Changeset 204 for trunk/email2trac.py.in

Timestamp:

05/29/08 15:51:21 (15 years ago)

Author:

bas

Message:

email2trac.py.in:

• close tickets #62, #55

File:

• trunk/email2trac.py.in (modified) (18 diffs)

trunk/email2trac.py.in

@@ -107 +107 @@
 trac_default_version = '0.10'
 m = None 
-DRY_RUN = False
-
-
 
 # A UTC class needed for trac version 0.11, added by
@@ -133 +130 @@
         comment = '> '
     
-        def __init__(self, env, parameters, version):
+        def __init__(self, env, parameters, version, dry_run):
                 self.env = env
 
@@ -147 +144 @@
 
                 self.VERSION = version
+                self.DRY_RUN = dry_run
+
                 self.get_config = self.env.config.get
 
@@ -260 +259 @@
                                 print 'This message is a SPAM. Automatic ticket insertion refused (SPAM level > %d' % self.SPAM_LEVEL
 
-                        return True     
+                        return 'drop'   
 
                 elif spam:
 
-                        return 'Spam'
-
-                else:
-
-                        return self.get_config('ticket', 'default_component') 
+                        return 'Spam'   
+
+                else:
+
+                        return False
 
 
@@ -530 +529 @@
                                         print  'ticket_field\t %s = %s' %(field,  ticket[field])
                                         
-        def ticket_update(self, m):
+        def ticket_update(self, m, spam):
                 """
                 If the current email is a reply to an existing ticket, this function
@@ -536 +535 @@
                 creating a new one.
                 """
-                global DRY_RUN
 
                 if not m['Subject']:
@@ -609 +607 @@
                         str = self.attachments(m, tkt)
 
-                if self.notification:
+                if self.notification and not spam:
                         self.notify(tkt, False, when)
 
@@ -654 +652 @@
 
 
-        def new_ticket(self, msg):
+        def new_ticket(self, msg, spam):
                 """
                 Create a new ticket
@@ -673 +671 @@
                 # Old style setting for component, will be removed
                 #
-                if settings.has_key('component'):
+                if spam:
+                        tkt['component'] = 'Spam'
+
+                elif settings.has_key('component'):
                         tkt['component'] = settings['component']
 
@@ -700 +701 @@
                 when = datetime.now(utc)
 
-                if DRY_RUN:
+                if self.DRY_RUN:
                         ticket_id = 'DRY_RUN'
                 else:
@@ -725 +726 @@
 
                 if changed:
-                        if DRY_RUN:
+                        if self.DRY_RUN:
                                 print 'DRY_RUN: tkt.save_changes(self.author, comment)'
                         else:
@@ -732 +733 @@
 
                 if self.notification:
-                        if DRY_RUN:
+                        if self.DRY_RUN:
                                 print 'DRY_RUN: self.notify(tkt, True)'
                         else:
-                                self.notify(tkt, True)
+                                if not spam:
+                                        self.notify(tkt, True)
                                 #self.notify(tkt, False)
 
@@ -757 +759 @@
                         return False
 
-            # If component is true then we drop the message drop message
-                #
-                if self.spam(m) == True:
+                # If drop the message
+                #
+                if self.spam(m) == 'drop':
                         return False
 
+                elif self.spam(m) == 'spam':
+                        spam_msg = True
+
+                else:
+                        spam_msg = False
 
                 if self.get_config('notification', 'smtp_enabled') in ['true']:
@@ -771 +778 @@
                 #
                 if self.TICKET_UPDATE > 0:
-                        if self.ticket_update(m):
+                        if self.ticket_update(m, spam_msg):
                                 return True
 
-                self.new_ticket(m)
+                self.new_ticket(m, spam_msg)
 
         def strip_signature(self, text):
@@ -905 +912 @@
                 A wrapper for the TRAC notify function. So we can use templates
                 """
-                if tkt['component'] == 'Spam':
-                        return  
-
                 try:
                         # create false {abs_}href properties, to trick Notify()
@@ -1097 +1101 @@
         # Default config file
         #
-        global DRY_RUN
-
         configfile = '@email2trac_conf@'
         project = ''
         component = ''
         ticket_prefix = 'default'
+        dry_run = None
 
         ENABLE_SYSLOG = 0
+
 
         SHORT_OPT = 'chf:np:t:'
@@ -1126 +1130 @@
                         configfile = value
                 elif opt in ['-n', '--dry-run']:
-                        DRY_RUN = True
+                        dry_run = True
                 elif opt in ['-p', '--project']:
                         project_name = value
@@ -1195 +1199 @@
 
                 env = Environment(settings['project'], create=0)
-                tktparser = TicketEmailParser(env, settings, float(version))
+                tktparser = TicketEmailParser(env, settings, float(version), dry_run)
                 tktparser.parse(sys.stdin)