'''Bugzilla integration for adding versions from tags+
−
+
−
The hook updates the Bugzilla database directly. Only Bugzilla installations+
−
using MySQL are supported.+
−
+
−
This hook uses the same .hgrc parameters as the default Bugzilla hook. There+
−
is no need for configuring the same stuff twice. (connection, etc.)+
−
+
−
Configuring the extension: (same as Bugzilla -hook)+
−
+
−
    [bugzilla]+
−
    host       Hostname of the MySQL server holding the Bugzilla database.+
−
    db         Name of the Bugzilla database in MySQL. Default 'bugs'.+
−
    user       Username to use to access MySQL server. Default 'bugs'.+
−
    password   Password to use to access MySQL server.+
−
    timeout    Database connection timeout (seconds). Default 5.+
−
+
−
Additional elements under Bugzilla -section: (new items)+
−
    [bugzilla]+
−
    product    The name on the Bugzilla product that is used for adding +
−
               the new versions.+
−
+
−
Activating the extension:+
−
+
−
    [extensions]+
−
    hgext.versiontobugzilla =+
−
+
−
    [hooks]+
−
    incoming.versiontobugzilla = python:hgext.versiontobugzilla.hook+
−
+
−
Example configuration in hgrc:+
−
    [bugzilla]+
−
    host = localhost+
−
    user = bugs+
−
    password = password+
−
    product = my_product+
−
+
−
    [extensions]+
−
    hgext.versiontobugzilla =+
−
+
−
    [hooks]+
−
    incoming.versiontobugzilla = python:hgext.versiontobugzilla.hook+
−
'''+
−
+
−
from mercurial import util+
−
import re+
−
+
−
MySQLdb = None+
−
+
−
class BugzillaClient:+
−
    +
−
    def __init__(self, ui, repo, node):+
−
        self.tag = None+
−
        self.ui = ui+
−
        self.repo = repo+
−
        self.node = node+
−
        self.product = ui.config('bugzilla', 'product')+
−
        self.host = ui.config('bugzilla', 'host', 'localhost')+
−
        self.user = ui.config('bugzilla', 'user', 'bugs')+
−
        self.passwd = ui.config('bugzilla', 'password')+
−
        self.db = ui.config('bugzilla', 'db', 'bugs')+
−
        self.timeout = int(ui.config('bugzilla', 'timeout', 10))+
−
        self.connection = MySQLdb.connect(host=self.host, user=self.user, passwd=self.passwd,+
−
                                    db=self.db, connect_timeout=self.timeout)+
−
        self.cursor = self.connection.cursor()+
−
+
−
    def printMessageInVerboseMode(self, message):+
−
        '''Prints a message to console if hg has been executed with -v option.'''+
−
        self.ui.note(message)+
−
+
−
    def executeDatabaseQuery(self, *args, **kwargs):+
−
        self.printMessageInVerboseMode('Bugzilla: query: %s %s\n' % (args, kwargs))+
−
        try:+
−
            self.cursor.execute(*args, **kwargs)+
−
        except MySQLdb.MySQLError:+
−
            self.printMessageInVerboseMode('Bugzilla: failed query: %s %s\n' % (args, kwargs))+
−
            raise+
−
+
−
    def commitContainsTag(self):+
−
        self.parseTagFromCommitMessage()+
−
        if self.tag:+
−
            return True+
−
        else:+
−
            return False+
−
+
−
    def parseTagFromCommitMessage(self):+
−
        ctx = self.repo[self.node]+
−
        version_re = re.compile(('Added tag (.+) for changeset [0-9a-h]+'), re.IGNORECASE)+
−
        m = version_re.search(ctx.description())+
−
        if m:+
−
            self.tag = m.group(1)+
−
+
−
    def insertTagIntoDatabase(self):+
−
        self.makeSureThatProductExists()+
−
        if not self.doesVersionAlreadyExist():+
−
            self.printMessageInVerboseMode("Bugzilla: adding version '%s' to product '%s' in database.\n" % (self.tag, self.product))+
−
            self.insertNewVersionIntoDatabase()+
−
        else:+
−
            self.printMessageInVerboseMode("Bugzilla: product '%s' already has a version '%s' in database. Not trying to add it again." % (self.product, self.tag))+
−
+
−
    def makeSureThatProductExists(self):+
−
        self.executeDatabaseQuery('select id from products where name = %s', (self.product,))+
−
        ids = self.cursor.fetchall()+
−
        if len(ids) != 1:+
−
            raise util.Abort("Product '%s' does not exist in database, please check the [bugzilla] -section in hgrc." % self.product)+
−
+
−
    def doesVersionAlreadyExist(self):+
−
        self.executeDatabaseQuery('select * from versions where value = %s and product_id in (select id from products where name=%s )', (self.tag, self.product))+
−
        ids = self.cursor.fetchall()+
−
        if len(ids) == 1:+
−
            return True+
−
        else:+
−
            return False+
−
+
−
    def insertNewVersionIntoDatabase(self):+
−
        self.executeDatabaseQuery('insert into versions (value, product_id) values (%s, (select id from products where name=%s ))', (self.tag, self.product))+
−
        self.connection.commit()+
−
+
−
def hook(ui, repo, hooktype, node=None, **kwargs):+
−
+
−
    try:+
−
        import MySQLdb as mysql+
−
        global MySQLdb+
−
        MySQLdb = mysql+
−
    except ImportError, err:+
−
        raise util.Abort('MySQL driver not installed: %s' % err)+
−
+
−
    if node is None:+
−
        raise util.Abort('Only hooks that have changesetid''s can be used.')+
−
+
−
    try: +
−
        bzClient = BugzillaClient(ui, repo, node)+
−
        if bzClient.commitContainsTag():+
−
            bzClient.insertTagIntoDatabase()+
−
    except MySQLdb.MySQLError, err:+
−
        raise util.Abort('Database error: %s' % err[1])+
−