server/post-receive-trac.py

   1 #!/usr/bin/env python
   2
   3 # This should work:
   4 #
   5 #    Changed blah and foo to do this or that. Re #10 and #12, and qa #12.
   6 #
   7
   8 import re
   9 import sys
  10 from datetime import datetime
  11
  12 from trac.env import open_environment
  13 from trac.ticket.notification import TicketNotifyEmail
  14 from trac.ticket import Ticket
  15 from trac.ticket.web_ui import TicketModule
  16 from trac.util.datefmt import utc
  17 from trac.versioncontrol.api import NoSuchChangeset
  18
  19 project = sys.argv[1]
  20 refname = sys.argv[2]
  21 describe = sys.argv[3]
  22 rev = sys.argv[4]
  23
  24 def refs(ticket):
  25         pass
  26
  27 def qa(ticket):
  28         if ticket['phase'] == 'Final Fixing':
  29                 ticket['phase'] = 'Final QA'
  30         else:
  31                 ticket['phase'] = 'Initial QA'
  32         ticket['owner'] = ''
  33         ticket['status'] = 'new'
  34
  35 commands = { 're': refs, 'refs': refs, 'qa': qa }
  36 commandPattern = re.compile(r'(?P<action>[A-Za-z]*).?(?P<ticket>#[0-9]+(?:(?:[, &]*|[ ]?and[ ]?)#[0-9]+)*)')
  37 ticketPattern = re.compile(r'#([0-9]*)')
  38 authorPattern = re.compile(r'<(.+)@')
  39 tickets = {}
  40
  41 env = open_environment(project)
  42 repos = env.get_repository()
  43 repos.sync()
  44
  45 changeset = repos.get_changeset(rev)
  46
  47 for command, ticketList in commandPattern.findall(changeset.message):
  48         if commands.has_key(command.lower()):
  49                 for ticketId in ticketPattern.findall(ticketList):
  50                         tickets.setdefault(ticketId, []).append(commands[command.lower()])
  51
  52 for ticketId, commands in tickets.iteritems():
  53         db = env.get_db_cnx()
  54
  55         ticket = Ticket(env, int(ticketId), db)
  56         for command in commands:
  57                 command(ticket)
  58
  59         # determine sequence number...
  60         cnum = 0
  61         tm = TicketModule(env)
  62         for change in tm.grouped_changelog_entries(ticket, db):
  63                 if change['permanent']:
  64                         cnum += 1
  65
  66         username = authorPattern.findall(changeset.author)[0]
  67         now = datetime.now(utc)
  68         message = "(On %s [changeset:%s %s]) %s" % (refname, rev, describe, changeset.message)
  69         ticket['branch'] = refname
  70         ticket.save_changes(username, message, now, db, cnum+1)
  71         db.commit()
  72
  73         tn = TicketNotifyEmail(env)
  74         tn.notify(ticket, newticket=0, modtime=now)
  75