source: WAeUP_SRP/trunk/WAeUPTables.py @ 1391

Last change on this file since 1391 was 1306, checked in by Henrik Bettermann, 18 years ago

iteration over non sequence fixed
  • Property svn:keywords set to Id
File size: 8.1 KB
Line 
1#-*- mode: python; mode: fold -*-
2# (C) Copyright 2005 AixtraWare <http://aixtraware.de>
3# Author: Joachim Schmitz <js@aixtraware.de>
4#
5# This program is free software; you can redistribute it and/or modify
6# it under the terms of the GNU General Public License version 2 as published
7# by the Free Software Foundation.
8#
9# This program is distributed in the hope that it will be useful,
10# but WITHOUT ANY WARRANTY; without even the implied warranty of
11# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12# GNU General Public License for more details.
13#
14# You should have received a copy of the GNU General Public License
15# along with this program; if not, write to the Free Software
16# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
17# 02111-1307, USA.
18#
19# $Id: WAeUPTables.py 1306 2007-01-16 06:33:04Z henrik $
20
21from zope.interface import implements
22from Globals import InitializeClass
23from Products.ZCatalog.ZCatalog import ZCatalog
24from AccessControl import ClassSecurityInfo
25from Products.CMFCore.permissions import ModifyPortalContent
26
27import DateTime
28import csv,re
29import logging
30import Globals
31p_home = Globals.package_home(globals())
32i_home = Globals.INSTANCE_HOME
33
34from interfaces import IWAeUPTable
35
36class AttributeHolder(object):
37    pass
38
39def dict2ob(dict):
40    ob = AttributeHolder()
41    for key, value in dict.items():
42        setattr(ob, key, value)
43    return ob
44
45
46class WAeUPTable(ZCatalog): ###(
47
48    implements(IWAeUPTable)
49    security = ClassSecurityInfo()
50
51    def addRecord(self, **data):
52        # The uid is the same as "bed".
53        uid = data[self.key]
54        res = self.searchResults({"%s" % self.key : uid})
55        if len(res) > 0:
56            raise ValueError("More than one record with uid %s" % uid)
57        self.catalog_object(dict2ob(data), uid=uid)
58        return uid
59
60    def deleteRecord(self, uid):
61        #import pdb;pdb.set_trace()
62        self.uncatalog_object(uid)
63
64    def searchAndSetRecord(self, **data):
65        raise NotImplemented
66
67    def modifyRecord(self, **data):
68        #records = self.searchResults(uid=uid)
69        uid = data[self.key]
70        records = self.searchResults({"%s" % self.key : uid})
71        if len(records) > 1:
72            # Can not happen, but anyway...
73            raise ValueError("More than one record with uid %s" % uid)
74        if len(records) == 0:
75            raise KeyError("No record for uid %s" % uid)
76        record = records[0]
77        record_data = {}
78        for field in self.schema() + self.indexes():
79            record_data[field] = getattr(record, field)
80        record_data = {}
81        for field in self.schema() + self.indexes():
82            record_data[field] = getattr(record, field)
83        # Add the updated data:
84        record_data.update(data)
85        self.catalog_object(dict2ob(record_data), uid)
86
87    def reindexIndex(self, name, REQUEST,pghandler=None):
88        if isinstance(name, str):
89            name = (name,)
90        paths = self._catalog.uids.items()
91        i = 0
92        #import pdb;pdb.set_trace()
93        for p,rid in paths:
94            i += 1
95            metadata = self.getMetadataForRID(rid)
96            record_data = {}
97            for field in name:
98                record_data[field] = metadata.get(field)
99            uid = metadata.get(self.key)
100            self.catalog_object(dict2ob(record_data), uid, idxs=name,
101                                update_metadata=0)
102
103    security.declareProtected(ModifyPortalContent,"exportAllRecords")
104    def exportAllRecords(self):
105        "export a WAeUPTable"
106        #import pdb;pdb.set_trace()
107        fields = [field for field in self.schema()]
108        format = ','.join(['"%%(%s)s"' % fn for fn in fields])
109        csv = []
110        csv.append(','.join(['"%s"' % fn for fn in fields]))
111        for uid in self._catalog.uids:
112            records = self.searchResults({"%s" % self.key : uid})
113            if len(records) > 1:
114                # Can not happen, but anyway...
115                raise ValueError("More than one record with uid %s" % uid)
116            if len(records) == 0:
117                raise KeyError("No record for uid %s" % uid)
118            rec = records[0]
119            csv.append(format % rec)
120        current = DateTime.DateTime().strftime("%d-%m-%y_%H_%M_%S")
121        open("%s/import/%s-%s.csv" % (i_home,self.getId(),current),"w+").write('\n'.join(csv))
122
123###)
124
125class AccommodationTable(WAeUPTable): ###(
126
127    meta_type = 'WAeUP Accommodation Tool'
128    name = "accommodation"
129    key = "bed"
130    def __init__(self):
131        WAeUPTable.__init__(self, 'portal_accommodation')
132
133    def searchAndReserveBed(self, student_id,bed_type):
134        records = self.searchResults({'student' : student_id})
135        if len(records) > 0:
136            return -1,"Student with Id %s already booked bed %s." % (student_id,records[0].bed)
137
138        records = [r for r in self.searchResults({'bed_type' : bed_type}) if not r.student]
139        #import pdb;pdb.set_trace()
140        if len(records) == 0:
141            return -2,"No bed available"
142        rec = records[0]
143        self.modifyRecord(bed=rec.bed,student=student_id)
144        s_logger = logging.getLogger('Accommodation.SearchAndReserveBed')
145        s_logger.info("%s reserved bed %s" % (student_id,rec.bed))
146        return 1,rec.bed
147
148
149InitializeClass(AccommodationTable)
150
151###)
152
153class PinTable(WAeUPTable): ###(
154    from ZODB.POSException import ConflictError
155    meta_type = 'WAeUP Pin Tool'
156    name = "pins"
157    key = 'pin'
158    def __init__(self):
159        WAeUPTable.__init__(self, 'portal_pins')
160
161
162    def searchAndSetRecord(self, uid, student_id,prefix):
163        #records = self.searchResults(uid=uid)
164        records = self.searchResults(student = student_id)
165        #import pdb;pdb.set_trace()
166        if len(records) > 0:
167            for r in records:
168                if r.pin != uid and r.prefix_batch.startswith(prefix):
169                    return -2
170        records = self.searchResults({"%s" % self.key : uid})
171        if len(records) > 1:
172            # Can not happen, but anyway...
173            raise ValueError("More than one record with uid %s" % uid)
174        if len(records) == 0:
175            return -1
176        record = records[0]
177        if record.student == "":
178            record_data = {}
179            for field in self.schema() + self.indexes():
180                record_data[field] = getattr(record, field)
181            # Add the updated data:
182            record_data['student'] = student_id
183            try:
184                self.catalog_object(dict2ob(record_data), uid)
185                return 1
186            except ConflictError:
187                return 2
188        if record.student.upper() != student_id.upper():
189            return 0
190        if record.student.upper() == student_id.upper():
191            return 2
192        return -3
193
194InitializeClass(PinTable)
195
196###)
197
198class PumeResultsTable(WAeUPTable): ###(
199
200    meta_type = 'WAeUP PumeResults Tool'
201    name = "pumeresults"
202    key = "jamb_reg_no"
203    def __init__(self):
204        WAeUPTable.__init__(self, 'portal_pumeresults')
205
206
207InitializeClass(PumeResultsTable)
208
209###)
210
211class StudentsCatalog(WAeUPTable): ###(
212
213    meta_type = 'WAeUP Students Catalog'
214    name = "students_catalog"
215    key = "id"
216    def __init__(self):
217        WAeUPTable.__init__(self, 'students_catalog')
218
219
220InitializeClass(StudentsCatalog)
221
222###)
223
224class CoursesCatalog(WAeUPTable): ###(
225
226    meta_type = 'WAeUP Courses Catalog'
227    name = "students_catalog"
228    key = "code"
229    def __init__(self):
230        WAeUPTable.__init__(self, 'courses_catalog')
231
232
233InitializeClass(CoursesCatalog)
234###)
235
236class ReturningImport(WAeUPTable): ###(
237
238    meta_type = 'Returning Import Table'
239    name = "returning_import"
240    key = "matric_no"
241    def __init__(self):
242        WAeUPTable.__init__(self, 'returning_import')
243
244
245InitializeClass(ReturningImport)
246###)
247
248class ResultsImport(WAeUPTable): ###(
249
250    meta_type = 'Results Import Table'
251    name = "results_import"
252    key = "key"
253    def __init__(self):
254        WAeUPTable.__init__(self, 'results_import')
255
256
257InitializeClass(ResultsImport)
258
259###)
260
261class PaymentsCatalog(WAeUPTable): ###(
262
263    meta_type = 'WAeUP Payments Catalog'
264    name = "students_catalog"
265    key = "id"
266    def __init__(self):
267        WAeUPTable.__init__(self, 'payments_catalog')
268
269
270InitializeClass(PaymentsCatalog)
271
272###)
273
274# BBB:
275AccomodationTable = AccommodationTable
Note: See TracBrowser for help on using the repository browser.