5 from optparse import OptionParser
12 sender_header = 'mail-merge-sender@mit.edu'
13 rtkit_path = '/afs/athena.mit.edu/user/a/d/adehnert/arch/common/lib/python/'
16 def dictize_line(header, line,):
18 for key, elem in zip(header, line, ):
22 def setup_sendmail_smtp():
26 def sendmail_smtp(addrs, text):
28 smtp.sendmail(sender_header, addrs, text, )
29 smtp_funcs = (setup_sendmail_smtp, sendmail_smtp, )
31 def sendmail_cmd(addrs, text):
32 args = ["/usr/lib/sendmail", "--", ]
34 proc = subprocess.Popen(args, stdin=subprocess.PIPE)
35 proc.communicate(text)
36 cmd_funcs = (lambda: True, sendmail_cmd)
38 setup_sendmail, sendmail = smtp_funcs
39 setup_sendmail, sendmail = cmd_funcs
41 def parse_arguments():
42 parser = OptionParser(usage='usage: %prog [options] cc_addr template recipients')
43 parser.add_option('-q', '--rt-queue', dest='rt_queue',
44 help='Automatically create a ticket in queue QUEUE',
47 parser.add_option('-o', '--rt-owner', dest='rt_owner',
48 help='Set RT owner and AdminCC to USER',
51 parser.add_option('--split', dest='split',
52 help='Split "email" field on SPLIT and send to each recipient',
55 (options, args) = parser.parse_args()
57 parser.error("incorrect number of arguments")
58 if options.rt_owner and not options.rt_queue:
59 parser.error("--rt-owner requires specifying a queue")
62 def nop_msg_filter(rcpts, body):
65 def msg_filter_factory(opts):
70 import rtkit.tracker, rtkit.authenticators, rtkit.errors
72 print "Note: using rtkit from %s" % (rtkit_path, )
73 sys.path.append(rtkit_path)
74 import rtkit.tracker, rtkit.authenticators, rtkit.errors
76 cookie = rtkit.authenticators.CookieAuthenticator
77 resource = rtkit.resource.RTResource.from_rtrc(cookie)
78 parser = email.parser.Parser()
80 def filter_rt(rcpts, body, ):
81 msg = parser.parsestr(body)
84 'Requestors': ", ".join(rcpts),
85 'Queue': opts.rt_queue,
86 'Subject' : msg['Subject'],
91 content['content']['AdminCC'] = opts.rt_owner
92 content['content']['Owner'] = opts.rt_owner
95 response = resource.post(path='ticket/new', payload=content,)
96 results = dict(response.parsed[0])
97 ticket, ticket_number = results['id'].split('/')
98 assert ticket == 'ticket', 'unexpected value "%s" instead of ticket' % (ticket, )
99 subject = "%s [help.mit.edu #%s]" % (msg['Subject'], ticket_number)
101 msg['Subject'] = subject
102 except rtkit.errors.RTResourceError as e:
103 logger.error(e.response.status_int)
104 logger.error(e.response.status)
105 logger.error(e.response.parsed)
107 # We don't want to send mail to the real recipient, because RT
108 # will send them a copy too.
109 return [], msg.as_string()
113 def mail_merge(opts, cc_addr, email_file, recipients_file):
114 email_tmpl = open(email_file, 'r').read()
115 reader = csv.reader(open(recipients_file, 'r'))
116 header = reader.next()
117 msg_filter = msg_filter_factory(opts)
120 dct = dictize_line(header, line, )
122 text = email_tmpl % dct
124 prop_rcpts = dct['email'].split(opts.split)
126 prop_rcpts = [dct['email']]
127 rcpts, text = msg_filter(prop_rcpts, text, )
128 rcpts.append(cc_addr)
129 sendmail(rcpts, text, )
131 if __name__=='__main__':
132 options, args = parse_arguments()
134 mail_merge(options, *args)