+
−#!/usr/bin/env python3
+
−
+
−
+
−# SàT plugin for import external blogs
+
−# Copyright (C) 2009-2021 Jérôme Poisson (goffi@goffi.org)
+
−
+
−# This program is free software: you can redistribute it and/or modify
+
−# it under the terms of the GNU Affero General Public License as published by
+
−# the Free Software Foundation, either version 3 of the License, or
+
−# (at your option) any later version.
+
−
+
−# This program is distributed in the hope that it will be useful,
+
−# but WITHOUT ANY WARRANTY; without even the implied warranty of
+
−# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+
−# GNU Affero General Public License for more details.
+
−
+
−# You should have received a copy of the GNU Affero General Public License
+
−# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
−
+
−from sat.core.i18n import _, D_
+
−from sat.core.constants import Const as C
+
−from sat.core.log import getLogger
+
−
+
−log = getLogger(__name__)
+
−from sat.core import exceptions
+
−
+
−# from twisted.internet import threads
+
−from twisted.internet import defer
+
−import os.path
+
−from lxml import etree
+
−from sat.tools.common import date_utils
+
−
+
−
+
−PLUGIN_INFO = {
+
−    C.PI_NAME: "Bugzilla import",
+
−    C.PI_IMPORT_NAME: "IMPORT_BUGZILLA",
+
−    C.PI_TYPE: C.PLUG_TYPE_BLOG,
+
−    C.PI_DEPENDENCIES: ["TICKETS_IMPORT"],
+
−    C.PI_MAIN: "BugzillaImport",
+
−    C.PI_HANDLER: "no",
+
−    C.PI_DESCRIPTION: _("""Tickets importer for Bugzilla"""),
+
−}
+
−
+
−SHORT_DESC = D_("import tickets from Bugzilla xml export file")
+
−
+
−LONG_DESC = D_(
+
−    """This importer handle Bugzilla xml export file.
+
−
+
−To use it, you'll need to export tickets using XML.
+
−Tickets will be uploaded with the same ID as for Bugzilla, any existing ticket with this ID will be replaced.
+
−
+
−location: you must use the absolute path to your .xml file
+
−"""
+
−)
+
−
+
−STATUS_MAP = {
+
−    "NEW": "queued",
+
−    "ASSIGNED": "started",
+
−    "RESOLVED": "review",
+
−    "CLOSED": "closed",
+
−    "REOPENED": "started",  # we loose data here because there is no need on basic workflow to have a reopened status
+
−}
+
−
+
−
+
−class BugzillaParser(object):
+
−    # TODO: add a way to reassign values
+
−
+
−    def parse(self, file_path):
+
−        tickets = []
+
−        root = etree.parse(file_path)
+
−
+
−        for bug in root.xpath("bug"):
+
−            ticket = {}
+
−            ticket["id"] = bug.findtext("bug_id")
+
−            ticket["created"] = date_utils.date_parse(bug.findtext("creation_ts"))
+
−            ticket["updated"] = date_utils.date_parse(bug.findtext("delta_ts"))
+
−            ticket["title"] = bug.findtext("short_desc")
+
−            reporter_elt = bug.find("reporter")
+
−            ticket["author"] = reporter_elt.get("name")
+
−            if ticket["author"] is None:
+
−                if "@" in reporter_elt.text:
+
−                    ticket["author"] = reporter_elt.text[
+
−                        : reporter_elt.text.find("@")
+
−                    ].title()
+
−                else:
+
−                    ticket["author"] = "no name"
+
−            ticket["author_email"] = reporter_elt.text
+
−            assigned_to_elt = bug.find("assigned_to")
+
−            ticket["assigned_to_name"] = assigned_to_elt.get("name")
+
−            ticket["assigned_to_email"] = assigned_to_elt.text
+
−            ticket["cc_emails"] = [e.text for e in bug.findall("cc")]
+
−            ticket["priority"] = bug.findtext("priority").lower().strip()
+
−            ticket["severity"] = bug.findtext("bug_severity").lower().strip()
+
−            ticket["product"] = bug.findtext("product")
+
−            ticket["component"] = bug.findtext("component")
+
−            ticket["version"] = bug.findtext("version")
+
−            ticket["platform"] = bug.findtext("rep_platform")
+
−            ticket["os"] = bug.findtext("op_sys")
+
−            ticket["status"] = STATUS_MAP.get(bug.findtext("bug_status"), "queued")
+
−            ticket["milestone"] = bug.findtext("target_milestone")
+
−
+
−            body = None
+
−            comments = []
+
−            for longdesc in bug.findall("long_desc"):
+
−                if body is None:
+
−                    body = longdesc.findtext("thetext")
+
−                else:
+
−                    who = longdesc.find("who")
+
−                    comment = {
+
−                        "id": longdesc.findtext("commentid"),
+
−                        "author_email": who.text,
+
−                        "published": date_utils.date_parse(longdesc.findtext("bug_when")),
+
−                        "author": who.get("name", who.text),
+
−                        "content": longdesc.findtext("thetext"),
+
−                    }
+
−                    comments.append(comment)
+
−
+
−            ticket["body"] = body
+
−            ticket["comments"] = comments
+
−            tickets.append(ticket)
+
−
+
−        tickets.sort(key=lambda t: int(t["id"]))
+
−        return (tickets, len(tickets))
+
−
+
−
+
−class BugzillaImport(object):
+
−    def __init__(self, host):
+
−        log.info(_("Bugilla import plugin initialization"))
+
−        self.host = host
+
−        host.plugins["TICKETS_IMPORT"].register(
+
−            "bugzilla", self.import_, SHORT_DESC, LONG_DESC
+
−        )
+
−
+
−    def import_(self, client, location, options=None):
+
−        if not os.path.isabs(location):
+
−            raise exceptions.DataError(
+
−                "An absolute path to XML data need to be given as location"
+
−            )
+
−        bugzilla_parser = BugzillaParser()
+
−        # d = threads.deferToThread(bugzilla_parser.parse, location)
+
−        d = defer.maybeDeferred(bugzilla_parser.parse, location)
+
−        return d