Gmail api to create and upload files

chirathr · chirathr · commit 2fc0d57d1fab · 2018-03-13T10:28:06.000+05:30
diff --git a/msql-backup/google-drive-api.py b/msql-backup/google-drive-api.py
@@ -0,0 +1,86 @@
+import httplib2
+import os
+
+from googleapiclient.http import MediaFileUpload
+from oauth2client import client
+from oauth2client import tools
+from oauth2client.file import Storage
+
+try:
+    import argparse
+    flags = argparse.ArgumentParser(parents=[tools.argparser]).parse_args()
+except ImportError:
+    flags = None
+
+# If modifying these scopes, delete your previously saved credentials
+# at ~/.credentials/drive-python-quickstart.json
+SCOPES = 'https://www.googleapis.com/auth/drive.metadata.readonly'
+CLIENT_SECRET_FILE = 'client_secret.json'
+APPLICATION_NAME = 'Drive API Python Quickstart'
+
+
+def get_credentials():
+    """Gets valid user credentials from storage.
+
+    If nothing has been stored, or if the stored credentials are invalid,
+    the OAuth2 flow is completed to obtain the new credentials.
+
+    Returns:
+        Credentials, the obtained credential.
+    """
+    home_dir = os.path.expanduser('~')
+    credential_dir = os.path.join(home_dir, '.credentials')
+    if not os.path.exists(credential_dir):
+        os.makedirs(credential_dir)
+    credential_path = os.path.join(credential_dir,
+                                   'drive-python-quickstart.json')
+
+    store = Storage(credential_path)
+    credentials = store.get()
+    if not credentials or credentials.invalid:
+        flow = client.flow_from_clientsecrets(CLIENT_SECRET_FILE, SCOPES)
+        flow.user_agent = APPLICATION_NAME
+        credentials = tools.run_flow(flow, store, flags)
+        print('Storing credentials to ' + credential_path)
+    return credentials
+
+
+def upload_file(
+        drive_service, parent_folder_id,  file_name, file_path, mime_type):
+    """
+    :param drive_service: discovery.build('drive', 'v3', http=http)
+    :param parent_folder_id: ID of folder to upload the file
+    :param file_name: name of the file
+    :param file_path: path to the file
+    :param mime_type: mime type of file (ex: image/jpeg)
+    :return: File id from Google drive
+    """
+    file_metadata = {'name': file_name, 'parents': [parent_folder_id]}
+    media = MediaFileUpload(file_path,
+                            mimetype=mime_type)
+    file = drive_service.files().create(body=file_metadata,
+                                        media_body=media,
+                                        fields='id').execute()
+    return file.get('id')
+
+
+def create_folder(drive_service, parent_folder_id, folder_name):
+    """
+    :param drive_service: discovery.build('drive', 'v3', http=http)
+    :param parent_folder_id: ID of folder to upload the file
+    :param folder_name: Name of the folder to be created
+    :return:
+    """
+    file_metadata = {
+        'name': folder_name,
+        'parents': [parent_folder_id],
+        'mimeType': 'application/vnd.google-apps.folder'
+    }
+    folder = drive_service.files().create(body=file_metadata,
+                                          fields='id').execute()
+    return folder.get('id')
+
+
+if __name__ == '__main__':
+    credentials = get_credentials()
+    http = credentials.authorize(httplib2.Http())
diff --git a/status-update-checker/test.py b/status-update-checker/test.py
@@ -1,21 +1,17 @@
-import httplib2
 import os
-import base64
-import email
+import httplib2
 from apiclient import discovery
 from oauth2client import client
 from oauth2client import tools
 from oauth2client.file import Storage
 from apiclient import errors
 
-email_list = []
+try:
+    import argparse
 
-# try:
-#     import argparse
-#
-#     flags = argparse.ArgumentParser(parents=[tools.argparser]).parse_args()
-# except ImportError:
-#     flags = None
+    flags = argparse.ArgumentParser(parents=[tools.argparser]).parse_args()
+except ImportError:
+    flags = None
 
 # If modifying these scopes, delete your previously saved credentials
 # at ~/.credentials/gmail-python-quickstart.json
