Context Navigation

email2trac.py.in @ 152

Last change on this file since 152 was 152, checked in by bas, 17 years ago

email2trac.py.in:

Some bugfixes for attachments that are too big

Property svn:executable set to *
Property svn:keywords set to Id

File size: 23.9 KB

Line
1	#!@PYTHON@
2	# Copyright (C) 2002
3	#
4	# This file is part of the email2trac utils
5	#
6	# This program is free software; you can redistribute it and/or modify it
7	# under the terms of the GNU General Public License as published by the
8	# Free Software Foundation; either version 2, or (at your option) any
9	# later version.
10	#
11	# This program is distributed in the hope that it will be useful,
12	# but WITHOUT ANY WARRANTY; without even the implied warranty of
13	# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14	# GNU General Public License for more details.
15	#
16	# You should have received a copy of the GNU General Public License
17	# along with this program; if not, write to the Free Software
18	# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
19	#
20	# For vi/emacs or other use tabstop=4 (vi: set ts=4)
21	#
22	"""
23	email2trac.py -- Email tickets to Trac.
24
25	A simple MTA filter to create Trac tickets from inbound emails.
26
27	Copyright 2005, Daniel Lundin <daniel@edgewall.com>
28	Copyright 2005, Edgewall Software
29
30	Changed By: Bas van der Vlies <basv@sara.nl>
31	Date : 13 September 2005
32	Descr. : Added config file and command line options, spam level
33	detection, reply address and mailto option. Unicode support
34
35	Changed By: Walter de Jong <walter@sara.nl>
36	Descr. : multipart-message code and trac attachments
37
38
39	The scripts reads emails from stdin and inserts directly into a Trac database.
40	MIME headers are mapped as follows:
41
42	* From: => Reporter
43	=> CC (Optional via reply_all option)
44	* Subject: => Summary
45	* Body => Description
46	* Component => Can be set to SPAM via spam_level option
47
48	How to use
49	----------
50	* Create an config file:
51	[DEFAULT] # REQUIRED
52	project : /data/trac/test # REQUIRED
53	debug : 1 # OPTIONAL, if set print some DEBUG info
54	spam_level : 4 # OPTIONAL, if set check for SPAM mail
55	reply_all : 1 # OPTIONAL, if set then fill in ticket CC field
56	umask : 022 # OPTIONAL, if set then use this umask for creation of the attachments
57	mailto_link : 1 # OPTIONAL, if set then [mailto:<>] in description
58	mailto_cc : basv@sara.nl # OPTIONAL, use this address as CC in mailto line
59	ticket_update: 1 # OPTIONAL, if set then check if this is an update for a ticket
60	trac_version : 0.8 # OPTIONAL, default is 0.10
61
62	[jouvin] # OPTIONAL project declaration, if set both fields necessary
63	project : /data/trac/jouvin # use -p\|--project jouvin.
64
65	* default config file is : /etc/email2trac.conf
66
67	* Commandline opions:
68	-h \| --help
69	-c <value> \| --component=<value>
70	-f <config file> \| --file=<config file>
71	-p <project name> \| --project=<project name>
72
73	SVN Info:
74	$Id: email2trac.py.in 152 2007-04-24 05:52:46Z bas $
75	"""
76	import os
77	import sys
78	import string
79	import getopt
80	import stat
81	import time
82	import email
83	import email.Iterators
84	import email.Header
85	import re
86	import urllib
87	import unicodedata
88	import ConfigParser
89	from stat import *
90	import mimetypes
91	import syslog
92	import traceback
93
94
95	# Some global variables
96	#
97	trac_default_version = 0.10
98	m = None
99
100
101	class TicketEmailParser(object):
102	env = None
103	comment = '> '
104
105	def __init__(self, env, parameters, version):
106	self.env = env
107
108	# Database connection
109	#
110	self.db = None
111
112	# Some useful mail constants
113	#
114	self.author = None
115	self.email_addr = None
116	self.email_field = None
117
118	self.VERSION = version
119	if self.VERSION == 0.8:
120	self.get_config = self.env.get_config
121	else:
122	self.get_config = self.env.config.get
123
124	if parameters.has_key('umask'):
125	os.umask(int(parameters['umask'], 8))
126
127	if parameters.has_key('debug'):
128	self.DEBUG = int(parameters['debug'])
129	else:
130	self.DEBUG = 0
131
132	if parameters.has_key('mailto_link'):
133	self.MAILTO = int(parameters['mailto_link'])
134	if parameters.has_key('mailto_cc'):
135	self.MAILTO_CC = parameters['mailto_cc']
136	else:
137	self.MAILTO_CC = ''
138	else:
139	self.MAILTO = 0
140
141	if parameters.has_key('spam_level'):
142	self.SPAM_LEVEL = int(parameters['spam_level'])
143	else:
144	self.SPAM_LEVEL = 0
145
146	if parameters.has_key('email_comment'):
147	self.comment = str(parameters['email_comment'])
148
149	if parameters.has_key('email_header'):
150	self.EMAIL_HEADER = int(parameters['email_header'])
151	else:
152	self.EMAIL_HEADER = 0
153
154	if parameters.has_key('alternate_notify_template'):
155	self.notify_template = str(parameters['alternate_notify_template'])
156	else:
157	self.notify_template = None
158
159	if parameters.has_key('reply_all'):
160	self.REPLY_ALL = int(parameters['reply_all'])
161	else:
162	self.REPLY_ALL = 0
163
164	if parameters.has_key('ticket_update'):
165	self.TICKET_UPDATE = int(parameters['ticket_update'])
166	else:
167	self.TICKET_UPDATE = 0
168
169	if parameters.has_key('drop_spam'):
170	self.DROP_SPAM = int(parameters['drop_spam'])
171	else:
172	self.DROP_SPAM = 0
173
174	if parameters.has_key('verbatim_format'):
175	self.VERBATIM_FORMAT = int(parameters['verbatim_format'])
176	else:
177	self.VERBATIM_FORMAT = 1
178
179	if parameters.has_key('strip_signature'):
180	self.STRIP_SIGNATURE = int(parameters['strip_signature'])
181	else:
182	self.STRIP_SIGNATURE = 0
183
184	if parameters.has_key('use_textwrap'):
185	self.USE_TEXTWRAP = int(parameters['use_textwrap'])
186	else:
187	self.USE_TEXTWRAP = 0
188
189	# X-Spam-Score: *** (3.255) BAYES_50,DNS_FROM_AHBL_RHSBL,HTML_
190	# Note if Spam_level then '*' are included
191	def spam(self, message):
192	if message.has_key('X-Spam-Score'):
193	spam_l = string.split(message['X-Spam-Score'])
194	number = spam_l[0].count('*')
195
196	if number >= self.SPAM_LEVEL:
197	return 'Spam'
198
199	elif message.has_key('X-Virus-found'): # treat virus mails as spam
200	return 'Spam'
201
202	return self.get_config('ticket', 'default_component')
203
204	def blacklisted_from(self):
205	FROM_RE = re.compile(r"""
206	MAILER-DAEMON@
207	""", re.VERBOSE)
208	result = FROM_RE.search(self.email_addr)
209	if result:
210	return True
211	else:
212	return False
213
214	def email_to_unicode(self, message_str):
215	"""
216	Email has 7 bit ASCII code, convert it to unicode with the charset
217	that is encoded in 7-bit ASCII code and encode it as utf-8 so Trac
218	understands it.
219	"""
220	results = email.Header.decode_header(message_str)
221	str = None
222	for text,format in results:
223	if format:
224	try:
225	temp = unicode(text, format)
226	except UnicodeError, detail:
227	# This always works
228	#
229	temp = unicode(text, 'iso-8859-15')
230	except LookupError, detail:
231	#text = 'ERROR: Could not find charset: %s, please install' %format
232	#temp = unicode(text, 'iso-8859-15')
233	temp = message_str
234
235	else:
236	temp = string.strip(text)
237	temp = unicode(text, 'iso-8859-15')
238
239	if str:
240	str = '%s %s' %(str, temp)
241	else:
242	str = '%s' %temp
243
244	#str = str.encode('utf-8')
245	return str
246
247	def debug_attachments(self, message):
248	n = 0
249	for part in message.walk():
250	if part.get_content_maintype() == 'multipart': # multipart/* is just a container
251	print 'TD: multipart container'
252	continue
253
254	n = n + 1
255	print 'TD: part%d: Content-Type: %s' % (n, part.get_content_type())
256	print 'TD: part%d: filename: %s' % (n, part.get_filename())
257
258	if part.is_multipart():
259	print 'TD: this part is multipart'
260	payload = part.get_payload(decode=1)
261	print 'TD: payload:', payload
262	else:
263	print 'TD: this part is not multipart'
264
265	part_file = '/var/tmp/part%d' % n
266	print 'TD: writing part%d (%s)' % (n,part_file)
267	fx = open(part_file, 'wb')
268	text = part.get_payload(decode=1)
269	if not text:
270	text = '(None)'
271	fx.write(text)
272	fx.close()
273	try:
274	os.chmod(part_file,S_IRWXU\|S_IRWXG\|S_IRWXO)
275	except OSError:
276	pass
277
278	def email_header_txt(self, m):
279	"""
280	Display To and CC addresses in description field
281	"""
282	str = ''
283	if m['To'] and len(m['To']) > 0 and m['To'] != 'hic@sara.nl':
284	str = "'''To:''' %s [[BR]]" %(m['To'])
285	if m['Cc'] and len(m['Cc']) > 0:
286	str = "%s'''Cc:''' %s [[BR]]" % (str, m['Cc'])
287
288	return self.email_to_unicode(str)
289
290
291	def set_owner(self, ticket):
292	"""
293	Select default owner for ticket component
294	"""
295	cursor = self.db.cursor()
296	sql = "SELECT owner FROM component WHERE name='%s'" % ticket['component']
297	cursor.execute(sql)
298	try:
299	ticket['owner'] = cursor.fetchone()[0]
300	except TypeError, detail:
301	ticket['owner'] = "UNKNOWN"
302
303	def get_author_emailaddrs(self, message):
304	"""
305	Get the default author name and email address from the message
306	"""
307	temp = self.email_to_unicode(message['from'])
308	#print temp.encode('utf-8')
309
310	self.author, self.email_addr = email.Utils.parseaddr(temp)
311	#print self.author.encode('utf-8', 'replace')
312
313	# Look for email address in registered trac users
314	#
315	if self.VERSION == 0.8:
316	users = []
317	else:
318	users = [ u for (u, n, e) in self.env.get_known_users(self.db)
319	if e == self.email_addr ]
320
321	if len(users) == 1:
322	self.email_field = users[0]
323	else:
324	self.email_field = self.email_to_unicode(message['from'])
325
326	def set_reply_fields(self, ticket, message):
327	"""
328	Set all the right fields for a new ticket
329	"""
330	ticket['reporter'] = self.email_field
331
332	# Put all CC-addresses in ticket CC field
333	#
334	if self.REPLY_ALL:
335	#tos = message.get_all('to', [])
336	ccs = message.get_all('cc', [])
337
338	addrs = email.Utils.getaddresses(ccs)
339	if not addrs:
340	return
341
342	# Remove reporter email address if notification is
343	# on
344	#
345	if self.notification:
346	try:
347	addrs.remove((self.author, self.email_addr))
348	except ValueError, detail:
349	pass
350
351	for name,mail in addrs:
352	try:
353	mail_list = '%s, %s' %(mail_list, mail)
354	except:
355	mail_list = mail
356
357	if mail_list:
358	ticket['cc'] = self.email_to_unicode(mail_list)
359
360	def save_email_for_debug(self, message, tempfile=False):
361	if tempfile:
362	import tempfile
363	msg_file = tempfile.mktemp('.email2trac')
364	else:
365	msg_file = '/var/tmp/msg.txt'
366	print 'TD: saving email to %s' % msg_file
367	fx = open(msg_file, 'wb')
368	fx.write('%s' % message)
369	fx.close()
370	try:
371	os.chmod(msg_file,S_IRWXU\|S_IRWXG\|S_IRWXO)
372	except OSError:
373	pass
374
375	def ticket_update(self, m):
376	"""
377	If the current email is a reply to an existing ticket, this function
378	will append the contents of this email to that ticket, instead of
379	creating a new one.
380	"""
381	if not m['Subject']:
382	return False
383	else:
384	subject = self.email_to_unicode(m['Subject'])
385
386	TICKET_RE = re.compile(r"""
387	(?P<ticketnr>[#][0-9]+:)
388	""", re.VERBOSE)
389
390	result = TICKET_RE.search(subject)
391	if not result:
392	return False
393
394	body_text = self.get_body_text(m)
395
396	# Strip '#' and ':' from ticket_id
397	#
398	ticket_id = result.group('ticketnr')
399	ticket_id = int(ticket_id[1:-1])
400
401	# Get current time
402	#
403	when = int(time.time())
404
405	if self.VERSION == 0.8:
406	tkt = Ticket(self.db, ticket_id)
407	tkt.save_changes(self.db, self.author, body_text, when)
408	else:
409	try:
410	tkt = Ticket(self.env, ticket_id, self.db)
411	except util.TracError, detail:
412	return False
413
414	tkt.save_changes(self.author, body_text, when)
415	tkt['id'] = ticket_id
416
417	if self.VERSION == 0.9:
418	str = self.attachments(m, tkt, True)
419	else:
420	str = self.attachments(m, tkt)
421
422	if self.notification:
423	self.notify(tkt, False, when)
424
425	return True
426
427	def new_ticket(self, msg):
428	"""
429	Create a new ticket
430	"""
431	tkt = Ticket(self.env)
432	tkt['status'] = 'new'
433
434	# Some defaults
435	#
436	tkt['milestone'] = self.get_config('ticket', 'default_milestone')
437	tkt['priority'] = self.get_config('ticket', 'default_priority')
438	tkt['severity'] = self.get_config('ticket', 'default_severity')
439	tkt['version'] = self.get_config('ticket', 'default_version')
440
441	if not msg['Subject']:
442	tkt['summary'] = u'(No subject)'
443	else:
444	tkt['summary'] = self.email_to_unicode(msg['Subject'])
445
446
447	if settings.has_key('component'):
448	tkt['component'] = settings['component']
449	else:
450	tkt['component'] = self.spam(msg)
451
452	# Discard SPAM messages.
453	#
454	if self.DROP_SPAM and (tkt['component'] == 'Spam'):
455	if self.DEBUG > 2 :
456	print 'This message is a SPAM. Automatic ticket insertion refused (SPAM level > %d' % self.SPAM_LEVEL
457	return False
458
459	# Set default owner for component
460	#
461	self.set_owner(tkt)
462	self.set_reply_fields(tkt, msg)
463
464	# produce e-mail like header
465	#
466	head = ''
467	if self.EMAIL_HEADER > 0:
468	head = self.email_header_txt(msg)
469
470	body_text = self.get_body_text(msg)
471
472	tkt['description'] = '\r\n%s\r\n%s' \
473	%(head, body_text)
474
475	when = int(time.time())
476	if self.VERSION == 0.8:
477	ticket_id = tkt.insert(self.db)
478	else:
479	ticket_id = tkt.insert()
480	tkt['id'] = ticket_id
481
482	changed = False
483	comment = ''
484
485	# Rewrite the description if we have mailto enabled
486	#
487	if self.MAILTO:
488	changed = True
489	comment = u'\nadded mailto line\n'
490	mailto = self.html_mailto_link(tkt['summary'], ticket_id, body_text)
491	tkt['description'] = u'\r\n%s\r\n%s%s\r\n' \
492	%(head, mailto, body_text)
493
494	str = self.attachments(msg, tkt)
495	if str:
496	changed = True
497	comment = '%s\n%s\n' %(comment, str)
498
499	if changed:
500	if self.VERSION == 0.8:
501	tkt.save_changes(self.db, self.author, comment)
502	else:
503	tkt.save_changes(self.author, comment)
504
505	#print tkt.get_changelog(self.db, when)
506
507	if self.notification:
508	self.notify(tkt, True)
509	#self.notify(tkt, False)
510
511	def parse(self, fp):
512	global m
513
514	m = email.message_from_file(fp)
515	if not m:
516	return
517
518	if self.DEBUG > 1: # save the entire e-mail message text
519	self.save_email_for_debug(m)
520	self.debug_attachments(m)
521
522	self.db = self.env.get_db_cnx()
523	self.get_author_emailaddrs(m)
524
525	if self.blacklisted_from():
526	if self.DEBUG > 1 :
527	print 'Message rejected : From: in blacklist'
528	return False
529
530	if self.get_config('notification', 'smtp_enabled') in ['true']:
531	self.notification = 1
532	else:
533	self.notification = 0
534
535	# Must we update existing tickets
536	#
537	if self.TICKET_UPDATE > 0:
538	if self.ticket_update(m):
539	return True
540
541	self.new_ticket(m)
542
543	def strip_signature(self, text):
544	"""
545	Strip signature from message, inspired by Mailman software
546	"""
547	body = []
548	for line in text.splitlines():
549	if line == '-- ':
550	break
551	body.append(line)
552
553	return ('\n'.join(body))
554
555
556	def wrap_text(str='', width = self.USE_TEXTWRAP, replace_whitespace = False):
557	"""
558	Will break a lines longer then given length into several small lines of size
559	given length
560	"""
561	import textwrap
562	LINESEPARATOR = '\n'
563	return LINESEPARATOR.join(textwrap.fill(s,width) for s in str.split(LINESEPARATOR))
564
565	def get_body_text(self, msg):
566	"""
567	put the message text in the ticket description or in the changes field.
568	message text can be plain text or html or something else
569	"""
570	has_description = 0
571	encoding = True
572	ubody_text = u'No plain text message'
573	for part in msg.walk():
574
575	# 'multipart/*' is a container for multipart messages
576	#
577	if part.get_content_maintype() == 'multipart':
578	continue
579
580	if part.get_content_type() == 'text/plain':
581	# Try to decode, if fails then do not decode
582	#
583	body_text = part.get_payload(decode=1)
584	if not body_text:
585	body_text = part.get_payload(decode=0)
586
587	if self.STRIP_SIGNATURE:
588	body_text = self.strip_signature(body_text)
589
590	if self.USE_TEXTWRAP:
591	body_text = self.wrap_text(body_text)
592
593	# Get contents charset (iso-8859-15 if not defined in mail headers)
594	#
595	charset = part.get_content_charset()
596	if not charset:
597	charset = 'iso-8859-15'
598
599	try:
600	ubody_text = unicode(body_text, charset)
601
602	except UnicodeError, detail:
603	ubody_text = unicode(body_text, 'iso-8859-15')
604
605	except LookupError, detail:
606	ubody_text = 'ERROR: Could not find charset: %s, please install' %(charset)
607
608	elif part.get_content_type() == 'text/html':
609	ubody_text = '(see attachment for HTML mail message)'
610
611	else:
612	ubody_text = '(see attachment for message)'
613
614	has_description = 1
615	break # we have the description, so break
616
617	if not has_description:
618	ubody_text = '(see attachment for message)'
619
620	# A patch so that the web-interface will not update the description
621	# field of a ticket
622	#
623	ubody_text = ('\r\n'.join(ubody_text.splitlines()))
624
625	# If we can unicode it try to encode it for trac
626	# else we a lot of garbage
627	#
628	#if encoding:
629	# ubody_text = ubody_text.encode('utf-8')
630
631	if self.VERBATIM_FORMAT:
632	ubody_text = '{{{\r\n%s\r\n}}}' %ubody_text
633	else:
634	ubody_text = '%s' %ubody_text
635
636	return ubody_text
637
638	def notify(self, tkt , new=True, modtime=0):
639	"""
640	A wrapper for the TRAC notify function. So we can use templates
641	"""
642	if tkt['component'] == 'Spam':
643	return
644
645	try:
646	# create false {abs_}href properties, to trick Notify()
647	#
648	self.env.abs_href = Href(self.get_config('project', 'url'))
649	self.env.href = Href(self.get_config('project', 'url'))
650
651	tn = TicketNotifyEmail(self.env)
652	if self.notify_template:
653	tn.template_name = self.notify_template;
654
655	tn.notify(tkt, new, modtime)
656
657	except Exception, e:
658	print 'TD: Failure sending notification on creation of ticket #%s: %s' %(tkt['id'], e)
659
660	def mail_line(self, str):
661	return '%s %s' % (self.comment, str)
662
663
664	def html_mailto_link(self, subject, id, body):
665	if not self.author:
666	author = self.email_addr
667	else:
668	author = self.author
669
670	# Must find a fix
671	#
672	#arr = string.split(body, '\n')
673	#arr = map(self.mail_line, arr)
674	#body = string.join(arr, '\n')
675	#body = '%s wrote:\n%s' %(author, body)
676
677	# Temporary fix
678	#
679	str = 'mailto:%s?Subject=%s&Cc=%s' %(
680	urllib.quote(self.email_addr),
681	urllib.quote('Re: #%s: %s' %(id, subject)),
682	urllib.quote(self.MAILTO_CC)
683	)
684
685	str = '\r\n{{{\r\n#!html\r\n<a href="%s">Reply to: %s</a>\r\n}}}\r\n' %(str, author)
686	return str
687
688	def attachments(self, message, ticket, update=False):
689	'''
690	save any attachments as files in the ticket's directory
691	'''
692	count = 0
693	first = 0
694	number = 0
695
696	# Get Maxium attachment size
697	#
698	max_size = int(self.get_config('attachment', 'max_size'))
699	message = ''
700
701	for part in message.walk():
702	if part.get_content_maintype() == 'multipart': # multipart/* is just a container
703	continue
704
705	if not first: # first content is the message
706	first = 1
707	if part.get_content_type() == 'text/plain': # if first is text, is was already put in the description
708	continue
709
710	filename = part.get_filename()
711	if not filename:
712	number = number + 1
713	filename = 'part%04d' % number
714
715	ext = mimetypes.guess_extension(part.get_content_type())
716	if not ext:
717	ext = '.bin'
718
719	filename = '%s%s' % (filename, ext)
720	else:
721	filename = self.email_to_unicode(filename)
722
723	# From the trac code
724	#
725	filename = filename.replace('\\', '/').replace(':', '/')
726	filename = os.path.basename(filename)
727
728	# We try to normalize the filename to utf-8 NFC if we can.
729	# Files uploaded from OS X might be in NFD.
730	# Check python version and then try it
731	#
732	if sys.version_info[0] > 2 or (sys.version_info[0] == 2 and sys.version_info[1] >= 3):
733	try:
734	filename = unicodedata.normalize('NFC', unicode(filename, 'utf-8')).encode('utf-8')
735	except TypeError:
736	pass
737
738	url_filename = urllib.quote(filename)
739	if self.VERSION == 0.8:
740	dir = os.path.join(self.env.get_attachments_dir(), 'ticket',
741	urllib.quote(str(ticket['id'])))
742	if not os.path.exists(dir):
743	mkdir_p(dir, 0755)
744	else:
745	dir = '/tmp'
746
747	path, fd = util.create_unique_file(os.path.join(dir, url_filename))
748	text = part.get_payload(decode=1)
749	if not text:
750	text = '(None)'
751	fd.write(text)
752	fd.close()
753
754	# get the filesize
755	#
756	stats = os.lstat(path)
757	filesize = stats[stat.ST_SIZE]
758
759	# Check if the attachment size is allowed
760	#
761	if file_size > max_size:
762	message = '%s\n%sFile %s is larger then allowed attachment size (%d > %d)'
763	%(message, filename, file_size, max_size)
764
765	os.unlink(path)
766	continue
767	else:
768	count = count + 1
769
770	# Insert the attachment it differs for the different TRAC versions
771	#
772	if self.VERSION == 0.8:
773	cursor = self.db.cursor()
774	try:
775	cursor.execute('INSERT INTO attachment VALUES("%s","%s","%s",%d,%d,"%s","%s","%s")'
776	%('ticket', urllib.quote(str(ticket['id'])), filename + '?format=raw', filesize,
777	int(time.time()),'', self.author, 'e-mail') )
778
779	# Attachment is already known
780	#
781	except sqlite.IntegrityError:
782	#self.db.close()
783	return count
784
785	self.db.commit()
786
787	else:
788	fd = open(path)
789	att = attachment.Attachment(self.env, 'ticket', ticket['id'])
790
791	# This will break the ticket_update system, the body_text is vaporized
792	# ;-(
793	#
794	if not update:
795	att.author = self.author
796	att.description = self.email_to_unicode('Added by email2trac')
797
798	att.insert(url_filename, fd, filesize)
799
800	#except util.TracError, detail:
801	# print detail
802
803	fd.close()
804
805	# Remove the created temporary filename
806	#
807	os.unlink(path)
808
809	# Return how many attachments
810	#
811	message = 'This message has %d attachment(s)\n%s' %(count, message)
812	return message
813
814
815	def mkdir_p(dir, mode):
816	'''do a mkdir -p'''
817
818	arr = string.split(dir, '/')
819	path = ''
820	for part in arr:
821	path = '%s/%s' % (path, part)
822	try:
823	stats = os.stat(path)
824	except OSError:
825	os.mkdir(path, mode)
826
827
828	def ReadConfig(file, name):
829	"""
830	Parse the config file
831	"""
832
833	if not os.path.isfile(file):
834	print 'File %s does not exist' %file
835	sys.exit(1)
836
837	config = ConfigParser.ConfigParser()
838	try:
839	config.read(file)
840	except ConfigParser.MissingSectionHeaderError,detail:
841	print detail
842	sys.exit(1)
843
844
845	# Use given project name else use defaults
846	#
847	if name:
848	if not config.has_section(name):
849	print "Not a valid project name: %s" %name
850	print "Valid names: %s" %config.sections()
851	sys.exit(1)
852
853	project = dict()
854	for option in config.options(name):
855	project[option] = config.get(name, option)
856
857	else:
858	project = config.defaults()
859
860	return project
861
862
863	if __name__ == '__main__':
864	# Default config file
865	#
866	configfile = '@email2trac_conf@'
867	project = ''
868	component = ''
869	ENABLE_SYSLOG = 0
870
871	try:
872	opts, args = getopt.getopt(sys.argv[1:], 'chf:p:', ['component=','help', 'file=', 'project='])
873	except getopt.error,detail:
874	print __doc__
875	print detail
876	sys.exit(1)
877
878	project_name = None
879	for opt,value in opts:
880	if opt in [ '-h', '--help']:
881	print __doc__
882	sys.exit(0)
883	elif opt in ['-c', '--component']:
884	component = value
885	elif opt in ['-f', '--file']:
886	configfile = value
887	elif opt in ['-p', '--project']:
888	project_name = value
889
890	settings = ReadConfig(configfile, project_name)
891	if not settings.has_key('project'):
892	print __doc__
893	print 'No Trac project is defined in the email2trac config file.'
894	sys.exit(1)
895
896	if component:
897	settings['component'] = component
898
899	if settings.has_key('trac_version'):
900	version = float(settings['trac_version'])
901	else:
902	version = trac_default_version
903
904	if settings.has_key('enable_syslog'):
905	ENABLE_SYSLOG = float(settings['enable_syslog'])
906
907	#debug HvB
908	#print settings
909
910	try:
911	if version == 0.8:
912	from trac.Environment import Environment
913	from trac.Ticket import Ticket
914	from trac.Notify import TicketNotifyEmail
915	from trac.Href import Href
916	from trac import util
917	import sqlite
918	elif version == 0.9:
919	from trac import attachment
920	from trac.env import Environment
921	from trac.ticket import Ticket
922	from trac.web.href import Href
923	from trac import util
924	from trac.Notify import TicketNotifyEmail
925	elif version == 0.10:
926	from trac import attachment
927	from trac.env import Environment
928	from trac.ticket import Ticket
929	from trac.web.href import Href
930	from trac import util
931	#
932	# return util.text.to_unicode(str)
933	#
934	# see http://projects.edgewall.com/trac/changeset/2799
935	from trac.ticket.notification import TicketNotifyEmail
936
937	env = Environment(settings['project'], create=0)
938	tktparser = TicketEmailParser(env, settings, version)
939	tktparser.parse(sys.stdin)
940
941	# Catch all errors ans log to SYSLOG if we have enabled this
942	# else stdout
943	#
944	except Exception, error:
945	if ENABLE_SYSLOG:
946	syslog.openlog('email2trac', syslog.LOG_NOWAIT)
947	etype, evalue, etb = sys.exc_info()
948	for e in traceback.format_exception(etype, evalue, etb):
949	syslog.syslog(e)
950	syslog.closelog()
951	else:
952	traceback.print_exc()
953
954	if m:
955	tktparser.save_email_for_debug(m, True)
956
957	# EOB

Note: See TracBrowser for help on using the repository browser.

Download in other formats:

Original Format