diff sat/plugins/plugin_tickets_import.py @ 2624:56f94936df1e

code style reformatting using black
author Goffi <goffi@goffi.org>
date Wed, 27 Jun 2018 20:14:46 +0200
parents 26edcf3a30eb
children 003b8b4b56a7
line wrap: on
line diff
--- a/sat/plugins/plugin_tickets_import.py	Wed Jun 27 07:51:29 2018 +0200
+++ b/sat/plugins/plugin_tickets_import.py	Wed Jun 27 20:14:46 2018 +0200
@@ -21,6 +21,7 @@
 from sat.core.constants import Const as C
 from sat.core import exceptions
 from sat.core.log import getLogger
+
 log = getLogger(__name__)
 from twisted.internet import defer
 from sat.tools.common import uri
@@ -34,15 +35,17 @@
     C.PI_DEPENDENCIES: ["IMPORT", "XEP-0060", "XEP-0277", "PUBSUB_SCHEMA"],
     C.PI_MAIN: "TicketsImportPlugin",
     C.PI_HANDLER: "no",
-    C.PI_DESCRIPTION: _(u"""Tickets import management:
-This plugin manage the different tickets importers which can register to it, and handle generic importing tasks.""")
+    C.PI_DESCRIPTION: _(
+        u"""Tickets import management:
+This plugin manage the different tickets importers which can register to it, and handle generic importing tasks."""
+    ),
 }
 
-OPT_MAPPING = 'mapping'
-FIELDS_LIST = (u'labels', u'cc_emails')  # fields which must have a list as value
-FIELDS_DATE = (u'created', u'updated')
+OPT_MAPPING = "mapping"
+FIELDS_LIST = (u"labels", u"cc_emails")  # fields which must have a list as value
+FIELDS_DATE = (u"created", u"updated")
 
-NS_TICKETS = 'org.salut-a-toi.tickets:0'
+NS_TICKETS = "org.salut-a-toi.tickets:0"
 
 
 class TicketsImportPlugin(object):
@@ -54,13 +57,15 @@
         log.info(_("plugin Tickets Import initialization"))
         self.host = host
         self._importers = {}
-        self._p = host.plugins['XEP-0060']
-        self._m = host.plugins['XEP-0277']
-        self._s = host.plugins['PUBSUB_SCHEMA']
-        host.plugins['IMPORT'].initialize(self, u'tickets')
+        self._p = host.plugins["XEP-0060"]
+        self._m = host.plugins["XEP-0277"]
+        self._s = host.plugins["PUBSUB_SCHEMA"]
+        host.plugins["IMPORT"].initialize(self, u"tickets")
 
     @defer.inlineCallbacks
-    def importItem(self, client, item_import_data, session, options, return_data, service, node):
+    def importItem(
+        self, client, item_import_data, session, options, return_data, service, node
+    ):
         """
 
         @param item_import_data(dict): no key is mandatory, but if a key doesn't exists in dest form, it will be ignored.
@@ -101,20 +106,24 @@
                     If you specify several import ticket key to the same dest key,
                     the values will be joined with line feeds
         """
-        if 'comments_uri' in item_import_data:
-            raise exceptions.DataError(_(u'comments_uri key will be generated and must not be used by importer'))
+        if "comments_uri" in item_import_data:
+            raise exceptions.DataError(
+                _(u"comments_uri key will be generated and must not be used by importer")
+            )
         for key in FIELDS_LIST:
             if not isinstance(item_import_data.get(key, []), list):
-                raise exceptions.DataError(_(u'{key} must be a list').format(key=key))
+                raise exceptions.DataError(_(u"{key} must be a list").format(key=key))
         for key in FIELDS_DATE:
             try:
                 item_import_data[key] = utils.xmpp_date(item_import_data[key])
             except KeyError:
                 continue
-        if session[u'root_node'] is None:
-            session[u'root_node'] = NS_TICKETS
-        if not 'schema' in session:
-            session['schema'] = yield self._s.getSchemaForm(client, service, node or session[u'root_node'])
+        if session[u"root_node"] is None:
+            session[u"root_node"] = NS_TICKETS
+        if not "schema" in session:
+            session["schema"] = yield self._s.getSchemaForm(
+                client, service, node or session[u"root_node"]
+            )
         defer.returnValue(item_import_data)
 
     @defer.inlineCallbacks
