comparison hgext/bugzilla.py @ 15870:f4c859293ed4 stable

bugzilla: make XMLRPC interface support http and https access

Inadvertently support is currently only for https. For some reason I
thought xmlrpclib.SafeTransport did http and https, but it is https only.

So create http and https XMLRPC transports that retain cookies. Decide which
to use by inspecting the Bugzilla URL.

15848:012b285cf643

'''

from mercurial.i18n import _
from mercurial.node import short
from mercurial import cmdutil, mail, templater, util
import re, time, xmlrpclib

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

    def __init__(self, ui):

            raise util.Abort(_('unknown database schema'))
        return ids[0][0]

# Buzgilla via XMLRPC interface.

class CookieSafeTransport(xmlrpclib.SafeTransport):
    """A SafeTransport that retains cookies over its lifetime.

    The regular xmlrpclib transports ignore cookies. Which causes
    a bit of a problem when you need a cookie-based login, as with
    the Bugzilla XMLRPC interface.

    So this is a SafeTransport which looks for cookies being set
    in responses and saves them to add to all future requests.
    It appears a SafeTransport can do both HTTP and HTTPS sessions,
    which saves us having to do a CookieTransport too.
    """

    # Inspiration drawn from
    # http://blog.godson.in/2010/09/how-to-make-python-xmlrpclib-client.html
    # http://www.itkovian.net/base/transport-class-for-pythons-xml-rpc-lib/

        parser.feed(payload)
        parser.close()

        return unmarshaller.close()

class bzxmlrpc(bzaccess):
    """Support for access to Bugzilla via the Bugzilla XMLRPC API.

    Requires a minimum Bugzilla version 3.4.
    """

        bzweb = bzweb.rstrip("/") + "/xmlrpc.cgi"

        user = self.ui.config('bugzilla', 'user', 'bugs')
        passwd = self.ui.config('bugzilla', 'password')

        self.bzproxy = xmlrpclib.ServerProxy(bzweb, CookieSafeTransport())
        self.bzproxy.User.login(dict(login=user, password=passwd))

    def get_bug_comments(self, id):
        """Return a string with all comment text for a bug."""
        c = self.bzproxy.Bug.comments(dict(ids=[id]))
        return ''.join([t['text'] for t in c['bugs'][str(id)]['comments']])

15870:f4c859293ed4

'''

from mercurial.i18n import _
from mercurial.node import short
from mercurial import cmdutil, mail, templater, util
import re, time, urlparse, xmlrpclib

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

    def __init__(self, ui):

            raise util.Abort(_('unknown database schema'))
        return ids[0][0]

# Buzgilla via XMLRPC interface.

class cookietransportrequest(object):
    """A Transport request method that retains cookies over its lifetime.

    The regular xmlrpclib transports ignore cookies. Which causes
    a bit of a problem when you need a cookie-based login, as with
    the Bugzilla XMLRPC interface.

    So this is a helper for defining a Transport which looks for
    cookies being set in responses and saves them to add to all future
    requests.
    """

    # Inspiration drawn from
    # http://blog.godson.in/2010/09/how-to-make-python-xmlrpclib-client.html
    # http://www.itkovian.net/base/transport-class-for-pythons-xml-rpc-lib/

        parser.feed(payload)
        parser.close()

        return unmarshaller.close()

# The explicit calls to the underlying xmlrpclib __init__() methods are
# necessary. The xmlrpclib.Transport classes are old-style classes, and
# it turns out their __init__() doesn't get called when doing multiple
# inheritance with a new-style class.
class cookietransport(cookietransportrequest, xmlrpclib.Transport):
    def __init__(self, use_datetime=0):
        xmlrpclib.Transport.__init__(self, use_datetime)

class cookiesafetransport(cookietransportrequest, xmlrpclib.SafeTransport):
    def __init__(self, use_datetime=0):
        xmlrpclib.SafeTransport.__init__(self, use_datetime)

class bzxmlrpc(bzaccess):
    """Support for access to Bugzilla via the Bugzilla XMLRPC API.

    Requires a minimum Bugzilla version 3.4.
    """

        bzweb = bzweb.rstrip("/") + "/xmlrpc.cgi"

        user = self.ui.config('bugzilla', 'user', 'bugs')
        passwd = self.ui.config('bugzilla', 'password')

        self.bzproxy = xmlrpclib.ServerProxy(bzweb, self.transport(bzweb))
        self.bzproxy.User.login(dict(login=user, password=passwd))

    def transport(self, uri):
        if urlparse.urlparse(uri, "http")[0] == "https":
            return cookiesafetransport()
        else:
            return cookietransport()

    def get_bug_comments(self, id):
        """Return a string with all comment text for a bug."""
        c = self.bzproxy.Bug.comments(dict(ids=[id]))
        return ''.join([t['text'] for t in c['bugs'][str(id)]['comments']])