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]}
|