Skip to content

Commit 7d1aacc

Browse files
add type hints
1 parent c923471 commit 7d1aacc

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

parsons/google/google_drive.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -192,7 +192,7 @@ def upsert_file(self, file_path: str, parent_folder_id: str) -> str:
192192
result = self.upload_file(file_path, parent_folder_id)
193193
return result
194194

195-
def get_permissions(self, file_id):
195+
def get_permissions(self, file_id: str) -> dict:
196196
"""
197197
`Args:`
198198
file_id: str
@@ -205,13 +205,19 @@ def get_permissions(self, file_id):
205205

206206
return p
207207

208-
def _share_object(self, file_id, permission_dict):
208+
def _share_object(self, file_id: str, permission_dict: dict) -> dict:
209209
# Send the request to share the file
210210
p = self.client.permissions().create(fileId=file_id, body=permission_dict).execute()
211211

212212
return p
213213

214-
def share_object(self, file_id, email_addresses=None, role="reader", type="user"):
214+
def share_object(
215+
self,
216+
file_id: str,
217+
email_addresses: Optional[list[str]] = None,
218+
role: str = "reader",
219+
type: str = "user",
220+
) -> list[dict]:
215221
"""
216222
`Args:`
217223
file_id: str
@@ -263,7 +269,7 @@ def share_object(self, file_id, email_addresses=None, role="reader", type="user"
263269

264270
return new_permissions
265271

266-
def transfer_ownership(self, file_id, new_owner_email):
272+
def transfer_ownership(self, file_id: str, new_owner_email: str) -> None:
267273
"""
268274
`Args:`
269275
file_id: str
@@ -295,5 +301,3 @@ def transfer_ownership(self, file_id, new_owner_email):
295301
logger.info(f"Ownership transferred successfully to {new_owner_email}.")
296302
else:
297303
logger.info("File does not have a current owner.")
298-
299-
return None

0 commit comments

Comments
 (0)