Context Navigation

email2trac.py.in @ 153

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

email2trac.py.in:

Code runs again and added check for attachment size

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

File size: 24.0 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 153 2007-04-24 06:51:51Z 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='', 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	reformat = ''
564
565	for s in str.split(LINESEPARATOR):
566	reformat = '%s%s' %(reformat, LINESEPARATOR.join(textwrap.fill(s,self.USE_TEXTWRAP)))
567
568	return reformat
569
570	def get_body_text(self, msg):
571	"""
572	put the message text in the ticket description or in the changes field.
573	message text can be plain text or html or something else
574	"""
575	has_description = 0
576	encoding = True
577	ubody_text = u'No plain text message'
578	for part in msg.walk():
579
580	# 'multipart/*' is a container for multipart messages
581	#
582	if part.get_content_maintype() == 'multipart':
583	continue
584
585	if part.get_content_type() == 'text/plain':
586	# Try to decode, if fails then do not decode
587	#
588	body_text = part.get_payload(decode=1)
589	if not body_text:
590	body_text = part.get_payload(decode=0)
591
592	if self.STRIP_SIGNATURE:
593	body_text = self.strip_signature(body_text)
594
595	if self.USE_TEXTWRAP:
596	body_text = self.wrap_text(body_text)
597
598	# Get contents charset (iso-8859-15 if not defined in mail headers)
599	#
600	charset = part.get_content_charset()
601	if not charset:
602	charset = 'iso-8859-15'
603
604	try:
605	ubody_text = unicode(body_text, charset)
606
607	except UnicodeError, detail:
608	ubody_text = unicode(body_text, 'iso-8859-15')
609
610	except LookupError, detail:
611	ubody_text = 'ERROR: Could not find charset: %s, please install' %(charset)
612
613	elif part.get_content_type() == 'text/html':
614	ubody_text = '(see attachment for HTML mail message)'
615
616	else:
617	ubody_text = '(see attachment for message)'
618
619	has_description = 1
620	break # we have the description, so break
621
622	if not has_description:
623	ubody_text = '(see attachment for message)'
624
625	# A patch so that the web-interface will not update the description
626	# field of a ticket
627	#
628	ubody_text = ('\r\n'.join(ubody_text.splitlines()))
629
630	# If we can unicode it try to encode it for trac
631	# else we a lot of garbage
632	#
633	#if encoding:
634	# ubody_text = ubody_text.encode('utf-8')
635
636	if self.VERBATIM_FORMAT:
637	ubody_text = '{{{\r\n%s\r\n}}}' %ubody_text
638	else:
639	ubody_text = '%s' %ubody_text
640
641	return ubody_text
642
643	def notify(self, tkt , new=True, modtime=0):
644	"""
645	A wrapper for the TRAC notify function. So we can use templates
646	"""
647	if tkt['component'] == 'Spam':
648	return
649
650	try:
651	# create false {abs_}href properties, to trick Notify()
652	#
653	self.env.abs_href = Href(self.get_config('project', 'url'))
654	self.env.href = Href(self.get_config('project', 'url'))
655
656	tn = TicketNotifyEmail(self.env)
657	if self.notify_template:
658	tn.template_name = self.notify_template;
659
660	tn.notify(tkt, new, modtime)
661
662	except Exception, e:
663	print 'TD: Failure sending notification on creation of ticket #%s: %s' %(tkt['id'], e)
664
665	def mail_line(self, str):
666	return '%s %s' % (self.comment, str)
667
668
669	def html_mailto_link(self, subject, id, body):
670	if not self.author:
671	author = self.email_addr
672	else:
673	author = self.author
674
675	# Must find a fix
676	#
677	#arr = string.split(body, '\n')
678	#arr = map(self.mail_line, arr)
679	#body = string.join(arr, '\n')
680	#body = '%s wrote:\n%s' %(author, body)
681
682	# Temporary fix
683	#
684	str = 'mailto:%s?Subject=%s&Cc=%s' %(
685	urllib.quote(self.email_addr),
686	urllib.quote('Re: #%s: %s' %(id, subject)),
687	urllib.quote(self.MAILTO_CC)
688	)
689
690	str = '\r\n{{{\r\n#!html\r\n<a href="%s">Reply to: %s</a>\r\n}}}\r\n' %(str, author)
691	return str
692
693	def attachments(self, message, ticket, update=False):
694	'''
695	save any attachments as files in the ticket's directory
696	'''
697	count = 0
698	first = 0
699	number = 0
700
701	# Get Maxium attachment size
702	#
703	max_size = int(self.get_config('attachment', 'max_size'))
704	status = ''
705
706	for part in message.walk():
707	if part.get_content_maintype() == 'multipart': # multipart/* is just a container
708	continue
709
710	if not first: # first content is the message
711	first = 1
712	if part.get_content_type() == 'text/plain': # if first is text, is was already put in the description
713	continue
714
715	filename = part.get_filename()
716	if not filename:
717	number = number + 1
718	filename = 'part%04d' % number
719
720	ext = mimetypes.guess_extension(part.get_content_type())
721	if not ext:
722	ext = '.bin'
723
724	filename = '%s%s' % (filename, ext)
725	else:
726	filename = self.email_to_unicode(filename)
727
728	# From the trac code
729	#
730	filename = filename.replace('\\', '/').replace(':', '/')
731	filename = os.path.basename(filename)
732
733	# We try to normalize the filename to utf-8 NFC if we can.
734	# Files uploaded from OS X might be in NFD.
735	# Check python version and then try it
736	#
737	if sys.version_info[0] > 2 or (sys.version_info[0] == 2 and sys.version_info[1] >= 3):
738	try:
739	filename = unicodedata.normalize('NFC', unicode(filename, 'utf-8')).encode('utf-8')
740	except TypeError:
741	pass
742
743	url_filename = urllib.quote(filename)
744	if self.VERSION == 0.8:
745	dir = os.path.join(self.env.get_attachments_dir(), 'ticket',
746	urllib.quote(str(ticket['id'])))
747	if not os.path.exists(dir):
748	mkdir_p(dir, 0755)
749	else:
750	dir = '/tmp'
751
752	path, fd = util.create_unique_file(os.path.join(dir, url_filename))
753	text = part.get_payload(decode=1)
754	if not text:
755	text = '(None)'
756	fd.write(text)
757	fd.close()
758
759	# get the file_size
760	#
761	stats = os.lstat(path)
762	file_size = stats[stat.ST_SIZE]
763
764	# Check if the attachment size is allowed
765	#
766	if (max_size != -1) and (file_size > max_size):
767	status = '%s\nFile %s is larger then allowed attachment size (%d > %d)\n\n' \
768	%(status, filename, file_size, max_size)
769
770	os.unlink(path)
771	continue
772	else:
773	count = count + 1
774
775	# Insert the attachment it differs for the different TRAC versions
776	#
777	if self.VERSION == 0.8:
778	cursor = self.db.cursor()
779	try:
780	cursor.execute('INSERT INTO attachment VALUES("%s","%s","%s",%d,%d,"%s","%s","%s")'
781	%('ticket', urllib.quote(str(ticket['id'])), filename + '?format=raw', file_size,
782	int(time.time()),'', self.author, 'e-mail') )
783
784	# Attachment is already known
785	#
786	except sqlite.IntegrityError:
787	#self.db.close()
788	return count
789
790	self.db.commit()
791
792	else:
793	fd = open(path)
794	att = attachment.Attachment(self.env, 'ticket', ticket['id'])
795
796	# This will break the ticket_update system, the body_text is vaporized
797	# ;-(
798	#
799	if not update:
800	att.author = self.author
801	att.description = self.email_to_unicode('Added by email2trac')
802
803	att.insert(url_filename, fd, file_size)
804
805	#except util.TracError, detail:
806	# print detail
807
808	fd.close()
809
810	# Remove the created temporary filename
811	#
812	os.unlink(path)
813
814	# Return how many attachments
815	#
816	status = 'This message has %d attachment(s)\n%s' %(count, status)
817	return status
818
819
820	def mkdir_p(dir, mode):
821	'''do a mkdir -p'''
822
823	arr = string.split(dir, '/')
824	path = ''
825	for part in arr:
826	path = '%s/%s' % (path, part)
827	try:
828	stats = os.stat(path)
829	except OSError:
830	os.mkdir(path, mode)
831
832
833	def ReadConfig(file, name):
834	"""
835	Parse the config file
836	"""
837
838	if not os.path.isfile(file):
839	print 'File %s does not exist' %file
840	sys.exit(1)
841
842	config = ConfigParser.ConfigParser()
843	try:
844	config.read(file)
845	except ConfigParser.MissingSectionHeaderError,detail:
846	print detail
847	sys.exit(1)
848
849
850	# Use given project name else use defaults
851	#
852	if name:
853	if not config.has_section(name):
854	print "Not a valid project name: %s" %name
855	print "Valid names: %s" %config.sections()
856	sys.exit(1)
857
858	project = dict()
859	for option in config.options(name):
860	project[option] = config.get(name, option)
861
862	else:
863	project = config.defaults()
864
865	return project
866
867
868	if __name__ == '__main__':
869	# Default config file
870	#
871	configfile = '@email2trac_conf@'
872	project = ''
873	component = ''
874	ENABLE_SYSLOG = 0
875
876	try:
877	opts, args = getopt.getopt(sys.argv[1:], 'chf:p:', ['component=','help', 'file=', 'project='])
878	except getopt.error,detail:
879	print __doc__
880	print detail
881	sys.exit(1)
882
883	project_name = None
884	for opt,value in opts:
885	if opt in [ '-h', '--help']:
886	print __doc__
887	sys.exit(0)
888	elif opt in ['-c', '--component']:
889	component = value
890	elif opt in ['-f', '--file']:
891	configfile = value
892	elif opt in ['-p', '--project']:
893	project_name = value
894
895	settings = ReadConfig(configfile, project_name)
896	if not settings.has_key('project'):
897	print __doc__
898	print 'No Trac project is defined in the email2trac config file.'
899	sys.exit(1)
900
901	if component:
902	settings['component'] = component
903
904	if settings.has_key('trac_version'):
905	version = float(settings['trac_version'])
906	else:
907	version = trac_default_version
908
909	if settings.has_key('enable_syslog'):
910	ENABLE_SYSLOG = float(settings['enable_syslog'])
911
912	#debug HvB
913	#print settings
914
915	try:
916	if version == 0.8:
917	from trac.Environment import Environment
918	from trac.Ticket import Ticket
919	from trac.Notify import TicketNotifyEmail
920	from trac.Href import Href
921	from trac import util
922	import sqlite
923	elif version == 0.9:
924	from trac import attachment
925	from trac.env import Environment
926	from trac.ticket import Ticket
927	from trac.web.href import Href
928	from trac import util
929	from trac.Notify import TicketNotifyEmail
930	elif version == 0.10:
931	from trac import attachment
932	from trac.env import Environment
933	from trac.ticket import Ticket
934	from trac.web.href import Href
935	from trac import util
936	#
937	# return util.text.to_unicode(str)
938	#
939	# see http://projects.edgewall.com/trac/changeset/2799
940	from trac.ticket.notification import TicketNotifyEmail
941
942	env = Environment(settings['project'], create=0)
943	tktparser = TicketEmailParser(env, settings, version)
944	tktparser.parse(sys.stdin)
945
946	# Catch all errors ans log to SYSLOG if we have enabled this
947	# else stdout
948	#
949	except Exception, error:
950	if ENABLE_SYSLOG:
951	syslog.openlog('email2trac', syslog.LOG_NOWAIT)
952	etype, evalue, etb = sys.exc_info()
953	for e in traceback.format_exception(etype, evalue, etb):
954	syslog.syslog(e)
955	syslog.closelog()
956	else:
957	traceback.print_exc()
958
959	if m:
960	tktparser.save_email_for_debug(m, True)
961
962	# EOB

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

Download in other formats:

Original Format