1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
# ===================================================
# Send mail out through gmail server with
# Function:
#    send(sender,password,recipient,subject,message,afilename,afilepath)
# Send email through your own gmail account, attachment is supported.
# Notice: Make Sure to Have "Allow Less Secure Apps" On Your Gmail Account
#
# Fetch gmail with
# Function:
#   fetch(account,password,cnt,diskfile)
# Fetch email from gmail, extract texts, and serialize as a JSON file.
#
# Execute in command line as
#   Usage:
#    python gmail.py [send|fetch]
#
# ====================================================
import sys
import smtplib
import time
import imaplib
import email
import json
from email.MIMEMultipart import MIMEMultipart
from email.MIMEText import MIMEText
from email.MIMEBase import MIMEBase
from email import encoders

def send(sender,password,recipient,subject,message,afilename,afilepath):

 msg = MIMEMultipart()
 msg['From'] = sender
 msg['To'] = recipient
 msg['Subject'] = subject

 msg.attach(MIMEText(message,'plain'))

 # attachment
 if afilename != None:
  filename = afilename
  attachment = open(afilepath,'rb')

  part = MIMEBase('application','octet-stream')
  part.set_payload((attachment).read())
  encoders.encode_base64(part)
  part.add_header('Content-Disposition',"attachment; filename= %s" % afilename)

  msg.attach(part)

 server = smtplib.SMTP('smtp.gmail.com',587)
 server.starttls()
 server.login(sender,password)
 text = msg.as_string()
 server.sendmail(sender,recipient,text)
 server.quit()

def fetch(account,password,cnt,diskfile):
 #try:
  ext_mails = []
  mail = imaplib.IMAP4_SSL("imap.gmail.com")
  mail.login(account,password)
  mail.select('inbox')

  type,data = mail.search(None,'ALL')
  mail_ids = data[0]

  id_list = mail_ids.split()
  first_email_id = int(id_list[0])
  latest_email_id = int(id_list[-1])
  count = 0

  for i in range(latest_email_id,first_email_id,-1):
   count = count + 1
   if count > cnt:
    break
   ext_mail = {}
   type, data = mail.fetch(i,'(RFC822)')

   for response_part in data:
    if isinstance(response_part,tuple):
     msg = email.message_from_string(response_part[1])
     email_subject = msg['subject']
     ext_mail['subject'] = email_subject
     email_from = msg['from']
     ext_mail['from'] = email_from
     email_to = msg['to']
     ext_mail['to'] = email_to
     email_date = msg['date']
     ext_mail['date'] = email_date
     email_message = getText(msg)
     ext_mail['message'] = email_message
     ext_mails.append(ext_mail)

     print 'From : ' + email_from + '\n'
     print 'To : ' + email_to + '\n'
     print 'Subject: ' + email_subject + '\n'
     print 'Date: ' + email_date + '\n'
     print 'Message: \n' + email_message + '\n'
     print '-'*60 + '\n'
  # Write to disk as a json file
  with open(diskfile,'w+') as f:
   json.dump(ext_mails,f)
 #except:
 # print sys.exc_info()


def getText(msg):
 maintype = msg.get_content_maintype()
 if maintype == 'multipart':
  for part in msg.get_payload():
   parttype = part.get_content_maintype()
   #print parttype
   if parttype == 'multipart':
    # ------Recursively find the text part-------
    getText(part.get_payload())
   elif parttype == 'text':
    return part.get_payload()
 elif maintype == 'text':
  return msg.get_payload()

#  =================================
#  Usage:
#  python gmail.py [send|fetch]
#  =================================

if __name__ == "__main__":
 if len(sys.argv) > 1:
  if sys.argv[1] == "send":
   send('xxxx@gmail.com','xxxx','yyyy@example.com','Python Tutorial','To test is my python script works for sending email from this gmail account',None,None)
  elif sys.argv[1] == "fetch":
   fetch('xxxx@gmail.com','xxxx',3,'out/'+str(time.time()) + '.gmail.json')
  else:
   print "Usage: python gmail.py [send|fetch] \n"
 else:
  print "Usage: python gmail.py [send|fetch] \n"