fix: attribute errors (#429)

Author: vinay-google

drive/snippets/drive-v2/file snippet/create_folder.py

@@ -43,13 +43,12 @@ def create_folder():
         # pylint: disable=maybe-no-member
         file = service.files().insert(body=file_metadata,
                                       fields='id').execute()
-        print(F'Folder has created with ID: "{file.get("id")}".')
+        print(F'Folder ID: "{file.get("id")}".')
+        return file.get('id')
 
     except HttpError as error:
         print(F'An error occurred: {error}')
-        file = None
-
-    return file.get('id')
+        return None
 
 
 if __name__ == '__main__':

drive/snippets/drive-v2/file snippet/move_file_to_folder.py

@@ -22,12 +22,13 @@
 from googleapiclient.errors import HttpError
 
 
-def move_file_to_folder(real_file_id, real_folder_id):
+def move_file_to_folder(file_id, folder_id):
     """Move specified file to the specified folder.
     Args:
-        real_file_id: Id of the file to move.
-        real_folder_id: Id of the folder
+        file_id: Id of the file to move.
+        folder_id: Id of the folder
     Print: An object containing the new parent folder and other meta data
+    Returns : Parent Ids for the file
 
     Load pre-authorized user credentials from the environment.
     TODO(developer) - See https://developers.google.com/identity
@@ -39,9 +40,6 @@ def move_file_to_folder(real_file_id, real_folder_id):
         # call drive api client
         service = build('drive', 'v2', credentials=creds)
 
-        file_id = real_file_id
-        folder_id = real_folder_id
-
         # Retrieve the existing parents to remove
         # pylint: disable=maybe-no-member
         file = service.files().get(fileId=file_id, fields='parents').execute()
@@ -52,16 +50,16 @@ def move_file_to_folder(real_file_id, real_folder_id):
                                       removeParents=previous_parents,
                                       fields='id, parents').execute()
         new_parent_folder_id = [parent["id"] for parent in file.get('parents')]
-        print(F'file with ID : {file.get("id")} has moved to folder : '
+        print(F'file with ID : {file.get("id")} moved to folder : '
               F'{new_parent_folder_id}')
+        return [parent["id"] for parent in file.get('parents')]
 
     except HttpError as error:
         print(F'An error occurred: {error}')
-
-    return [parent["id"] for parent in file.get('parents')]
+        return None
 
 
 if __name__ == '__main__':
-    move_file_to_folder(real_file_id='14fesChjgzDA7lUu9ZeGqXOuXMPgaVkxS',
-                        real_folder_id='1KzT9gjq-AHfciwNzKjh7nUd6prrQOA4')
+    move_file_to_folder(file_id='14fesChjgzDA7lUu9ZeGqXOuXMPgaVkxS',
+                        folder_id='1KzT9gjq-AHfciwNzKjh7nUd6prrQOA4')
 # [END drive_move_file_to_folder]

drive/snippets/drive-v2/file snippet/upload_to_folder.py

@@ -23,7 +23,7 @@
 from googleapiclient.http import MediaFileUpload
 
 
-def upload_to_folder(real_folder_id):
+def upload_to_folder(folder_id):
     """Upload a file to the specified folder and prints file ID, folder ID
     Args: Id of the folder
     Returns: ID of the file uploaded
@@ -38,7 +38,6 @@ def upload_to_folder(real_folder_id):
         # create drive api client
         service = build('drive', 'v2', credentials=creds)
 
-        folder_id = real_folder_id
         file_metadata = {
             'title': 'photo.jpg',
             'parents': [{'id': folder_id}]
@@ -48,16 +47,14 @@ def upload_to_folder(real_folder_id):
         # pylint: disable=maybe-no-member
         file = service.files().insert(body=file_metadata,
                                       media_body=media, fields='id').execute()
-        print(F'File with ID: "{file.get("id")}" has added to the folder with '
-              F'ID "{real_folder_id}".')
+        print(F'File ID: "{file.get("id")}".')
+        return file.get('id')
 
     except HttpError as error:
         print(F'An error occurred: {error}')
-        file = None
-
-    return file.get('id')
+        return None
 
 
 if __name__ == '__main__':
-    upload_to_folder(real_folder_id='1s0oKEZZXjImNngxHGnY0xed6Mw-tvspu')
+    upload_to_folder(folder_id='1s0oKEZZXjImNngxHGnY0xed6Mw-tvspu')
 # [END drive_upload_to_folder]

drive/snippets/drive-v3/file_snippet/create_folder.py

@@ -43,13 +43,12 @@ def create_folder():
         # pylint: disable=maybe-no-member
         file = service.files().create(body=file_metadata, fields='id'
                                       ).execute()
-        print(F'Folder has created with ID: "{file.get("id")}".')
+        print(F'Folder ID: "{file.get("id")}".')
+        return file.get('id')
 
     except HttpError as error:
         print(F'An error occurred: {error}')
-        file = None
-
-    return file.get('id')
+        return None
 
 
 if __name__ == '__main__':

drive/snippets/drive-v3/file_snippet/move_file_to_folder.py

@@ -22,12 +22,13 @@
 from googleapiclient.errors import HttpError
 
 
-def move_file_to_folder(real_file_id, real_folder_id):
+def move_file_to_folder(file_id, folder_id):
     """Move specified file to the specified folder.
     Args:
-        real_file_id: Id of the file to move.
-        real_folder_id: Id of the folder
+        file_id: Id of the file to move.
+        folder_id: Id of the folder
     Print: An object containing the new parent folder and other meta data
+    Returns : Parent Ids for the file
 
     Load pre-authorized user credentials from the environment.
     TODO(developer) - See https://developers.google.com/identity
@@ -39,9 +40,6 @@ def move_file_to_folder(real_file_id, real_folder_id):
         # call drive api client
         service = build('drive', 'v3', credentials=creds)
 
-        file_id = real_file_id
-        folder_id = real_folder_id
-
         # pylint: disable=maybe-no-member
         # Retrieve the existing parents to remove
         file = service.files().get(fileId=file_id, fields='parents').execute()
@@ -50,15 +48,14 @@ def move_file_to_folder(real_file_id, real_folder_id):
         file = service.files().update(fileId=file_id, addParents=folder_id,
                                       removeParents=previous_parents,
                                       fields='id, parents').execute()
+        return file.get('parents')
 
     except HttpError as error:
         print(F'An error occurred: {error}')
-        file = None
-
-    return file.get('parents')
+        return None
 
 
 if __name__ == '__main__':
-    move_file_to_folder(real_file_id='1KuPmvGq8yoYgbfW74OENMCB5H0n_2Jm9',
-                        real_folder_id='1jvTFoyBhUspwDncOTB25kb9k0Fl0EqeN')
+    move_file_to_folder(file_id='1KuPmvGq8yoYgbfW74OENMCB5H0n_2Jm9',
+                        folder_id='1jvTFoyBhUspwDncOTB25kb9k0Fl0EqeN')
 # [END drive_move_file_to_folder]

drive/snippets/drive-v3/file_snippet/upload_to_folder.py

@@ -23,7 +23,7 @@
 from googleapiclient.http import MediaFileUpload
 
 
-def upload_to_folder(real_folder_id):
+def upload_to_folder(folder_id):
     """Upload a file to the specified folder and prints file ID, folder ID
     Args: Id of the folder
     Returns: ID of the file uploaded
@@ -38,7 +38,6 @@ def upload_to_folder(real_folder_id):
         # create drive api client
         service = build('drive', 'v3', credentials=creds)
 
-        folder_id = real_folder_id
         file_metadata = {
             'name': 'photo.jpg',
             'parents': [folder_id]
@@ -48,16 +47,14 @@ def upload_to_folder(real_folder_id):
         # pylint: disable=maybe-no-member
         file = service.files().create(body=file_metadata, media_body=media,
                                       fields='id').execute()
-        print(F'File with ID: "{file.get("id")}" has added to the folder with '
-              F'ID "{real_folder_id}".')
+        print(F'File ID: "{file.get("id")}".')
+        return file.get('id')
 
     except HttpError as error:
         print(F'An error occurred: {error}')
-        file = None
-
-    return file.get('id')
+        return None
 
 
 if __name__ == '__main__':
-    upload_to_folder(real_folder_id='1s0oKEZZXjImNngxHGnY0xed6Mw-tvspu')
+    upload_to_folder(folder_id='1s0oKEZZXjImNngxHGnY0xed6Mw-tvspu')
 # [END drive_upload_to_folder]