Changeset 3415 for WAeUP_SRP/base

Timestamp:

3 Apr 2008, 16:12:00 (17 years ago)

Author:

joachim

Message:

implemt restorePins
file must be named pins_to_restore
goto batch
call archiveBatch?delete=1
call restorePins

File:

• WAeUP_SRP/base/ScratchCards.py (modified) (2 diffs)

WAeUP_SRP/base/ScratchCards.py

@@ -128 +128 @@
     ###)
 
+    security.declareProtected(ModifyPortalContent,"restorePins") ###(
+    def restorePins(self):
+        "restore Pins from file named pins_to_restore.csv"
+        pins = self.portal_pins
+        filename = "%s/export/pins_to_restore.csv" % (i_home)
+        logger = logging.getLogger('ScratchCards.restore') 
+        if not os.path.exists(filename):
+            logger.info('no file %s' % filename)
+            return
+        src = open(filename,"rb")
+        records = csv.DictReader(src)
+        cards = []
+        already_in = []
+        #import pdb;pdb.set_trace()
+        records.next()
+        for record in records:
+            d = {}
+            d['pin'] = record['Pin'].replace('-','')
+            if pins(pin=d['pin']):
+                logger.info("pin %(pin)s already in Pins" % d)
+                already_in += d['pin'],
+                continue
+            d['serial'] = record['Serial']
+            d['cost'] = record['Cost']
+            d['prefix_batch'] = "".join(record['Pin'].split('-')[:2])
+            cards += d,
+        if already_in:
+            logger.info("found %d existing pins in %s" % (len(already_in),filename))
+            return
+        anz = len(cards)
+        logger.info("start restoring %d pins from %s" % (anz,filename))
+        for card in cards:
+            pins.addRecord(**card)
+        logger.info("sucessfully restored %d pins from %s" % (anz,filename))
+        url = self.REQUEST.get('URL1')
+        return self.REQUEST.RESPONSE.redirect(url)
+    
     security.declareProtected(ModifyPortalContent,"archiveBatch") ###(
     def archiveBatch(self,delete=False):
@@ -170 +207 @@
             msg = "deleted %d pin from portal_pins" % len(pin_list)
             logger.info(msg)
+        url = self.REQUEST.get('URL1')
+        return self.REQUEST.RESPONSE.redirect(url)
     ###)