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

Timestamp:

20 Jun 2011, 02:37:06 (13 years ago)

Author:

uli

Message:

Reorganize tests. For what is over, make most of the remaining accesscode stuff work again. Still lots to do.

File:

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

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

@@ -61 +61 @@
         return IWorkflowState(self).getState() == USED
 
-class AccessCodeBatch(grok.Model):
+class AccessCodeBatch(grok.Container):
     """A batch of access codes.
     """
@@ -77 +77 @@
         self.invalidated_num = 0
         self.disabled_num = 0
-        self._entries = list()
-        self._acids = OIBTree()
-        self._createEntries()
 
     def _createEntries(self):
@@ -98 +95 @@
             for x in range(10):
                 pin += str(random().randint(0, 9))
-            if not '%s-%s-%s' % (self.prefix, self.num, pin) in self._acids:
+            if not '%s-%s-%s' % (self.prefix, self.num, pin) in self.keys():
                 curr += 1
                 yield pin
@@ -116 +113 @@
         """Get all entries of this batch as generator.
         """
-        for x in self._entries:
+        for x in self.values():
             yield x
 
@@ -122 +119 @@
         """Get the AccessCode with ID ``ac_id`` or ``KeyError``.
         """
-        return self._entries[self._acids[ac_id]]
+        return self[ac_id]
 
     def addAccessCode(self, num, pin):
@@ -129 +126 @@
         ac = AccessCode(num, pin)
         ac.__parent__ = self
-        self._entries.append(ac)
-        self._acids.update({ac.representation: num})
+        self[ac.representation] = ac
         return
 
@@ -136 +132 @@
         """Invalidate the AC with ID ``ac_id``.
         """
-        num = self._acids[ac_id]
         self.invalidated_num += 1
 
@@ -157 +152 @@
         ACs are left untouched.
         """
-        num = self._acids[ac_id]
-        self.disabled_num -= 1
 
     def createCSVLogFile(self):
@@ -179 +172 @@
         writer.writerow([self.prefix, str(self.num), "%0.2f" % self.cost])
 
-        for value in self._entries:
+        for value in self.values():
             writer.writerow(
                 [str(value.batch_serial), str(value.representation)]
@@ -205 +198 @@
         writer.writerow([self.prefix, '%0.2f' % self.cost, str(self.num),
                          str(self.entry_num)])
-        for value in self._entries:
+        for value in self.values():
             writer.writerow([
                     self.prefix, value.batch_serial, value.representation,
@@ -223 +216 @@
                 return []
 
+@grok.subscribe(IAccessCodeBatch, grok.IObjectAddedEvent)
+def handle_local_role_changed(batch, event):
+    # A (maybe dirty?) workaround to make batch containers work
+    # without self-maintained acids: as batches should contain their
+    # set of data immediately after creation, but we cannot add
+    # subobjects as long as the batch was not added already to the
+    # ZODB, we trigger the item creation for the time after the batch
+    # was added to the ZODB.
+    batch._createEntries()
+    return
+
+
 class AccessCodeBatchContainer(grok.Container):
     grok.implements(IAccessCodeBatchContainer)
@@ -245 +250 @@
         self._p_changed = True
 
-    def createBatch(self, creation_date, creator, batch_prefix, cost,
+    def createBatch(self, creation_date, creator, prefix, cost,
                     entry_num):
         """Create and add a batch.
         """
-        batch_num = self.getNum(batch_prefix)
-        batch = AccessCodeBatch(creation_date, creator, batch_prefix,
+        batch_num = self.getNum(prefix)
+        batch = AccessCodeBatch(creation_date, creator, prefix,
                                 cost, entry_num, num=batch_num)
         self.addBatch(batch)
@@ -284 +289 @@
         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)
         batch = AccessCodeBatch(
-            datetime.now(), creator, entry['serial'], cost, 0, num=num)
+            datetime.now(), creator, entry['serial'], cost, 0, num=0)
         num_entries = 0
+        self[batch_name] = batch
         for row in reader:
             pin = row['ac']
@@ -294 +302 @@
             num_entries += 1
         batch.entry_num = num_entries
-        self[batch_name] = batch
+
         batch.createCSVLogFile()
         return