mail-merge

   1 #!/usr/bin/python
   2
   3 import csv
   4 import email.parser
   5 from optparse import OptionParser
   6 import os
   7 import smtplib
   8 import subprocess
   9 import sys
  10
  11
  12 sender_header = 'mail-merge-sender@mit.edu'
  13 smtp = None
  14
  15 def dictize_line(header, line,):
  16     line_dict = {}
  17     for key, elem in zip(header, line, ):
  18         line_dict[key]=elem
  19     return line_dict
  20
  21 def setup_sendmail_smtp():
  22     global smtp
  23     smtp = smtplib.SMTP()
  24     smtp.connect()
  25 def sendmail_smtp(addrs, text):
  26     global smtp
  27     smtp.sendmail(sender_header, addrs, text, )
  28 smtp_funcs = (setup_sendmail_smtp, sendmail_smtp, )
  29
  30 def sendmail_cmd(addrs, text):
  31     args = ["/usr/lib/sendmail", "--", ]
  32     args.extend(addrs)
  33     proc = subprocess.Popen(args, stdin=subprocess.PIPE)
  34     proc.communicate(text)
  35 cmd_funcs = (lambda: True, sendmail_cmd)
  36
  37 setup_sendmail, sendmail = smtp_funcs
  38 setup_sendmail, sendmail = cmd_funcs
  39
  40 def parse_arguments():
  41     parser = OptionParser(usage='usage: %prog [options] cc_addr template recipients')
  42     parser.add_option('-q', '--rt-queue', dest='rt_queue',
  43             help='Automatically create a ticket in queue QUEUE',
  44             metavar='QUEUE',
  45     )
  46     parser.add_option('-o', '--rt-owner', dest='rt_owner',
  47             help='Set RT owner and AdminCC to USER',
  48             metavar='USER',
  49     )
  50     (options, args) = parser.parse_args()
  51     if len(args) != 3:
  52         parser.error("incorrect number of arguments")
  53     if options.rt_owner and not options.rt_queue:
  54         parser.error("--rt-owner requires specifying a queue")
  55     return options, args
  56
  57 def msg_filter_factory(opts):
  58     if not opts.rt_queue:
  59         return lambda rcpt, body: body
  60
  61     import rtkit.tracker, rtkit.authenticators, rtkit.errors
  62     cookie = rtkit.authenticators.CookieAuthenticator
  63     resource = rtkit.resource.RTResource.from_rtrc(cookie)
  64     parser = email.parser.Parser()
  65
  66     def filter_rt(rcpt, body, ):
  67         msg = parser.parsestr(body)
  68         content = {
  69             'content': {
  70                 'Requestors': rcpt,
  71                 'Queue': opts.rt_queue,
  72                 'Subject' : msg['Subject'],
  73                 'Text' : '',
  74             }
  75         }
  76         if opts.rt_owner:
  77             content['content']['AdminCC'] = opts.rt_owner
  78             content['content']['Owner'] = opts.rt_owner
  79
  80         try:
  81             response = resource.post(path='ticket/new', payload=content,)
  82             results = dict(response.parsed[0])
  83             ticket, ticket_number = results['id'].split('/')
  84             assert ticket == 'ticket', 'unexpected value "%s" instead of ticket' % (ticket, )
  85             subject = "%s [help.mit.edu #%s]" % (msg['Subject'], ticket_number)
  86             del msg['Subject']
  87             msg['Subject'] = subject
  88         except rtkit.errors.RTResourceError as e:
  89             logger.error(e.response.status_int)
  90             logger.error(e.response.status)
  91             logger.error(e.response.parsed)
  92
  93         return msg.as_string()
  94
  95     return filter_rt
  96
  97 def mail_merge(opts, cc_addr, email_file, recipients_file):
  98     email = open(email_file, 'r').read()
  99     reader = csv.reader(open(recipients_file, 'r'))
 100     header = reader.next()
 101     msg_filter = msg_filter_factory(opts)
 102     print header
 103     for line in reader:
 104         dct = dictize_line(header, line, )
 105         print dct
 106         text = email % dct
 107         text = msg_filter(dct['email'], text, )
 108         print text
 109         sendmail([dct['email'], cc_addr, ], text, )
 110
 111 if __name__=='__main__':
 112     options, args = parse_arguments()
 113     setup_sendmail()
 114     mail_merge(options, *args)