gmail.py

#!/usr/bin/python
# Filename: gmail.py
"""Sends email using OAuth and Gmail.""" 

import pickle 

import base64
import httplib2

from email.mime.text import MIMEText
from email.mime.image import MIMEImage
from email.mime.multipart import MIMEMultipart

from apiclient.discovery import build
from oauth2client.client import flow_from_clientsecrets
from oauth2client.file import Storage
from oauth2client.tools import run

from os import environ as env
from datetime import datetime

body = ""
recipient = ""
sender = ""
subject = ""

def configure():
	"""This function allows you to configure the location of you client_secret.json file."""
	print ("gmail.configure allows you to set or change the location of client_secret.json file. This must be ran interactively!")
	loc = raw_input("Please type the absolute path of the client_secret.json file: ")
	print ("Hold on a sec'")
	pickle.dump(loc, open("gmConfig.p", "wb"))
	print ("Saved! You will not need to run this again unless you move your file or run this on a new computer.")

def sendMail(body, recipient, sender, subject):
	"""This function allows you to compose and send mail. It takes 4 arguments (body, recipient, sender, and subject)"""
	
	
	# Path to the client_secret.json file downloaded from the Developer Console
	CLIENT_SECRET_FILE = pickle.load( open( "gmConfig.p", "rb" ) )

	# Check https://developers.google.com/gmail/api/auth/scopes for all available scopes
	OAUTH_SCOPE = 'https://www.googleapis.com/auth/gmail.compose'

	# Location of the credentials storage file
	STORAGE = Storage('gmail.storage')

	# Start the OAuth flow to retrieve credentials
	flow = flow_from_clientsecrets(CLIENT_SECRET_FILE, scope=OAUTH_SCOPE)
	http = httplib2.Http()

	# Try to retrieve credentials from storage or run the flow to generate them
	credentials = STORAGE.get()
	if credentials is None or credentials.invalid:
	  credentials = run(flow, STORAGE, http=http)

	# Authorize the httplib2.Http object with our credentials
	http = credentials.authorize(http)

	# Build the Gmail service from discovery
	gmail_service = build('gmail', 'v1', http=http)

	# create a message to send
	message = MIMEText(body)
	message['to'] = (recipient)
	message['from'] = (sender)
	message['subject'] = (subject)
	body = {'raw': base64.b64encode(message.as_string())}

	# send it

	try:
	  message = (gmail_service.users().messages().send(userId="me", body=body).execute())
	  #print('Message Id: %s' % message['id'])
	  #print(message)
	  print("Message sent to: " + recipient + " @ " + datetime.now().strftime('%Y-%m-%d %I:%M:%S:%p'))
	except Exception as error:
	  print('An error occurred while sending a message. Maybe this error code will help: %s' % error)