Update sheets_integration.py

sheets_integration.py

@@ -1,9 +1,11 @@
+# sheets_integration.py
 import os
 import json
 from datetime import datetime
 from enum import Enum
 import gspread
 from google.oauth2 import service_account
+import traceback
 
 class RequestStatus(Enum):
     NEW = "New"
@@ -17,6 +19,7 @@ class RequestStatus(Enum):
 
 class SheetsLogger:
     def __init__(self):
+        print("Initializing SheetsLogger")
         self.gc = None
         self.sheet = None
         self.initialize_client()
@@ -27,10 +30,16 @@ class SheetsLogger:
             # Get credentials from environment variable
             creds_json = os.getenv('GOOGLE_SHEETS_CREDS')
             if not creds_json:
-                raise ValueError("Google Sheets credentials not found in environment")
-            
+                print("Google Sheets credentials not found in environment")
+                return
+
             # Parse credentials JSON
-            creds_dict = json.loads(creds_json)
+            try:
+                creds_dict = json.loads(creds_json)
+                print("Credentials JSON parsed successfully")
+            except json.JSONDecodeError as e:
+                print(f"Error parsing credentials JSON: {e}")
+                return
             
             # Create credentials object
             credentials = service_account.Credentials.from_service_account_info(
@@ -44,17 +53,20 @@ class SheetsLogger:
             self.gc = gspread.authorize(credentials)
             
             # Open spreadsheet by ID
-            SHEET_ID = os.getenv('GOOGLE_SHEET_ID')
-            if not SHEET_ID:
-                raise ValueError("Google Sheet ID not found in environment")
+            sheet_id = os.getenv('GOOGLE_SHEET_ID')
+            if not sheet_id:
+                print("Google Sheet ID not found in environment")
+                return
                 
-            self.sheet = self.gc.open_by_key(SHEET_ID).sheet1
+            self.sheet = self.gc.open_by_key(sheet_id).sheet1
+            print("Sheet connected successfully")
             
             # Initialize headers if needed
             self.initialize_sheet()
             
         except Exception as e:
             print(f"Error initializing Google Sheets client: {str(e)}")
+            print(traceback.format_exc())
             self.gc = None
             self.sheet = None
     
@@ -80,21 +92,22 @@ class SheetsLogger:
             # Check if headers exist
             existing_headers = self.sheet.row_values(1)
             if not existing_headers:
+                print("Adding headers to sheet")
                 self.sheet.insert_row(headers, 1)
                 # Format header row
                 self.sheet.format('A1:M1', {
                     "backgroundColor": {"red": 0.2, "green": 0.2, "blue": 0.2},
                     "textFormat": {"bold": True, "foregroundColor": {"red": 1, "green": 1, "blue": 1}}
                 })
-                # Set column widths
-                self.sheet.format('A:M', {
-                    "wrapStrategy": "WRAP"
-                })
+            else:
+                print("Headers already exist")
         except Exception as e:
             print(f"Error initializing headers: {str(e)}")
+            print(traceback.format_exc())
 
-    def log_request(self, data):
+    def log_request(self, data: dict) -> bool:
         """Log a request to Google Sheets"""
+        print("Attempting to log request to sheets")
         if not self.sheet:
             print("Sheet not initialized")
             return False
@@ -102,7 +115,7 @@ class SheetsLogger:
         try:
             # Prepare row data
             row = [
-                data.get('timestamp', datetime.now().strftime("%Y-%m-%d %H:%M:%S")),
+                datetime.now().strftime("%Y-%m-%d %H:%M:%S"),  # Timestamp
                 data.get('name', ''),
                 data.get('employee_id', ''),
                 data.get('email', ''),
@@ -118,12 +131,14 @@ class SheetsLogger:
             ]
             
             # Append row to sheet
+            print("Appending row to sheet")
             self.sheet.append_row(row)
             
             # Get the row number that was just added
             last_row = len(self.sheet.get_all_values())
             
             # Apply conditional formatting for status
+            print("Applying status formatting")
             self.apply_status_formatting(last_row)
             
             print(f"Successfully logged request to row {last_row}")
@@ -131,36 +146,19 @@ class SheetsLogger:
             
         except Exception as e:
             print(f"Error logging to sheet: {str(e)}")
+            print(traceback.format_exc())
             return False
 
-    def apply_status_formatting(self, row_number):
+    def apply_status_formatting(self, row_number: int):
         """Apply conditional formatting based on status"""
         status_cell = f'K{row_number}'
         formats = {
-            RequestStatus.NEW.value: {
-                "backgroundColor": {"red": 1, "green": 1, "blue": 0.8}  # Light yellow
-            },
-            RequestStatus.IN_PROGRESS.value: {
-                "backgroundColor": {"red": 0.8, "green": 0.9, "blue": 1}  # Light blue
-            },
-            RequestStatus.PENDING_INFO.value: {
-                "backgroundColor": {"red": 1, "green": 0.8, "blue": 0.8}  # Light red
-            },
-            RequestStatus.DATA_COLLECTION.value: {
-                "backgroundColor": {"red": 0.8, "green": 1, "blue": 0.8}  # Light green
-            },
-            RequestStatus.READY_FOR_REVIEW.value: {
-                "backgroundColor": {"red": 0.9, "green": 0.9, "blue": 1}  # Light purple
-            },
-            RequestStatus.COMPLETED.value: {
-                "backgroundColor": {"red": 0.9, "green": 1, "blue": 0.9}  # Lighter green
-            },
-            RequestStatus.CANCELLED.value: {
-                "backgroundColor": {"red": 0.9, "green": 0.9, "blue": 0.9}  # Gray
-            },
-            RequestStatus.ON_HOLD.value: {
-                "backgroundColor": {"red": 1, "green": 0.9, "blue": 0.8}  # Light orange
-            }
+            RequestStatus.NEW.value: {"backgroundColor": {"red": 1, "green": 1, "blue": 0.8}},
+            RequestStatus.IN_PROGRESS.value: {"backgroundColor": {"red": 0.8, "green": 0.9, "blue": 1}},
+            RequestStatus.PENDING_INFO.value: {"backgroundColor": {"red": 1, "green": 0.8, "blue": 0.8}},
+            RequestStatus.DATA_COLLECTION.value: {"backgroundColor": {"red": 0.8, "green": 1, "blue": 0.8}},
+            RequestStatus.READY_FOR_REVIEW.value: {"backgroundColor": {"red": 0.9, "green": 0.9, "blue": 1}},
+            RequestStatus.COMPLETED.value: {"backgroundColor": {"red": 0.9, "green": 1, "blue": 0.9}}
         }
         
         try:
@@ -168,59 +166,4 @@ class SheetsLogger:
             if status in formats:
                 self.sheet.format(status_cell, formats[status])
         except Exception as e:
-            print(f"Error applying formatting: {str(e)}")
-
-    def update_status(self, row_number, new_status, notes=None):
-        """Update the status and notes of a request"""
-        try:
-            if not isinstance(new_status, str):
-                if isinstance(new_status, RequestStatus):
-                    new_status = new_status.value
-                else:
-                    raise ValueError("Invalid status type")
-
-            # Update status
-            self.sheet.update_cell(row_number, 11, new_status)  # Column K
-            
-            # Update notes if provided
-            if notes:
-                current_notes = self.sheet.cell(row_number, 12).value  # Column L
-                timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-                updated_notes = f"{current_notes}\n{timestamp}: {notes}" if current_notes else f"{timestamp}: {notes}"
-                self.sheet.update_cell(row_number, 12, updated_notes)
-            
-            # Update last updated timestamp
-            self.sheet.update_cell(row_number, 13, datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
-            
-            # Apply formatting
-            self.apply_status_formatting(row_number)
-            
-            return True
-        except Exception as e:
-            print(f"Error updating status: {str(e)}")
-            return False
-
-    def get_request_history(self, request_id=None, employee_id=None):
-        """Get request history filtered by ID or employee"""
-        try:
-            all_data = self.sheet.get_all_records()
-            if request_id:
-                return [row for row in all_data if str(row.get('Employee ID', '')).strip() == str(request_id).strip()]
-            elif employee_id:
-                return [row for row in all_data if str(row.get('Employee ID', '')).strip() == str(employee_id).strip()]
-            return all_data
-        except Exception as e:
-            print(f"Error getting history: {str(e)}")
-            return []
-
-    def get_request_by_row(self, row_number):
-        """Get a specific request by row number"""
-        try:
-            if row_number < 2:  # Row 1 is headers
-                return None
-            row_data = self.sheet.row_values(row_number)
-            headers = self.sheet.row_values(1)
-            return dict(zip(headers, row_data))
-        except Exception as e:
-            print(f"Error getting row {row_number}: {str(e)}")
-            return None
+            print(f"Error applying formatting: {str(e)}")