Spaces:
Running
Running
Update pipeline.py
Browse files- pipeline.py +12 -0
pipeline.py
CHANGED
|
@@ -48,6 +48,18 @@ def get_or_create_drive_folder(name, parent_id=None):
|
|
| 48 |
file_metadata["parents"] = [parent_id]
|
| 49 |
file = drive_service.files().create(body=file_metadata, fields="id").execute()
|
| 50 |
return file["id"]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
|
| 52 |
def upload_file_to_drive(local_path, remote_name, folder_id):
|
| 53 |
file_metadata = {"name": remote_name, "parents": [folder_id]}
|
|
|
|
| 48 |
file_metadata["parents"] = [parent_id]
|
| 49 |
file = drive_service.files().create(body=file_metadata, fields="id").execute()
|
| 50 |
return file["id"]
|
| 51 |
+
def find_drive_file(filename, parent_id):
|
| 52 |
+
"""
|
| 53 |
+
Checks if a file with the given name exists inside the specified Google Drive folder.
|
| 54 |
+
Returns the file ID if found, else None.
|
| 55 |
+
"""
|
| 56 |
+
query = f"'{parent_id}' in parents and name = '{filename}' and trashed = false"
|
| 57 |
+
results = drive_service.files().list(q=query, spaces='drive', fields='files(id, name)', pageSize=1).execute()
|
| 58 |
+
files = results.get('files', [])
|
| 59 |
+
if files:
|
| 60 |
+
return files[0]["id"]
|
| 61 |
+
return None
|
| 62 |
+
|
| 63 |
|
| 64 |
def upload_file_to_drive(local_path, remote_name, folder_id):
|
| 65 |
file_metadata = {"name": remote_name, "parents": [folder_id]}
|