source: WAeUP_SRP/trunk_not_used/ScratchCards.py @ 2335

Last change on this file since 2335 was 2063, checked in by joachim, 17 years ago

set cost in pin table

  • Property svn:keywords set to Id
File size: 6.8 KB
Line 
1#-*- mode: python; mode: fold -*-
2# $Id: ScratchCards.py 2063 2007-07-27 06:55:39Z joachim $
3from Globals import InitializeClass
4from AccessControl import ClassSecurityInfo
5from AccessControl.SecurityManagement import newSecurityManager
6from zExceptions import BadRequest
7from Products.CMFCore.utils import UniqueObject, getToolByName
8from Products.CMFCore.permissions import View
9from Products.CMFCore.permissions import ModifyPortalContent
10from Products.CPSCore.CPSBase import CPSBase_adder, CPSBaseFolder
11#from Products.CPSCore.CPSBase import CPSBaseDocument as BaseDocument
12from Products.CPSDocument.CPSDocument import CPSDocument
13import DateTime
14import Globals
15import logging
16import os,csv,re
17import Globals
18MAX_TRANS = 1000
19p_home = Globals.package_home(globals())
20i_home = Globals.INSTANCE_HOME
21
22class ScratchCardBatchesFolder(CPSDocument): ###(
23    """
24    WAeUP Student container for the various student data
25    """
26    meta_type = 'ScratchCardBatchesFolder'
27    portal_type = meta_type
28    security = ClassSecurityInfo()
29
30    security.declareProtected(View,"Title")
31    def Title(self):
32        """compose title"""
33        return "Pin Batches"
34
35    security.declareProtected(ModifyPortalContent,"importPinBatch") ###(
36    def importPinBatch(self):
37        """import PinBatch"""
38        name = "PINS"
39        logger = logging.getLogger('Students.ScratchCardBatchesFolder.importPinBatch')
40        logger.info('Start loading from %s.csv' % name)
41        pins_section = self.portal_url.getPortalObject().campus.pins
42        p_pins = self.portal_pins
43        base_dir = "%s/import/pins" % (i_home)
44        files = os.listdir(base_dir)
45        for fn in files:
46            pins = csv.DictReader(open("%s/%s" % (base_dir,fn),"rb"))
47            n = 0
48            for pin in pins:
49                if n == 0:
50                    bid = "%(Serial)s_%(Pin)s" % pin
51                    pins_section.invokeFactory('ScratchCardBatch', bid)
52                    batch = getattr(pins_section,bid)
53                    dict = {}
54                    prefix = pin['Serial']
55                    dict['prefix'] = pin['Serial']
56                    batch_no = pin['Pin']
57                    dict['batch_no'] = int(batch_no)
58                    dict['cost'] = float(pin['Cost'])
59                    dict['sold_by'] = pin['Sold']
60                    #import pdb;pdb.set_trace()
61                    n += 1
62                    continue
63                puid = "".join(pin['Pin'].split('-'))
64                p_pins.addRecord(pin=puid,serial=n,student="",prefix_batch="%s%s" % (prefix, batch_no))
65                n += 1
66            dict['no_of_pins'] = n - 1
67            batch.getContent().edit(mapping=dict)
68            logger.info('%s with %d PINs added' % (bid,n-1))
69        return self.REQUEST.RESPONSE.redirect("%s" % self.REQUEST.get('URL1'))
70    ###)
71
72InitializeClass(ScratchCardBatchesFolder)
73
74def addScratchCardBatchesFolder(container, id, REQUEST=None, **kw):
75    """Add a Students personal data."""
76    ob = ScratchCardBatchesFolder(id, **kw)
77    return CPSBase_adder(container, ob, REQUEST=REQUEST)
78
79###)
80
81from Products.WAeUP_SRP.WAeUPTables import PinTable
82
83class ScratchCardBatch(CPSDocument): ###(
84    """
85    WAeUP Student container for the various student data
86    """
87    meta_type = 'ScratchCardBatch'
88    portal_type = meta_type
89    security = ClassSecurityInfo()
90
91    security.declareProtected(View,"Title")
92    def Title(self):
93        """compose title"""
94        doc = self.getContent()
95        return "Pin Batch %s BatchNo %d" % (doc.prefix, doc.batch_no)
96
97    security.declareProtected(ModifyPortalContent,"makePins") ###(
98    def makePins(self):
99        """generate Pins"""
100        batch = self.getContent()
101        nr = batch.no_of_pins
102        prefix = batch.prefix
103        sold = batch.sold_by
104        cost = batch.cost
105        import random
106        r = random
107        b_no = batch.batch_no
108        #import pdb;pdb.set_trace()
109        generated = []
110        generated.append('"Serial","Pin","Sold","Cost"' % vars())
111        generated.append('"%(prefix)s","%(b_no)d","%(sold)s","%(cost)f"' % vars())
112        pins = self.portal_pins
113        for i in range(1,nr+1):
114            ri = r.randint(1000000000,9999999999)
115            pin = "%s%d%d" % (prefix,b_no,ri)
116            while len(pins.searchResults({'pin': pin})) > 0:
117                ri = r.randint(1000000000,9999999999)
118                pin = "%s%d%d" % (prefix,b_no,ri)
119            pins.addRecord(pin=pin,
120                           serial=i,
121                           student="",
122                           cost = cost,
123                           prefix_batch="%s%d" % (prefix, b_no))
124            generated.append('"%(i)d","%(prefix)s-%(b_no)d-%(ri)d"' % vars())
125        print '\n'.join(generated)
126        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
127        open("%s/import/%s-%s-%s.csv" % (i_home,prefix,b_no,current),"w+").write('\n'.join(generated))
128    ###)
129
130
131    security.declareProtected(ModifyPortalContent,"getUnusedPins")
132    def getUnusedPins(self):
133        """return a list of unused Pins"""
134        max_pins = 10
135        #import pdb;pdb.set_trace()
136        pins_cat = self.portal_pins
137        doc = self.getContent()
138        unused = [{'pin': p.pin,
139              'serial': p.serial,
140              } for p in pins_cat.searchResults(prefix_batch = "%s%d"
141                                               % (doc.prefix,doc.batch_no),
142                                               ) if not p.student]
143        if len(unused) > max_pins:
144            return unused[-max_pins:]
145        return unused
146
147    security.declareProtected(View,"getUsedPins")
148    def getUsedPins(self):
149        """return a list of used Pins"""
150        #import pdb;pdb.set_trace()
151        pins_cat = self.portal_pins
152        doc = self.getContent()
153        used = [{'pin': p.pin,
154              'prefix_batch': p.prefix_batch,
155              'serial': p.serial,
156              'student': p.student,
157              } for p in pins_cat.searchResults(prefix_batch = "%s%d"
158                                               % (doc.prefix,doc.batch_no),
159                                               ) if p.student]
160        return used
161
162    security.declareProtected(View,"getNumberOfUsedPins")
163    def getNumberOfUsedPins(self):
164        """return the number of used Pins"""
165        #import pdb;pdb.set_trace()
166        pins_cat = self.portal_pins
167        doc = self.getContent()
168        used = [{'pin': p.pin,
169              'serial': p.serial,
170              'student': p.student,
171              } for p in pins_cat.searchResults(prefix_batch = "%s%d"
172                                               % (doc.prefix,doc.batch_no),
173                                               ) if p.student]
174        return len(used)
175
176InitializeClass(ScratchCardBatch)
177
178def addScratchCardBatch(container, id, REQUEST=None, **kw):
179    """Add a Students personal data."""
180    ob = ScratchCardBatch(id, **kw)
181    return CPSBase_adder(container, ob, REQUEST=REQUEST)
182
183###)
Note: See TracBrowser for help on using the repository browser.