Changeset 5132 for main/waeup.sirp/trunk/src/waeup/sirp/accesscodes/accesscodes.py

Timestamp:

5 Apr 2010, 18:16:56 (15 years ago)

Author:

uli

Message:

Support reimport of 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

@@ -198 +198 @@
     grok.implements(IAccessCodeBatchContainer)
 
+    def _getStoragePath(self):
+        """Get the directory, where batch import files are stored.
+        """
+        site = grok.getSite()
+        storagepath = site['datacenter'].storage
+        ac_storage = os.path.join(storagepath, 'accesscodes')
+        import_path = os.path.join(ac_storage, 'imports')
+        if not os.path.exists(import_path):
+            os.mkdir(import_path)
+        return import_path
+
     def addBatch(self, batch):
         """Add a batch.
@@ -224 +235 @@
         return num
 
+    def getImportFiles(self):
+        """Return a generator with basenames of available import files.
+        """
+        path = self._getStoragePath()
+        for filename in sorted(os.listdir(path)):
+            yield filename
+    
+    def reimport(self, filename, creator=u'UNKNOWN'):
+        """Reimport a batch given in CSV file.
+
+        CSV file must be of format as generated by createCSVLogFile
+        method.
+        """
+        path = os.path.join(self._getStoragePath(), filename)
+        reader = csv.DictReader(open(path, 'rb'), quoting=csv.QUOTE_ALL)
+        entry = reader.next()
+        cost = float(entry['cost'])
+        num = int(entry['ac'])
+        batch_name = '%s-%s' % (entry['serial'], num)
+        if batch_name in self.keys():
+            raise KeyError('Batch already exists: %s' % batch_name)
+        batch = AccessCodeBatch(
+            datetime.now(), creator, entry['serial'], cost, 0, num=num)
+        num_entries = 0
+        for row in reader:
+            pin = row['ac']
+            serial = int(row['serial'])
+            rand_num = pin.rsplit('-', 1)[-1]
+            batch.addAccessCode(serial, rand_num)
+            num_entries += 1
+        batch.entry_num = num_entries
+        self[batch_name] = batch
+        batch.createCSVLogFile()
+        return
+    
 class AccessCodePlugin(grok.GlobalUtility):
     grok.name('accesscodes')