comparison hgext/bugzilla.py @ 45942:89a2afe31e82

formating: upgrade to black 20.8b1

This required a couple of small tweaks to un-confuse black, but now it
works. Big formatting changes come from:

 * Dramatically improved collection-splitting logic upstream
 * Black having a strong (correct IMO) opinion that """ is better than '''

Differential Revision: https://phab.mercurial-scm.org/D9430

45941:346af7687c6f

configtable = {}
configitem = registrar.configitem(configtable)

configitem(
    b'bugzilla', b'apikey', default=b'',
)
configitem(
    b'bugzilla', b'bzdir', default=b'/var/www/html/bugzilla',
)
configitem(
    b'bugzilla', b'bzemail', default=None,
)
configitem(
    b'bugzilla', b'bzurl', default=b'http://localhost/bugzilla/',
)
configitem(
    b'bugzilla', b'bzuser', default=None,
)
configitem(
    b'bugzilla', b'db', default=b'bugs',
)
configitem(
    b'bugzilla',
    b'fixregexp',
    default=(

        br'(?P<ids>(?:#?\d+\s*(?:,?\s*(?:and)?)?\s*)+)'
        br'\.?\s*(?:h(?:ours?)?\s*(?P<hours>\d*(?:\.\d+)?))?'
    ),
)
configitem(
    b'bugzilla', b'fixresolution', default=b'FIXED',
)
configitem(
    b'bugzilla', b'fixstatus', default=b'RESOLVED',
)
configitem(
    b'bugzilla', b'host', default=b'localhost',
)
configitem(
    b'bugzilla', b'notify', default=configitem.dynamicdefault,
)
configitem(
    b'bugzilla', b'password', default=None,
)
configitem(
    b'bugzilla',
    b'regexp',
    default=(

        br'(?P<ids>(?:\d+\s*(?:,?\s*(?:and)?)?\s*)+)'
        br'\.?\s*(?:h(?:ours?)?\s*(?P<hours>\d*(?:\.\d+)?))?'
    ),
)
configitem(
    b'bugzilla', b'strip', default=0,
)
configitem(
    b'bugzilla', b'style', default=None,
)
configitem(
    b'bugzilla', b'template', default=None,
)
configitem(
    b'bugzilla', b'timeout', default=5,
)
configitem(
    b'bugzilla', b'user', default=b'bugs',
)
configitem(
    b'bugzilla', b'usermap', default=None,
)
configitem(
    b'bugzilla', b'version', default=None,
)


class bzaccess(object):
    '''Base class for access to Bugzilla.'''

    def filter_cset_known_bug_ids(self, node, bugs):
        '''remove bug IDs where node occurs in comment text from bugs.'''

    def updatebug(self, bugid, newstate, text, committer):
        '''update the specified bug. Add comment text and set new states.

        If possible add the comment as being from the committer of
        the changeset. Otherwise use the default Bugzilla user.
        '''

    def notify(self, bugs, committer):
        '''Force sending of Bugzilla notification emails.

        Only required if the access method does not trigger notification
        emails automatically.
        '''


# Bugzilla via direct access to MySQL database.
class bzmysql(bzaccess):
    '''Support for direct MySQL access to Bugzilla.

    The earliest Bugzilla version this is tested with is version 2.16.

    If your Bugzilla is version 3.4 or above, you are strongly
    recommended to use the XMLRPC access method instead.
    '''

    @staticmethod
    def sql_buglist(ids):
        '''return SQL-friendly list of bug ids'''
        return b'(' + b','.join(map(str, ids)) + b')'

                userid = int(all[0][0])
            self.user_ids[user] = userid
            return userid

    def get_bugzilla_user(self, committer):
        '''See if committer is a registered bugzilla user. Return
        bugzilla username and userid if so. If not, return default
        bugzilla username and userid.'''
        user = self.map_committer(committer)
        try:
            userid = self.get_user_id(user)
        except KeyError:
            try:

                    % (user, defaultuser)
                )
        return (user, userid)

    def updatebug(self, bugid, newstate, text, committer):
        '''update bug state with comment text.

        Try adding comment as committer of changeset, otherwise as
        default bugzilla user.'''
        if len(newstate) > 0:
            self.ui.warn(_(b"Bugzilla/MySQL cannot update bug state\n"))

        (user, userid) = self.get_bugzilla_user(committer)
        now = time.strftime('%Y-%m-%d %H:%M:%S')

            if fieldname == b"id":
                fieldname = b"bug_id"
            return b"@%s = %s" % (fieldname, pycompat.bytestr(value))

    def send_bug_modify_email(self, bugid, commands, comment, committer):
        '''send modification message to Bugzilla bug via email.

        The message format is documented in the Bugzilla email_in.pl
        specification. commands is a list of command lines, comment is the
        comment text.

        To stop users from crafting commit comments with
        Bugzilla commands, specify the bug ID via the message body, rather
        than the subject line, and leave a blank line after it.
        '''
        user = self.map_committer(committer)
        matches = self.bzproxy.User.get(
            {b'match': [user], b'token': self.bztoken}
        )
        if not matches[b'users']:

                    % (bugid, sn)
                )
                del bugs[bugid]

    def updatebug(self, bugid, newstate, text, committer):
        '''update the specified bug. Add comment text and set new states.

        If possible add the comment as being from the committer of
        the changeset. Otherwise use the default Bugzilla user.
        '''
        bugmod = {}
        if b'hours' in newstate:
            bugmod[b'work_time'] = newstate[b'hours']
        if b'fix' in newstate:
            bugmod[b'status'] = self.fixstatus

                },
            )
            self.ui.debug(b'added comment to bug %s\n' % bugid)

    def notify(self, bugs, committer):
        '''Force sending of Bugzilla notification emails.

        Only required if the access method does not trigger notification
        emails automatically.
        '''
        pass


class bugzilla(object):
    # supported versions of bugzilla. different versions have

            self.ui.config(b'bugzilla', b'fixregexp'), re.IGNORECASE
        )
        self.split_re = re.compile(br'\D+')

    def find_bugs(self, ctx):
        '''return bugs dictionary created from commit comment.

        Extract bug info from changeset comments. Filter out any that are
        not known to Bugzilla, and any that already have a reference to
        the given changeset in their comments.
        '''
        start = 0
        bugs = {}
        bugmatch = self.bug_re.search(ctx.description(), start)
        fixmatch = self.fix_re.search(ctx.description(), start)
        while True:

    def update(self, bugid, newstate, ctx):
        '''update bugzilla bug with reference to changeset.'''

        def webroot(root):
            '''strip leading prefix of repo root and turn into
            url-safe path.'''
            count = int(self.ui.config(b'bugzilla', b'strip'))
            root = util.pconvert(root)
            while count > 0:
                c = root.find(b'/')
                if c == -1:

        '''ensure Bugzilla users are notified of bug change.'''
        self.bzdriver.notify(bugs, committer)


def hook(ui, repo, hooktype, node=None, **kwargs):
    '''add comment to bugzilla for each changeset that refers to a
    bugzilla bug id. only add a comment once per bug, so same change
    seen multiple times does not fill bug with duplicate data.'''
    if node is None:
        raise error.Abort(
            _(b'hook type %s does not pass a changeset id') % hooktype
        )
    try:

45942:89a2afe31e82

configtable = {}
configitem = registrar.configitem(configtable)

configitem(
    b'bugzilla',
    b'apikey',
    default=b'',
)
configitem(
    b'bugzilla',
    b'bzdir',
    default=b'/var/www/html/bugzilla',
)
configitem(
    b'bugzilla',
    b'bzemail',
    default=None,
)
configitem(
    b'bugzilla',
    b'bzurl',
    default=b'http://localhost/bugzilla/',
)
configitem(
    b'bugzilla',
    b'bzuser',
    default=None,
)
configitem(
    b'bugzilla',
    b'db',
    default=b'bugs',
)
configitem(
    b'bugzilla',
    b'fixregexp',
    default=(

        br'(?P<ids>(?:#?\d+\s*(?:,?\s*(?:and)?)?\s*)+)'
        br'\.?\s*(?:h(?:ours?)?\s*(?P<hours>\d*(?:\.\d+)?))?'
    ),
)
configitem(
    b'bugzilla',
    b'fixresolution',
    default=b'FIXED',
)
configitem(
    b'bugzilla',
    b'fixstatus',
    default=b'RESOLVED',
)
configitem(
    b'bugzilla',
    b'host',
    default=b'localhost',
)
configitem(
    b'bugzilla',
    b'notify',
    default=configitem.dynamicdefault,
)
configitem(
    b'bugzilla',
    b'password',
    default=None,
)
configitem(
    b'bugzilla',
    b'regexp',
    default=(

        br'(?P<ids>(?:\d+\s*(?:,?\s*(?:and)?)?\s*)+)'
        br'\.?\s*(?:h(?:ours?)?\s*(?P<hours>\d*(?:\.\d+)?))?'
    ),
)
configitem(
    b'bugzilla',
    b'strip',
    default=0,
)
configitem(
    b'bugzilla',
    b'style',
    default=None,
)
configitem(
    b'bugzilla',
    b'template',
    default=None,
)
configitem(
    b'bugzilla',
    b'timeout',
    default=5,
)
configitem(
    b'bugzilla',
    b'user',
    default=b'bugs',
)
configitem(
    b'bugzilla',
    b'usermap',
    default=None,
)
configitem(
    b'bugzilla',
    b'version',
    default=None,
)


class bzaccess(object):
    '''Base class for access to Bugzilla.'''

    def filter_cset_known_bug_ids(self, node, bugs):
        '''remove bug IDs where node occurs in comment text from bugs.'''

    def updatebug(self, bugid, newstate, text, committer):
        """update the specified bug. Add comment text and set new states.

        If possible add the comment as being from the committer of
        the changeset. Otherwise use the default Bugzilla user.
        """

    def notify(self, bugs, committer):
        """Force sending of Bugzilla notification emails.

        Only required if the access method does not trigger notification
        emails automatically.
        """


# Bugzilla via direct access to MySQL database.
class bzmysql(bzaccess):
    """Support for direct MySQL access to Bugzilla.

    The earliest Bugzilla version this is tested with is version 2.16.

    If your Bugzilla is version 3.4 or above, you are strongly
    recommended to use the XMLRPC access method instead.
    """

    @staticmethod
    def sql_buglist(ids):
        '''return SQL-friendly list of bug ids'''
        return b'(' + b','.join(map(str, ids)) + b')'

                userid = int(all[0][0])
            self.user_ids[user] = userid
            return userid

    def get_bugzilla_user(self, committer):
        """See if committer is a registered bugzilla user. Return
        bugzilla username and userid if so. If not, return default
        bugzilla username and userid."""
        user = self.map_committer(committer)
        try:
            userid = self.get_user_id(user)
        except KeyError:
            try:

                    % (user, defaultuser)
                )
        return (user, userid)

    def updatebug(self, bugid, newstate, text, committer):
        """update bug state with comment text.

        Try adding comment as committer of changeset, otherwise as
        default bugzilla user."""
        if len(newstate) > 0:
            self.ui.warn(_(b"Bugzilla/MySQL cannot update bug state\n"))

        (user, userid) = self.get_bugzilla_user(committer)
        now = time.strftime('%Y-%m-%d %H:%M:%S')

            if fieldname == b"id":
                fieldname = b"bug_id"
            return b"@%s = %s" % (fieldname, pycompat.bytestr(value))

    def send_bug_modify_email(self, bugid, commands, comment, committer):
        """send modification message to Bugzilla bug via email.

        The message format is documented in the Bugzilla email_in.pl
        specification. commands is a list of command lines, comment is the
        comment text.

        To stop users from crafting commit comments with
        Bugzilla commands, specify the bug ID via the message body, rather
        than the subject line, and leave a blank line after it.
        """
        user = self.map_committer(committer)
        matches = self.bzproxy.User.get(
            {b'match': [user], b'token': self.bztoken}
        )
        if not matches[b'users']:

                    % (bugid, sn)
                )
                del bugs[bugid]

    def updatebug(self, bugid, newstate, text, committer):
        """update the specified bug. Add comment text and set new states.

        If possible add the comment as being from the committer of
        the changeset. Otherwise use the default Bugzilla user.
        """
        bugmod = {}
        if b'hours' in newstate:
            bugmod[b'work_time'] = newstate[b'hours']
        if b'fix' in newstate:
            bugmod[b'status'] = self.fixstatus

                },
            )
            self.ui.debug(b'added comment to bug %s\n' % bugid)

    def notify(self, bugs, committer):
        """Force sending of Bugzilla notification emails.

        Only required if the access method does not trigger notification
        emails automatically.
        """
        pass


class bugzilla(object):
    # supported versions of bugzilla. different versions have

            self.ui.config(b'bugzilla', b'fixregexp'), re.IGNORECASE
        )
        self.split_re = re.compile(br'\D+')

    def find_bugs(self, ctx):
        """return bugs dictionary created from commit comment.

        Extract bug info from changeset comments. Filter out any that are
        not known to Bugzilla, and any that already have a reference to
        the given changeset in their comments.
        """
        start = 0
        bugs = {}
        bugmatch = self.bug_re.search(ctx.description(), start)
        fixmatch = self.fix_re.search(ctx.description(), start)
        while True:

    def update(self, bugid, newstate, ctx):
        '''update bugzilla bug with reference to changeset.'''

        def webroot(root):
            """strip leading prefix of repo root and turn into
            url-safe path."""
            count = int(self.ui.config(b'bugzilla', b'strip'))
            root = util.pconvert(root)
            while count > 0:
                c = root.find(b'/')
                if c == -1:

        '''ensure Bugzilla users are notified of bug change.'''
        self.bzdriver.notify(bugs, committer)


def hook(ui, repo, hooktype, node=None, **kwargs):
    """add comment to bugzilla for each changeset that refers to a
    bugzilla bug id. only add a comment once per bug, so same change
    seen multiple times does not fill bug with duplicate data."""
    if node is None:
        raise error.Abort(
            _(b'hook type %s does not pass a changeset id') % hooktype
        )
    try: