Changeset 2365 in products for quintagroup.canonicalpath/trunk/quintagroup/canonicalpath/upgrades.py

Timestamp:

May 20, 2010 4:52:54 PM (14 years ago)

Author:

mylan

Message:

Add logging to upgrades.CanonicalConverter?

File:

• quintagroup.canonicalpath/trunk/quintagroup/canonicalpath/upgrades.py (modified) (6 diffs)

quintagroup.canonicalpath/trunk/quintagroup/canonicalpath/upgrades.py

@@ -1 @@
-import logging
-from zope.component import queryAdapter
+from logging import NOTSET, DEBUG, INFO, ERROR
+from logging import Logger, StreamHandler, Formatter
+from StringIO import StringIO
+from zope.component import getAdapter
 #from zope.component import queryMultiAdapter
 from Acquisition import aq_base, aq_inner
@@ -13 +15 @@
     """Convert canonical link to canonical path and vice versa."""
 
-    def __init__(self, portal_url, logger_name="quintagroup.canonicalpath"):
+    def __init__(self, portal_url):
         """Create instanse of convertor.
            - *portal_url* (string), add in the front of canonical path property
@@ -19 +21 @@
            - *logger_name* - name of the logger
         """
-        self.initLogger(logger_name)
+        self._initLogger()
         self.portal_url = portal_url
 
-    def initLogger(self, lname):
-        self.logger = logging.getLogger(lname)
+    def getLogs(self):
+        self._inout.flush()
+        return self._inout.getvalue()
+
+    def cleanupLogs(self):
+        self._inout = StringIO()
 
     def convertLinkToPath(self, obj):
@@ -42 +48 @@
            Log results in logger.
         """
-        src = queryAdapter(obj, src_iface)
-        dst = queryAdapter(obj, dst_iface)
-        # XXX: Check is this correct work XXX
-        obj = aq_base(aq_inner(obj))
-        # XXX
-        if src is not None \
-           and dst is not None:
-            msg = "Migrate %s into %s for %s object: " \
-                   % (src_iface, dst_iface, obj.absolute_url())
-            try:
-                converter(src, dst)
-            except Exception, e:
-                lev = logging.ERROR
-                msg += "WITH ERROR: %s" % str(e)
-            else:
-                lev = logging.INFO
-                msg += "SUCCESSFUL"
-
-            self.logger.log(lev, msg)
+        msg = "Migrate %s into %s for %s object: " \
+               % (src_iface.__name__, dst_iface.__name__, obj.absolute_url())
+        try:
+            src = getAdapter(obj, src_iface)
+            dst = getAdapter(obj, dst_iface)
+            # XXX: Check is this correct work XXX
+            obj = aq_base(aq_inner(obj))
+            # XXX
+            converter(src, dst)
+        except Exception:
+            import sys
+            et, em, etr = map(str, sys.exc_info())
+            lev = ERROR
+            msg += "ERROR: %s: %s" % (et, em)
+        else:
+            lev = INFO
+            msg += "SUCCESS"
+        self._logger.log(lev, msg)
             
-            return lev == logging.INFO and True or False
+        return lev == INFO and True or False
     
     def _convertP2L(self, src, dst):
@@ -69 +74 @@
         cpath = cpath.startswith('/') and cpath or '/' + cpath
         dst.canonical_link = self.portal_url + cpath
-        del dst.canonical_path
+        del src.canonical_path
 
     def _convertL2P(self, src, dst):
@@ -76 +81 @@
             "Convertion from canonical link to canonical path not implemented")
 
+    def _initLogger(self):
+        self._inout = StringIO()
+        handler = StreamHandler(self._inout)
+        handler.setLevel(DEBUG)
+        formatter = Formatter(fmt="[%(asctime)s]: %(message)s",
+                                      datefmt="%H:%M:%S")
+        handler.setFormatter(formatter)
+        self._logger = Logger("quintagroup.canonicalpath", NOTSET)
+        self._logger.addHandler(handler)