Update app.py

app.py

@@ -20,9 +20,9 @@ def get_image_as_base64(image_path):
 # Load credentials from environment variables
 SERVICE_ACCOUNT_INFO = {
   "type": "service_account",
-  "project_id": os.environ.get("project_id"),
+  "project_id": os.environ.get("project_id",),
   "private_key_id": os.environ.get("private_key_id"),
-  "private_key": "-----BEGIN PRIVATE KEY-----\nMIIEvwIBADANBgkqhkiG9w0BAQEFAASCBKkwggSlAgEAAoIBAQCU+RlXQVzTifBI\nqDoj1yt2sLvFepy9edxEjAwAdyM0ICesvB6SCFpVGV3dV19c1PiwqqsnzGSGz3ji\nnSOxFXEve5pXFziCDVVIeaQ7g59GYLHOfKUtvqGewpxXm6gP1yrzHprpbHpBomXW\nnZNvIZ/SWzn8n4CjWV5A4V1mqfZq9OkBlJRsjJ0gh84g7QMQZ45lBrev1BUK5DtY\n8lMjP5XkQJGT4Z4km+qdsjMf+QCKLU5U2dsvNp1vViElKCwok9PfCeF+/fYX7WW5\nddpNkFC8/iOK9hf6jYbRW9Luo3OEqHV5uyqfBDQLHiZw4WuOvuY92TpWhkSlQsaN\nZiAQNR/lAgMBAAECggEAFDPJt3cCNx+8KaZCqCycVM630BqGGf3mHJefYwrFB+Ma\nbkbeoHAoi+TSQVuBb63kAdVLO6zwrwWAv76Oogul20oYTpo0DBLxz0/LsVRyq5Ee\nW8G0hsUU1tpOaGUQAAHubaTnXiNbWv5CcD3TSh1Vtqmvbe5VfBF5AjRtvafr5moV\n/oiKiOHNCR4QkA0kgynafTGd4UkZ+xekh6nWAQcj/Ri4BgYYckzHgIn/4eXE88p4\nq1L6JF2w0GHolLHaDpDBqii4CIhSc7i1jmM5NRriGfetzinR+ch0dVOt0ssdpkKK\nRxntHK93MhBUfue9/ZKZXlRhk7LfNyqfL99ghkhrIQKBgQDQ+QP7PD+JKuw+b6r/\n+2KhOu3A8diWp0euJsiHBZeIzmx9c7XLHbnv/H3NhLdhfGIBel/YmrT+hGUFiCa2\nn+7hGdgBjVt0EYGSxfBqwDpgBsPfEqsCIhOiLQSuECjfJV2XTRffhwVVzAwEQMan\nRwY6Pfh7VBSC7YTT7tCW+ohkMQKBgQC2f3uy4m7PdvONybirm8GTvVB0HdR0uDig\nDdg7iRuDc1YVksQ3I9Mi0TUTeTDNxeZ6eKNNY9EwIi61FKlacg11dXBCgXIHxwvw\ntBmeJffYLLijrF5D30pYJ39sO6YTdhXz8k2JQ9kjPygRA35ynD7frUYY8wVh2vId\n7VzSiBLN9QKBgQC7yMl3HnDJxjR89sPnO/Mph1bXJ3zGBJQ5Psi77J57ZWPAaBZY\n6+GGcpJp+HUBuVAom9s9PeifPkG5ctu3CIkU5rBg1LH13zVcu1qL0ymh2PJypsfd\nJtoVxfaKpCLWQzGTIMfvKdn1e2Lg/4SEFSLXe5Cq4qd78AyIlfYDdibSQQKBgQCW\n/KAN3tLI6GQWgbc8m4+rOKywdOGKc8O9f9bPYt8VgT3OeW/HIueS3b1o0BJ1nG7U\nEq0g4R14Ej/MZzy57LR/4lqvW0NJdf/F9GthcY49FrroBzqyMrHif0Mn+62qWj9N\nZcGKeEzUoXL3qIlcLzLBgi0FhUKFtEwuQGxUbmoIZQKBgQCYL9by084u0Sf1858i\narIVVg6baBK+MDnxG0YNTVovZv31n30Q0saT0KYZEVWbPDYL/Q+0zfn4qmI1L9Sd\nHedzZ1fREI61XwP/WxDojgZ22fkGbMVfnINjBwbWFMiKf5k7wQzhI5t+oOT0l9ti\n2GsPiT9eCYuz6WV4+Q9RXolrhQ==\n-----END PRIVATE KEY-----\n",
+   "private_key": os.environ.get("private_key"),
   "client_email": os.environ.get("client_email"),
   "client_id": os.environ.get("client_id"),
   "auth_uri": "https://accounts.google.com/o/oauth2/auth",
@@ -32,7 +32,6 @@ SERVICE_ACCOUNT_INFO = {
   "universe_domain": "googleapis.com"
 }
 
-
 SCOPES = ['https://www.googleapis.com/auth/drive']
 
 
@@ -183,43 +182,49 @@ def save_mapping_file_to_drive():
         for file in files:
             # Get the ID and name of the first Excel file found in the folder
             existing_file_id = file['id']
-            existing_file_name = file['name']
 
-            # Delete the existing file
-            service.files().delete(fileId=existing_file_id).execute()
+            # Overwrite the existing file
+            media = MediaFileUpload(MAPPING_FILENAME, mimetype='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
+            service.files().update(fileId=existing_file_id,media_body=media).execute()
 
-    file_metadata = {'name': MAPPING_FILENAME }
-    media = MediaFileUpload(MAPPING_FILENAME, mimetype='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
-    service.files().create(body=file_metadata, media_body=media, fields='id').execute()
 
-def pull_mapping_file_from_drive():
-    creds = authenticate()
-    # service = build('drive', 'v3', credentials=creds)
-    # creds = Credentials.from_authorized_user_info(credentials_dict)
-    # Authenticate with Google Drive API
-    service = build('drive', 'v3', credentials=creds)
+    # file_metadata = {'name': MAPPING_FILENAME }
+    # media = MediaFileUpload(MAPPING_FILENAME, mimetype='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
+    # service.files().create(body=file_metadata, media_body=media, fields='id').execute()
 
-    results = service.files().list(
-        q="mimeType='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'",
-        fields="files(id, name)").execute()
-    files = results.get('files', [])
-    files = [i for i in files if i.get('name')==MAPPING_FILENAME]    
-    if files:
-        file_id = files[0]['id']
-        file_name = files[0]['name']
-        request = service.files().get_media(fileId=file_id)
-        fh = open(file_name, 'wb')
-        downloader = MediaIoBaseDownload(fh, request)
-
-        # Execute the download
-        done = False
-        while not done:
-            status, done = downloader.next_chunk()
-
-        fh.close()
-        return 1
-    
-    return 0
+def pull_mapping_file_from_drive():
+    try:
+        creds = authenticate()
+        # service = build('drive', 'v3', credentials=creds)
+        # creds = Credentials.from_authorized_user_info(credentials_dict)
+        # Authenticate with Google Drive API
+        service = build('drive', 'v3', credentials=creds)
+
+        results = service.files().list(
+            q="mimeType='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'",
+            fields="files(id, name)").execute()
+        
+        files = results.get('files', [])
+        files = [i for i in files if i.get('name')==MAPPING_FILENAME]    
+        if files:
+            file_id = files[0]['id']
+            file_name = files[0]['name']
+            request = service.files().get_media(fileId=file_id)
+            fh = open(file_name, 'wb')
+            downloader = MediaIoBaseDownload(fh, request)
+
+            # Execute the download
+            done = False
+            while not done:
+                status, done = downloader.next_chunk()
+
+            fh.close()
+            return 1
+        
+        return 0
+    except Exception as e:
+        print(e)
+        return 0
 
 def delete_master_file():
     creds = authenticate()