xmailer.py

   1 #!/usr/bin/python
   2 #- * - encoding: utf-8 - * -
   3 import email,mimetypes
   4 import string, sys, os
   5 import time
   6 import ConfigParser
   7 import smtplib
   8 import glob
   9
  10 from email import encoders
  11 from email.mime.audio import MIMEAudio
  12 from email.mime.base import MIMEBase
  13 from email.mime.image import MIMEImage
  14 from email.mime.multipart import MIMEMultipart
  15 from email.mime.text import MIMEText
  16
  17 from optparse import OptionParser
  18
  19 import smtpser
  20
  21 def ReadConfig(filename):
  22    """ read param from config file: filename """
  23    rc=ConfigParser.ConfigParser()
  24    rc.read(filename)
  25    maxlen=rc.getint('global','maillen')*1024
  26    smtpcount=rc.getint('global','smtp')
  27    defaultsmtp=rc.get('global','default')
  28    #only use default or use each smtp orderly
  29    onlydefault=rc.getint('global','onlydefault')
  30    tolist=rc.get('global','to').split(',')
  31    param=[]
  32    servernamelist=[]
  33    param.append([maxlen,smtpcount,defaultsmtp,onlydefault,tolist])
  34    smpts={}
  35    for i in xrange(1,smtpcount+1):
  36       sname='smtp'+'%d'%i
  37       key=rc.get(sname,'name')
  38       addr=rc.get(sname,'addr')
  39       port=rc.getint(sname,'port')
  40       ssl=rc.getint(sname,'ssl')
  41       username=rc.get(sname,'user')
  42       password=rc.get(sname,'pass')
  43       mailaddress=rc.get(sname,'mail')
  44       smpts[key]=[addr,port,ssl,username,password,mailaddress]
  45       servernamelist.append(key)
  46    param.append(smpts)
  47    param.append(servernamelist)
  48
  49    return param
  50
  51 def GetOption():
  52    """ get option from command line """
  53    parser = OptionParser(usage="""\
  54 Send the contents of a directory as a MIME message.
  55
  56 Usage: %prog [options] [directory]
  57
  58 """)
  59    parser.add_option('-s', '--subject',
  60                       type='string', action='store', metavar='SUBJECT',
  61                       help='The value of the Subject: header (required)')
  62    parser.add_option('-r', '--recipient',
  63                       type='string', action='append', metavar='RECIPIENT',
  64                       default=[], dest='recipients',
  65                       help='A To: header value (at least one required)')
  66    parser.add_option('', '--bcc',
  67                       type='string', action='append', metavar='BCC',
  68                       default=[], dest='bccs',
  69                       help='A BCC: send by BCC')
  70    parser.add_option('-f', '--file',
  71                       type='string', action='append', metavar='FILE',
  72                       default=[], dest='files',
  73                       help="A File point out the files to be mailed. Example: '*.jpg' , '../*.pdf' etc. DON NOT TO FORGET THE '. If this option , we will discard directory argument")
  74    parser.add_option('-c','--configfile',
  75                      type='string',action='store',metavar='CONFIGFILE',
  76                      help="The configfile, for Linux is '~/.xmailrc'")
  77    parser.add_option('-l','--listserver',
  78                      action='store_true',dest='listserver',
  79                      help="Only list the server in configfile")
  80    parser.add_option('', '--maxmaillen',
  81                       type='float', action='store', metavar='MAXMAILLEN',
  82                       help='The value of max length of each mail (M)')
  83
  84    opts, args = parser.parse_args()
  85
  86    return opts,args
  87
  88 def ReadAlias(filename):
  89    """ read alias from filename """
  90    aliasdict={}
  91    afile=open(filename,'r')
  92    s=afile.readline().strip()
  93    while s:
  94       name,email=s.split(':')
  95       aliasdict[name]=email.split(',')
  96       s=afile.readline().strip()
  97    return aliasdict
  98
  99 def MailAddressFromAlias(alist,adict):
 100    """ realias the address alist from adict,
 101  alist: list with alias
 102  adict: dictionary of alias """
 103    elist=[]
 104    for i in alist:
 105       if adict.has_key(i):
 106          elist.extend(adict[i])
 107       else:
 108          elist.append(i)
 109    return elist
 110
 111 def NextServ(cur,lists):
 112    """ get the Next of cur in lists """
 113    keylen=len(lists)
 114    for i in range(0,keylen):
 115       if lists[i]==cur:
 116          break
 117    if i==keylen-1:
 118       i=0
 119    else:
 120       i=i+1
 121    return lists[i]
 122
 123 opts,args=GetOption()
 124
 125 if args :
 126    direction=args[0]
 127 else:
 128    direction='.'
 129
 130 if opts.configfile:
 131    rcfile=opts.configfile
 132 else:
 133    if sys.platform=='win32':
 134       rcfile=os.getcwd()+'\\xmail.ini'
 135       aliasfile=os.getcwd()+'\\xmail.alias'
 136       charset='gb2312'
 137    else:
 138       rcfile=os.environ['HOME']+'/.xmailrc'
 139       aliasfile=os.environ['HOME']+'/.xmailalias'
 140       charset='utf-8'
 141 #print rcfile
 142 param=ReadConfig(rcfile)
 143 #print param
 144 glparam=param[0]
 145 # server dict
 146 serdict=param[1]
 147 # server name list
 148 snamelist=param[2]
 149 maxmaillen=glparam[0] # max mail length
 150 maxsmtpser=glparam[1] # max smtp server
 151 defaultsmtp=glparam[2] # default smtp server
 152 onlydefault=glparam[3] # only use default smtp server?
 153 tolist=glparam[4] # to list
 154
 155 mailalias=ReadAlias(aliasfile)
 156
 157 if opts.maxmaillen:
 158    maxmaillen=int(opts.maxmaillen*1024*1024)
 159 print 'Length for each mail is less then ',maxmaillen
 160 if opts.listserver:
 161    sender=smtpser.SMTPServer()
 162    for i in serdict:
 163       sender.SetInfoFromList(serdict[i])
 164       print i,sender
 165    exit()
 166 if opts.recipients: # 如果命令行指定了dest e-mail address
 167    tolist=opts.recipients # send to lists
 168 if not tolist:
 169    print 'use -r to add recipients'
 170    quit()
 171 tolist=MailAddressFromAlias(tolist,mailalias)
 172 if opts.bccs: # add bcc
 173    bcclist=opts.bccs
 174    bcclist=MailAddressFromAlias(bcclist,mailalias)
 175 else:
 176    bcclist=None
 177
 178 if opts.subject:
 179    subj=str(email.Header.Header(opts.subject,charset))
 180 else:
 181    subj='X-Mailer: send directory'
 182
 183 sender=smtpser.SMTPServer()
 184 msender=smtpser.MailSenders(serdict,defaultsmtp)
 185 msender.OnlyDefault(onlydefault)
 186 msender.SetSubject(subj,charset)
 187 msender.SetCharset(charset)
 188
 189 print '-'*60
 190 print 'X-Mailer , mail total directory to dest e-mail address'
 191 #print 'Directory:'+direction
 192 print 'To:'+' '.join(tolist)
 193 if bcclist:
 194    print 'BCC:'+' '.join(bcclist)
 195 print 'use %d SMTP server(s):'%len(serdict),serdict.keys()
 196 print '='*60
 197
 198 files=[]
 199 fileslen=0
 200 cursmtp=defaultsmtp # current used smtp server
 201
 202 # get file list
 203 if opts.files:
 204    fileslist=[]
 205    for fl in opts.files:
 206       a=glob.glob(fl)
 207       fileslist=fileslist+a
 208 else:
 209    fileslist=glob.glob(direction+'/*') # files in directory
 210 fileslist.sort()  # sort it
 211 print 'We will mail %d file(s):'%len(fileslist)+' '.join(fileslist)
 212
 213 # the sender smtp server
 214 sender.SetInfoFromList(serdict[cursmtp])
 215
 216 for filename in fileslist:
 217    path=filename #os.path.join(direction,filename)
 218    if not os.path.isfile(path):
 219       continue
 220    filelen=os.stat(path).st_size
 221    if filelen > maxmaillen : # one big file
 222       print 'Big file:'+filename
 223 #      print 'From:'+sender.email
 224 #      print 'Subject:'+subj
 225       bodymsg=msender.CreateBodyMail([filename,'<a href=mailto:r01ustc+xmailer@gmail.com>Contact me</a>'],charset)
 226       msender.MailBigFile(tolist,subj,bodymsg,path,maxmaillen,charset=charset,bcc=bcclist)
 227 #      sender.MailBigFile(tolist,subj,[filename,'<a href=mailto:r01ustc+xmailer@gmail.com>联系我</a>'],path,maxmaillen,str(email.Header.Header(os.path.basename(filename),charset)))
 228 #      if not onlydefault:
 229 #         cursmtp=NextServ(cursmtp,serdict.keys())
 230 #         sender.SetInfoFromList(serdict[cursmtp])
 231    else:
 232       if fileslen+filelen > maxmaillen : # 加上新文件后就超大了
 233          print 'Mail Files:'+' '.join(files)
 234 #         print 'From:'+sender.email
 235 #         print 'Subject:'+subj
 236          bodymsg=msender.CreateBodyMail(['\n'.join(files),'<a href=mailto:r01ustc+xmailer@gmail.com>Contact me</a>'],charset)
 237          msender.MailFiles(tolist,subj,bodymsg,files,charset=charset,bcc=bcclist)
 238 #         sender.MailFiles(tolist,subj,['\n'.join(files),'<a href=mailto:r01ustc+xmailer@gmail.com>联系我</a>'],files,charset)
 239 #         if not onlydefault:
 240 #            cursmtp=NextServ(cursmtp,serdict.keys())
 241 #            sender.SetInfoFromList(serdict[cursmtp])
 242          files=[path]
 243          fileslen=filelen
 244       else: # add new file name to list
 245          files.append(path)
 246          fileslen=fileslen+filelen
 247 if files: # has some file not mailed
 248    print 'Mail Files remained :'+' '.join(files)
 249 #   print 'From:'+sender.email
 250 #   print 'Subject:'+subj
 251    bodymsg=msender.CreateBodyMail(['\n'.join(files),'<a href=mailto:r01ustc+xmailer@gmail.com>Contact me</a>'],charset)
 252    msender.MailFiles(tolist,subj,bodymsg,files,charset=charset,bcc=bcclist)
 253 #   sender.MailFiles(tolist,subj,['\n'.join(files),'<a href=mailto:r01ustc+xmailer@gmail.com>联系我</a>'],files,charset)
 254