Changeset 5110 for main/waeup.sirp/trunk/src/waeup/sirp

Timestamp:

31 Mar 2010, 23:21:04 (15 years ago)

Author:

uli

Message:

Add CSV file creator for AC batches.

File:

• main/waeup.sirp/trunk/src/waeup/sirp/accesscodes/accesscodes.py (modified) (2 diffs)

main/waeup.sirp/trunk/src/waeup/sirp/accesscodes/accesscodes.py

@@ -1 +1 @@
 """Components to handle access codes.
 """
+import csv
 import grok
+import os
 from random import SystemRandom as random
 from waeup.sirp.interfaces import IWAeUPSIRPPluggable
@@ -81 +83 @@
         return result
 
+    def createCSVLogFile(self):
+        """Create a CSV file with data in batch.
+
+        Data will not contain invalidation date nor student ids.  File
+        will be created in ``accesscodes`` subdir of data center
+        storage path.
+
+        Returns name of created file.
+        """
+        site = grok.getSite()
+        storagepath = site['datacenter'].storage
+        ac_storage = os.path.join(storagepath, 'accesscodes')
+        if not os.path.exists(ac_storage):
+            os.mkdir(ac_storage)
+        date = self.creation_date.strftime('%Y_%m_%d_%H_%M_%S')
+        csv_path = os.path.join(
+            ac_storage, '%s-%s-%s-%s.csv' % (
+                self.prefix, self.num, date, self.creator)
+            )
+        writer = csv.writer(open(csv_path, 'w'), quoting=csv.QUOTE_ALL)
+        writer.writerow(['serial', 'ac', 'cost'])
+        writer.writerow([self.prefix, str(self.num), "%0.2f" % self.cost])
+        for key, value in self.items():
+            writer.writerow(
+                [str(value.batch_serial), str(value.representation)]
+                )
+        return os.path.basename(csv_path)
+
+    
 class AccessCodeBatchContainer(grok.Container):
     grok.implements(IAccessCodeBatchContainer)