Create sheets_integration.py

sheets_integration.py

@@ -0,0 +1,53 @@
+# sheets_integration.py
+import os
+import json
+from google.oauth2.credentials import Credentials
+from google.oauth2 import service_account
+from googleapiclient.discovery import build
+from google.oauth2.credentials import Credentials
+import pandas as pd
+
+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"""
+        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()
+            return True
+        except Exception as e:
+            print(f"Error appending to sheet: {str(e)}")
+            return False
+    
+    def get_all_requests(self):
+        """Get all requests from the spreadsheet"""
+        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', []))
+        except Exception as e:
+            print(f"Error reading sheet: {str(e)}")
+            return None