VyLala commited on
Commit
b29bcf1
·
verified ·
1 Parent(s): d015043

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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]}