Changeset 1755 in products

Timestamp:

Feb 16, 2010 4:04:07 PM (14 years ago)

Author:

mylan

Message:

Update utility, and interface for updating list of columns in catalog

Location:

quintagroup.catalogupdater/trunk/quintagroup/catalogupdater

Files:

• interfaces.py (modified) (1 diff)
• utility.py (modified) (5 diffs)

quintagroup.catalogupdater/trunk/quintagroup/catalogupdater/interfaces.py

@@ -3 +3 @@
 class ICatalogUpdater(Interface):
 
-    def updateMetadata4All(catalog, column):
-        """ Update only the column metadata in the catalog
-            for all records
+    def updateMetadata4All(catalog, columns):
+        """ Update metadata in the *catalog* for each column
+            in the *columns* list for all records.
+
+              * catalog - ZCatalog descendent catalog;
+              * columns - list of metadata names, or
+                          string with name of single
+                          metadata, which must be updated.
         """
 

quintagroup.catalogupdater/trunk/quintagroup/catalogupdater/utility.py

@@ -1 @@
-import logging
+import logging, types
 from zope.interface import implements
 from zope.component import queryMultiAdapter
@@ -22 +22 @@
     implements(ICatalogUpdater)
 
-    def validate(self, cat, col):
+    def validate(self, cat, cols):
         # Validate catalog and column name
+        AVAIL_COLTYPES = list(types.StringTypes) + [types.ListType, types.TupleType]
+
         _cat = getattr(cat, '_catalog', None)
-
         if _cat is None:
             raise AttributeError("%s - is not ZCatalog based catalog" % cat)
 
-        if not _cat.schema.has_key(col):
-            raise AttributeError("'%s' - not presented column in %s catalog " % (col, cat))
+        if not type(cols) in AVAIL_COLTYPES:
+            raise TypeError("'columns' parameter must be one of the following " \
+                "types: %s" % AVAIL_COLTYPES)
+        # Normalize columns
+        if type(cols) in types.StringTypes:
+            cols = [cols,]
+        # Check is every column present in the catalog
+        for col in cols:
+            if not _cat.schema.has_key(col):
+                raise AttributeError("'%s' - not presented column in %s catalog " % (col, cat))
+
+        return _cat, cols
 
     def getWrapedObject(self, obj, portal, catalog):
@@ -49 +60 @@
 
 
-    def updateMetadata4All(self, catalog, column):
+    def updateMetadata4All(self, catalog, columns):
         """ Look into appropriate method of ICatalogUpdate interface
         """
-        self.validate(catalog, column)
 
-        _catalog = catalog._catalog
+        _catalog, columns = self.validate(catalog, columns)
+
         portal = getToolByName(catalog, 'portal_url').getPortalObject()
         root = aq_parent(portal)
@@ -60 +71 @@
         data = _catalog.data
         schema = _catalog.schema
-        indx = schema[column]
         paths = _catalog.paths
 
@@ -71 +81 @@
                 obj = self.getWrapedObject(obj, portal, catalog)
             except:
-                LOG.error('update_metadata_column could not resolve '
+                LOG.error('updateMetadata4All could not resolve '
                           'an object from the uid %r.' % obj_uid)
                 continue
 
-            # calculate the column value
             mdlist = list(md)
-            attr=getattr(obj, column, MV)
-            if(attr is not MV and safe_callable(attr)): attr=attr()
+            for column in columns:
+                # calculate the column value
+                attr=getattr(obj, column, MV)
+                if(attr is not MV and safe_callable(attr)): attr=attr()
+                # Update metadata value
+                indx = schema[column]
+                mdlist[indx] = attr
 
-            # update metadata value
-            mdlist[indx] = attr
+            # Update catalog record
             data[rid] = tuple(mdlist)