gmail_api.py

# for parsing commandline arguments
import argparse
from common import search_messages, gmail_authenticate
from read_emails import read_message
from send_emails import send_message
from delete_emails import delete_messages
from mark_emails import mark_as_read, mark_as_unread


if __name__ == '__main__':
 parser = argparse.ArgumentParser(description="Send/Search/Delete/Mark messages using gmail's API.")
 subparsers = parser.add_subparsers(help='Subcommands')
 parser_1 = subparsers.add_parser('send', help='Send an email')
 parser_1.add_argument('destination', type=str, help='The destination email address')
 parser_1.add_argument('subject', type=str, help='The subject of the email')
 parser_1.add_argument('body', type=str, help='The body of the email')
 parser_1.add_argument('files', type=str, help='email attachments', nargs='+')
 parser_1.set_defaults(action='send')
 parser_2 = subparsers.add_parser('delete', help='Delete a set of emails')
 parser_2.add_argument('query', type=str, help='a search query that selects emails to delete')
 parser_2.set_defaults(action='delete')
 parser_3 = subparsers.add_parser('mark', help='Marks a set of emails as read or unread')
 parser_3.add_argument('query', type=str, help='a search query that selects emails to mark')
 parser_3.add_argument('read_status', type=bool, help='Whether to mark the message as unread, or as read')
 parser_3.set_defaults(action='mark')
 parser_4 = subparsers.add_parser('search', help='Marks a set of emails as read or unread')
 parser_4.add_argument('query', type=str, help='a search query, which messages to display')
 parser_4.set_defaults(action='search')
 args = parser.parse_args()
 service = gmail_authenticate()
 if args.action == 'send':
 # TODO: add attachements
 send_message(service, args.destination, args.subject, args.body, args.files)
 elif args.action == 'delete':
 delete_messages(service, args.query)
 elif args.action == 'mark':
 print(args.unread_status)
 if args.read_status:
 mark_as_read(service, args.query)
 else:
 mark_as_unread(service, args.query)
 elif args.action == 'search':
 results = search_messages(service, args.query)
 for msg in results:
 read_message(service, msg)