@@ -122,40 +131,53 @@
         # TODO: force "open" permission (except if private, check below)
         # TODO: handle "private" metadata, to have non public access for node
         # TODO: node access/publish model should be customisable
-        comments = ticket_data.get('comments', [])
+        comments = ticket_data.get("comments", [])
         service = yield self._m.getCommentsService(client)
-        node = self._m.getCommentsNode(session['root_node'] + u'_' + ticket_data['id'])
-        node_options = {self._p.OPT_ACCESS_MODEL: self._p.ACCESS_OPEN,
-                        self._p.OPT_PERSIST_ITEMS: 1,
-                        self._p.OPT_MAX_ITEMS: -1,
-                        self._p.OPT_DELIVER_PAYLOADS: 1,
-                        self._p.OPT_SEND_ITEM_SUBSCRIBE: 1,
-                        self._p.OPT_PUBLISH_MODEL: self._p.ACCESS_OPEN,
-                        }
+        node = self._m.getCommentsNode(session["root_node"] + u"_" + ticket_data["id"])
+        node_options = {
+            self._p.OPT_ACCESS_MODEL: self._p.ACCESS_OPEN,
+            self._p.OPT_PERSIST_ITEMS: 1,
+            self._p.OPT_MAX_ITEMS: -1,
+            self._p.OPT_DELIVER_PAYLOADS: 1,
+            self._p.OPT_SEND_ITEM_SUBSCRIBE: 1,
+            self._p.OPT_PUBLISH_MODEL: self._p.ACCESS_OPEN,
+        }
         yield self._p.createIfNewNode(client, service, node, options=node_options)
-        ticket_data['comments_uri'] = uri.buildXMPPUri(u'pubsub', subtype='microblog', path=service.full(), node=node)
+        ticket_data["comments_uri"] = uri.buildXMPPUri(
+            u"pubsub", subtype="microblog", path=service.full(), node=node
+        )
         for comment in comments:
-            if 'updated' not in comment and 'published' in comment:
+            if "updated" not in comment and "published" in comment:
                 # we don't want an automatic update date
-                comment['updated'] = comment['published']
+                comment["updated"] = comment["published"]
             yield self._m.send(client, comment, service, node)
 
     def publishItem(self, client, ticket_data, service, node, session):
         if node is None:
             node = NS_TICKETS
-        id_ = ticket_data.pop('id', None)
-        log.debug(u"uploading item [{id}]: {title}".format(id=id_, title=ticket_data.get('title','')))
-        return self._s.sendDataFormItem(client, service, node, ticket_data, session['schema'], id_)
+        id_ = ticket_data.pop("id", None)
+        log.debug(
+            u"uploading item [{id}]: {title}".format(
+                id=id_, title=ticket_data.get("title", "")
+            )
+        )
+        return self._s.sendDataFormItem(
+            client, service, node, ticket_data, session["schema"], id_
+        )
 
     def itemFilters(self, client, ticket_data, session, options):
         mapping = options.get(OPT_MAPPING)
         if mapping is not None:
             if not isinstance(mapping, dict):
-                raise exceptions.DataError(_(u'mapping option must be a dictionary'))
+                raise exceptions.DataError(_(u"mapping option must be a dictionary"))
 
             for source, dest in mapping.iteritems():
                 if not isinstance(source, unicode) or not isinstance(dest, unicode):
-                    raise exceptions.DataError(_(u'keys and values of mapping must be sources and destinations ticket fields'))
+                    raise exceptions.DataError(
+                        _(
+                            u"keys and values of mapping must be sources and destinations ticket fields"
+                        )
+                    )
                 if source in ticket_data:
                     value = ticket_data.pop(source)
                     if dest in FIELDS_LIST:
@@ -163,6 +185,6 @@
                         values.append(value)
                     else:
                         if dest in ticket_data:
-                            ticket_data[dest] = ticket_data[dest] + u'\n' + value
+                            ticket_data[dest] = ticket_data[dest] + u"\n" + value
                         else:
                             ticket_data[dest] = value