Update sheets_integration.py

sheets_integration.py

@@ -1,53 +1,83 @@
-# sheets_integration.py
 import os
 import json
-from google.oauth2.credentials import Credentials
+import gspread
 from google.oauth2 import service_account
-from googleapiclient.discovery import build
-from google.oauth2.credentials import Credentials
-import pandas as pd
+from datetime import datetime
 
-class SheetsManager:
-    def __init__(self, credentials_path, spreadsheet_id):
-        self.credentials_path = credentials_path
-        self.spreadsheet_id = spreadsheet_id
-        self.credentials = None
-        self.service = None
-        self.initialize_service()
-    
-    def initialize_service(self):
-        """Initialize Google Sheets API service"""
-        self.credentials = service_account.Credentials.from_service_account_file(
-            self.credentials_path,
-            scopes=['https://www.googleapis.com/auth/spreadsheets']
-        )
-        self.service = build('sheets', 'v4', credentials=self.credentials)
-    
-    def append_row(self, row_data):
-        """Append a row to the spreadsheet"""
+class SheetsLogger:
+    def __init__(self):
+        self.gc = None
+        self.sheet = None
+        self.initialize_client()
+        
+    def initialize_client(self):
+        """Initialize Google Sheets client"""
         try:
-            body = {
-                'values': [list(row_data.values())]
-            }
-            result = self.service.spreadsheets().values().append(
-                spreadsheetId=self.spreadsheet_id,
-                range='Sheet1!A:I',  # Adjust range as needed
-                valueInputOption='RAW',
-                body=body
-            ).execute()
+            # 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")
+            
+            # Parse credentials JSON
+            creds_dict = json.loads(creds_json)
+            
+            # Create credentials object
+            credentials = service_account.Credentials.from_service_account_info(
+                creds_dict,
+                scopes=['https://spreadsheets.google.com/feeds',
+                       'https://www.googleapis.com/auth/spreadsheets',
+                       'https://www.googleapis.com/auth/drive']
+            )
+            
+            # Initialize gspread client
+            self.gc = gspread.authorize(credentials)
+            
+            # Open spreadsheet by ID (get this from your sheet's URL)
+            SHEET_ID = os.getenv('GOOGLE_SHEET_ID')
+            self.sheet = self.gc.open_by_key(SHEET_ID).sheet1
+            
+        except Exception as e:
+            print(f"Error initializing Google Sheets client: {str(e)}")
+            self.gc = None
+            self.sheet = None
+
+    def log_request(self, request_data):
+        """Log a request to Google Sheets"""
+        if not self.sheet:
+            print("Sheet not initialized")
+            return False
+            
+        try:
+            # Prepare row data
+            row = [
+                datetime.now().strftime("%Y-%m-%d %H:%M:%S"),  # Timestamp
+                request_data.get('name', ''),
+                request_data.get('employee_id', ''),
+                request_data.get('email', ''),
+                request_data.get('department', ''),
+                request_data.get('request_details', ''),
+                request_data.get('frequency', ''),
+                request_data.get('urgency', ''),
+                request_data.get('system_response', '')
+            ]
+            
+            # Append row to sheet
+            self.sheet.append_row(row)
             return True
+            
         except Exception as e:
-            print(f"Error appending to sheet: {str(e)}")
+            print(f"Error logging to sheet: {str(e)}")
             return False
-    
+
     def get_all_requests(self):
-        """Get all requests from the spreadsheet"""
+        """Get all requests from the sheet"""
+        if not self.sheet:
+            return None
+            
         try:
-            result = self.service.spreadsheets().values().get(
-                spreadsheetId=self.spreadsheet_id,
-                range='Sheet1!A:I'  # Adjust range as needed
-            ).execute()
-            return pd.DataFrame(result.get('values', []))
+            # Get all records
+            records = self.sheet.get_all_records()
+            return records
         except Exception as e:
-            print(f"Error reading sheet: {str(e)}")
+            print(f"Error getting records: {str(e)}")
             return None