Changeset 12077 for main/waeup.ikoba/trunk

Timestamp:

28 Nov 2014, 07:29:14 (10 years ago)

Author:

Henrik Bettermann

Message:

Add product exporter with tests.

Location:

main/waeup.ikoba/trunk/src/waeup/ikoba/products

Files:

• export.py (modified) (2 diffs)
• tests/test_batching.py (modified) (2 diffs)
• tests/test_browser.py (modified) (4 diffs)
• tests/test_export.py (added)

main/waeup.ikoba/trunk/src/waeup/ikoba/products/export.py

@@ -19 +19 @@
 """
 import grok
+from zope.securitypolicy.interfaces import IPrincipalRoleMap
 from waeup.ikoba.interfaces import ICSVExporter
 from waeup.ikoba.interfaces import MessageFactory as _
@@ -24 +25 @@
 from waeup.ikoba.utils.helpers import iface_names
 from waeup.ikoba.products.interfaces import IProduct
+
+
+class ProductExporter(grok.GlobalUtility, ExporterBase):
+    """Exporter for products.
+    """
+    grok.implements(ICSVExporter)
+    grok.name('products')
+
+    #: Fieldnames considered by this exporter
+    fields = tuple(sorted(iface_names(IProduct))) + (
+        'users_with_local_roles',)
+
+    #: The title under which this exporter will be displayed
+    title = _(u'Products')
+
+    def mangle_value(self, value, name, context=None):
+        """Hook for mangling values in derived classes
+        """
+        if name == 'users_with_local_roles':
+            value = []
+            role_map = IPrincipalRoleMap(context)
+            for local_role, user_name, setting in role_map.getPrincipalsAndRoles():
+                value.append({'user_name':user_name,'local_role':local_role})
+        return super(ProductExporter, self).mangle_value(
+            value, name, context)
+
+    def export(self, products, filepath=None):
+        """Export `products`, an iterable, as CSV file.
+
+        If `filepath` is ``None``, a raw string with CSV data is returned.
+        """
+        writer, outfile = self.get_csv_writer(filepath)
+        for product in products:
+            self.write_item(product, writer)
+        return self.close_outfile(filepath, outfile)
+
+    def export_all(self, site, filepath=None):
+        """Export products in productscontainer into filepath as CSV data.
+
+        If `filepath` is ``None``, a raw string with CSV data is returned.
+        """
+        products = site.get('products', {})
+        return self.export(products.values(), filepath)

main/waeup.ikoba/trunk/src/waeup/ikoba/products/tests/test_batching.py

@@ -31 +31 @@
 from zope.catalog.interfaces import ICatalog
 from zope.interface.verify import verifyClass, verifyObject
+from zope.securitypolicy.interfaces import IPrincipalRoleManager
 
 from waeup.ikoba.app import Company
@@ -82 +83 @@
         self.app['products'][product.product_id] = self.product = product
         self.outfile = os.path.join(self.workdir, 'myoutput.csv')
+        role_manager = IPrincipalRoleManager(product)
+        role_manager.assignRoleToPrincipal(u'johnsrole', u'john')
         return
 

main/waeup.ikoba/trunk/src/waeup/ikoba/products/tests/test_browser.py

@@ -55 +55 @@
     return None
 
-class ProductssFullSetup(FunctionalTestCase):
+class ProductsFullSetup(FunctionalTestCase):
     # A test case that only contains a setup and teardown
     #
@@ -67 +67 @@
 
     def setUp(self):
-        super(ProductssFullSetup, self).setUp()
+        super(ProductsFullSetup, self).setUp()
 
         # Setup a sample site for each test
@@ -113 +113 @@
 
     def tearDown(self):
-        super(ProductssFullSetup, self).tearDown()
+        super(ProductsFullSetup, self).tearDown()
         clearSite()
         shutil.rmtree(self.dc_root)
 
-class ProductsContainerUITests(ProductssFullSetup):
+class ProductsContainerUITests(ProductsFullSetup):
     # Tests for ProductsContainer class views and pages
 
@@ -163 +163 @@
 
 
-class ProductsUITests(ProductssFullSetup):
+class ProductsUITests(ProductsFullSetup):
     # Tests for Products class views and pages