@@ -33,7 +29,6 @@ def get_credentials():
     Returns:
         Credentials, the obtained credential.
     """
-    # This function generates the required credentials required to make api calls
     home_dir = os.path.expanduser('~')
     credential_dir = os.path.join(home_dir, '.credentials')
     if not os.path.exists(credential_dir):
@@ -54,117 +49,111 @@ def get_credentials():
     return credentials
 
 
-def GetMsg(service, user_id, msg_id):
-    # This user defined message prints the email id's  and their total number emails,of the senders who has sent status updates except user's
-    try:
-        message = service.users().messages().get(userId=user_id, id=msg_id,
-                                                 format='metadata').execute()
-
-        payload = message["payload"]["headers"]
-
-        a = "X-Original-Sender"
-        # The below part prints the email id's
-        for d in payload:
-            if a in d["name"]:
-                email_list.append(d["value"])
-        return (email_list)
-        # mime_msg = email.message_from_string(msg_str)
-
-        return "mime_msg"
-    except errors.HttpError, error:
-        print ('An error occurred: %s' % error)
-
-
 def get_label_id(service, user_id, label_name):
-    foss_label_id = ""
+    """
+    Return the label id of a Gmail label
+    :param service: Authorized Gmail API service instance.
+    :param user_id: User's email address. The special value "me"
+    :param label_name: The name of the label id required
+    :return: string(label_id)
+    """
+    label_id = ''
     try:
         response = service.users().labels().list(userId=user_id).execute()
         labels = response['labels']
         for label in labels:
             if label['name'] == label_name:
-                # print('Label id: %s - Label name: %s' % (label['id'], label['name']))
-                foss_label_id = label['id']
-    except errors.HttpError, error:
-        print('An error occurred: %s' % error)
+                label_id = label['id']
+    except errors.HttpError:
+        print('An error occurred')
 
-    return foss_label_id
+    return label_id
 
 
-def ListMessagesWithLabels(service, user_id, label_ids=[]):
-    """List all Messages of the user's mailbox with label_ids applied.
+def list_messages_matching_query(service, user_id, query=''):
+    """List all Messages of the user's mailbox matching the query.
 
     Args:
       service: Authorized Gmail API service instance.
       user_id: User's email address. The special value "me"
       can be used to indicate the authenticated user.
-      label_ids: Only return Messages with these labelIds applied.
+      query: String used to filter messages returned.
+      Eg.- 'from:user@some_domain.com' for Messages from a particular sender.
 
     Returns:
-      List of Messages that have all required Labels applied. Note that the
+      List of Messages that match the criteria of the query. Note that the
       returned list contains Message IDs, you must use get with the
-      appropriate id to get the details of a Message.
+      appropriate ID to get the details of a Message.
     """
     try:
         response = service.users().messages().list(userId=user_id,
-                                                   labelIds=label_ids).execute()
+                                                   q=query).execute()
         messages = []
         if 'messages' in response:
             messages.extend(response['messages'])
 
         while 'nextPageToken' in response:
             page_token = response['nextPageToken']
-            response = service.users().messages().list(userId=user_id,
-                                                       labelIds=label_ids,
-                                                       pageToken=page_token).execute()
+            response = service.users().messages().list(
+                userId=user_id, q=query, pageToken=page_token).execute()
             messages.extend(response['messages'])
-
         return messages
-    except errors.HttpError, error:
-        print ('An error occurred: %s' % error)
-
 
-def ListMessagesMatchingQuery(service, user_id, query=''):
-    """List all Messages of the user's mailbox matching the query.
+    except errors.HttpError:
+        print('An error occurred')
 
-    Args:
-      service: Authorized Gmail API service instance.
-      user_id: User's email address. The special value "me"
-      can be used to indicate the authenticated user.
-      query: String used to filter messages returned.
-      Eg.- 'from:user@some_domain.com' for Messages from a particular sender.
 
-    Returns:
-      List of Messages that match the criteria of the query. Note that the
-      returned list contains Message IDs, you must use get with the
-      appropriate ID to get the details of a Message.
+def get_sender_email_id(service, user_id, msg_id):
+    """
+    This functions return the email id of the sender
+    :param service: The google service(here Gmail)
+    :param user_id: The user id of the user
+    :param msg_id: The message id that is to be retrived
+    :return: string(email id)
     """
+    email_id = ''
+
     try:
-        response = service.users().messages().list(userId=user_id,
-                                                   q=query).execute()
-        messages = []
-        if 'messages' in response:
-            messages.extend(response['messages'])
+        message = service.users().messages().get(userId=user_id, id=msg_id,
+                                                 format='metadata').execute()
 
-        while 'nextPageToken' in response:
-            page_token = response['nextPageToken']
-            response = service.users().messages().list(userId=user_id, q=query,
-                                                       pageToken=page_token).execute()
-            messages.extend(response['messages'])
+        header_data = message["payload"]["headers"]
 
-        return messages
-    except errors.HttpError, error:
-        print ('An error occurred: %s' % error)
+        sender_text = "X-Original-Sender"
+
+        # Get email id from header data
+        for data in header_data:
+            if sender_text in data["name"]:
+                email_id = data["value"]
+        return email_id
+
+    except errors.HttpError:
+        print('An error occurred')
 
 
-def main(date):
-    # This is the function which asks the user for his query
+def get_status_update_emails(status_update_date):
+    """
+
+    :param status_update_date: Date of the status update you want to query
+    :return:
+    """
     credentials = get_credentials()
     http = credentials.authorize(httplib2.Http())
     service = discovery.build('gmail', 'v1', http=http)
-    query = '[foss-2017] Status Update [' + date + ']'
-    message = []
-    messgs = ListMessagesMatchingQuery(service, user_id='me', query=query)
-    for item in messgs:
-        message = GetMsg(service, user_id="me", msg_id=item['id'])
-    return message
+    status_update_date_string = status_update_date.strftime('%d-%m-%Y')
+    query = '[foss-2017] Status Update [%s]' % status_update_date_string
 
+    emails = []
+    messages = list_messages_matching_query(service, user_id='me', query=query)
+
+    for message in messages:
+        email = get_sender_email_id(
+            service, user_id="me", msg_id=message['id'])
+        emails.append(email)
+
+    return emails
+
+
+if __name__ == '__main__':
+    credentials = get_credentials()
+    http = credentials.authorize(httplib2.Http())
