Add GoogleApi Script

main/googleapi.py

@@ -0,0 +1,102 @@
+import os
+from google.oauth2 import service_account
+from googleapiclient.discovery import build
+from googleapiclient.http import MediaIoBaseDownload, MediaFileUpload
+import io
+
+# Define the Google Drive API scopes and service account file path
+SCOPES = ['https://www.googleapis.com/auth/drive']
+SERVICE_ACCOUNT_FILE = "mo-key.json"
+
+# Create credentials using the service account file
+credentials = service_account.Credentials.from_service_account_file(SERVICE_ACCOUNT_FILE, scopes=SCOPES)
+
+# Build the Google Drive service
+drive_service = build('drive', 'v3', credentials=credentials)
+
+def create_folder(folder_name, parent_folder_id=None):
+    """Create a folder in Google Drive and return its ID."""
+    folder_metadata = {
+        'name': folder_name,
+        "mimeType": "application/vnd.google-apps.folder",
+        'parents': [parent_folder_id] if parent_folder_id else []
+    }
+
+    created_folder = drive_service.files().create(
+        body=folder_metadata,
+        fields='id'
+    ).execute()
+
+    print(f'Created Folder ID: {created_folder["id"]}')
+    return created_folder["id"]
+
+def list_folder(parent_folder_id=None, delete=False):
+    """List folders and files in Google Drive."""
+    results = drive_service.files().list(
+        q=f"'{parent_folder_id}' in parents and trashed=false" if parent_folder_id else None,
+        pageSize=1000,
+        fields="nextPageToken, files(id, name, mimeType)"
+    ).execute()
+    items = results.get('files', [])
+
+    if not items:
+        print("No folders or files found in Google Drive.")
+    else:
+        print("Folders and files in Google Drive:")
+        for item in items:
+            print(f"Name: {item['name']}, ID: {item['id']}, Type: {item['mimeType']}")
+            if delete:
+                delete_files(item['id'])
+
+def delete_files(file_or_folder_id):
+    """Delete a file or folder in Google Drive by ID."""
+    try:
+        drive_service.files().delete(fileId=file_or_folder_id).execute()
+        print(f"Successfully deleted file/folder with ID: {file_or_folder_id}")
+    except Exception as e:
+        print(f"Error deleting file/folder with ID: {file_or_folder_id}")
+        print(f"Error details: {str(e)}")
+
+def download_file(file_id, destination_path):
+    """Download a file from Google Drive by its ID."""
+    request = drive_service.files().get_media(fileId=file_id)
+    fh = io.FileIO(destination_path, mode='wb')
+    
+    downloader = MediaIoBaseDownload(fh, request)
+    
+    done = False
+    while not done:
+        status, done = downloader.next_chunk()
+        print(f"Download {int(status.progress() * 100)}%.")
+
+def upload_basic():
+
+    file_metadata = {
+                    "name": "COMBO_MO_MOBIL.htm",
+                    'addParents': ['1M48IP3EIZ6fsrf_vFLLeDa_LOLqTpByH']
+                    }
+    media = MediaFileUpload("COMBO_MO_MOBIL.htm", mimetype="text/html")
+
+    file = (
+        drive_service.files()
+        .update(body=file_metadata, media_body=media, fileId="18CUUVcUcMvkIMkfB-Hi5SxztsJrwtiBc")
+        .execute()
+    )
+
+    return
+
+if __name__ == '__main__':
+    # Example usage:
+
+    # Create a new folder
+    # create_folder("MyNewFolder")
+    
+    # List folders and files
+    upload_basic()
+    # list_folder(parent_folder_id='1M48IP3EIZ6fsrf_vFLLeDa_LOLqTpByH')
+    
+    # Delete a file or folder by ID
+    # delete_files("1hw-p4uHKdYvlk7l3VnEWwO_dwSNoGWV4")
+
+    # Download a file by its ID
+    # download_file("your_file_id", "destination_path/file_name.extension")