Upload logs.py

logs.py

@@ -0,0 +1,50 @@
+import os
+from google.oauth2 import service_account
+from googleapiclient.discovery import build
+from googleapiclient.http import MediaFileUpload
+from datetime import datetime
+
+def save_logs(query,response, folder_id = "16Vv728HPW2J0BYzgTaBV00nUEc5pRKT-"):
+    to_save = f"LOG ENTRY\nQUERY\n{query}\n=================================\nRESPONSE\n{response}\n****************************************\n"
+
+    # Get the current date and time
+    now = datetime.now()
+    filename = str(now).replace(":","").replace(" ","").replace("-","").replace(".","")+".txt"
+    with open(filename, 'w') as file:
+        file.write(to_save)
+    # Path to the service account key file
+    SERVICE_ACCOUNT_FILE = 'secret_google_service_account.json'
+
+    # Define the required scopes
+    SCOPES = ['https://www.googleapis.com/auth/drive.file']
+
+    # Authenticate using the service account key file
+    credentials = service_account.Credentials.from_service_account_file(
+        SERVICE_ACCOUNT_FILE, scopes=SCOPES)
+
+    # Build the Google Drive API client
+    service = build('drive', 'v3', credentials=credentials)
+
+    # Specify the folder ID where you want to upload the file
+
+    # Metadata of the file to be uploaded
+    file_metadata = {
+        'name': filename,  # Name of the file to be uploaded
+        'parents': [folder_id]  # Folder ID
+    }
+
+    # Path to the file you want to upload
+    file_path = filename
+
+    # Create a MediaFileUpload object to upload the file
+    media = MediaFileUpload(file_path, mimetype='text/plain')
+
+    # Use the Drive API to upload the file
+    file = service.files().create(
+        body=file_metadata,
+        media_body=media,
+        fields='id'
+    ).execute()
+
+    # Print the file ID of the uploaded file
+    print('Saved in Google Drive - File ID: %s' % file.get('id'))