source: WAeUP_SRP/trunk/Widgets.py @ 537

Last change on this file since 537 was 537, checked in by joachim, 18 years ago

added WAeUPImage Widget
  • Property svn:keywords set to Id
File size: 13.6 KB
Line 
1#-*- mode: python; mode: fold -*-
2
3from cgi import escape
4from types import *
5from Globals import InitializeClass
6##from Products.CPSSchemas.Widget import CPSWidgetType
7from Products.CMFCore.utils import getToolByName
8from Products.CPSSchemas.BasicWidgets import CPSWidget, CPSStringWidget, CPSEmailWidget,CPSImageWidget
9from Products.CPSSchemas.BasicWidgets import renderHtmlTag,CPSSelectWidget, CPSStringWidget
10from Products.CPSSchemas.ExtendedWidgets import CPSDateTimeWidget
11from Products.CPSSchemas.Widget import widgetRegistry
12##from Products.CPSSchemas.WidgetTypesTool import WidgetTypeRegistry
13from DateTime.DateTime import DateTime
14from AccessControl import getSecurityManager
15from Products.WAeUP_SRP.Students import getStudentByRegNo
16from re import compile
17
18from zLOG import LOG, DEBUG
19
20class CPSSelectWidgetForRecord(CPSSelectWidget): ###(
21    """Select widget. with record names"""
22    meta_type = 'Select Widget for Records'
23
24    field_types = ('CPS String Field',)
25    field_inits = ({'is_searchabletext': 1,},)
26
27    _properties = CPSSelectWidget._properties + (
28        {'id': 'record_id', 'type': 'string', 'mode': 'w',
29         'label': 'Record Id', 'is_required' : 1},
30        )
31
32    def render(self, mode, datastructure, **kw):
33        """Render in mode from datastructure."""
34        value = datastructure[self.getWidgetId()]
35        vocabulary = self._getVocabulary(datastructure)
36        portal = getToolByName(self, 'portal_url').getPortalObject()
37        cpsmcat = portal.translation_service
38        if mode == 'view':
39            if self.translated:
40                return escape(cpsmcat(vocabulary.getMsgid(value, value)).encode('ISO-8859-15', 'ignore'))
41            else:
42                return escape(vocabulary.get(value, value))
43        elif mode == 'edit':
44            html_widget_id = self.getHtmlWidgetId()
45            res = renderHtmlTag('select',
46                                name='%s.%s:records' % (self.record_id,html_widget_id),
47                                id=html_widget_id)
48            in_selection = 0
49            for k, v in vocabulary.items():
50                if self.translated:
51                    kw = {'value': k,
52                          'contents': cpsmcat(vocabulary.getMsgid(k, k)).encode('ISO-8859-15', 'ignore')
53                          }
54                else:
55                    kw = {'value': k, 'contents': v}
56                if value == k:
57                    kw['selected'] = 'selected'
58                    in_selection = 1
59                res += renderHtmlTag('option', **kw)
60            if value and not in_selection:
61                kw = {'value': value, 'contents': 'invalid: '+ str(value),
62                      'selected': 'selected'}
63                res += renderHtmlTag('option', **kw)
64            res += '</select>'
65            return res
66        raise RuntimeError('unknown mode %s' % mode)
67
68InitializeClass(CPSSelectWidgetForRecord)
69
70widgetRegistry.register(CPSSelectWidgetForRecord)
71
72###)
73
74class CPSStringWidgetForRecord(CPSStringWidget): ###(
75    """String widget."""
76    meta_type = 'String Widget For Record'
77
78    field_types = ('CPS String Field',)
79    field_inits = ({'is_searchabletext': 1,},)
80    _properties = CPSStringWidget._properties + (
81        {'id': 'record_id', 'type': 'string', 'mode': 'w',
82         'label': 'Record Id', 'is_required' : 1},
83        )
84
85    def render(self, mode, datastructure, **kw):
86        """Render in mode from datastructure."""
87        value = datastructure[self.getWidgetId()]
88        if mode == 'view':
89            return escape(value)
90        elif mode == 'edit':
91            # XXX TODO should use an other name than kw !
92            # XXX change this everywhere
93            html_widget_id = self.getHtmlWidgetId()
94            kw = {'type': 'text',
95                  'id'  : html_widget_id,
96                  'name': '%s.%s:records' % (self.record_id,html_widget_id),
97                  'value': escape(value),
98                  'size': self.display_width,
99                  }
100            if self.size_max:
101                kw['maxlength'] = self.size_max
102            return renderHtmlTag('input', **kw)
103        raise RuntimeError('unknown mode %s' % mode)
104
105InitializeClass(CPSStringWidgetForRecord)
106
107widgetRegistry.register(CPSStringWidgetForRecord)
108
109###)
110
111class CertificateCourseIdWidget(CPSStringWidget): ###(
112    """ CertificateCourseId Widget"""
113    meta_type = "CertificateCourseId Widget"
114
115    def validate(self, datastructure, **kw):
116        """Validate datastructure and update datamodel."""
117
118        valid = CPSStringWidget.validate(self, datastructure, **kw)
119        if not valid:
120            return 0
121        else:
122            widget_id = self.getWidgetId()
123            value = datastructure[widget_id].upper()
124            err = 0
125            c_ids = [c.id for c in self.portal_catalog({'meta_type': "Course"})]
126            if hasattr(self.aq_parent,value):
127                err = 'Course %s already exists' % (value)
128            elif value not in c_ids:
129                err = 'Course %s does not exist' % (value)
130            if err:
131                datastructure.setError(widget_id, err)
132            else:
133                datamodel = datastructure.getDataModel()
134                datamodel[self.fields[0]] = value
135
136            return not err
137
138InitializeClass(CertificateCourseIdWidget)
139
140widgetRegistry.register(CertificateCourseIdWidget)
141
142
143###)
144
145class WAeUPIdWidget(CPSStringWidget): ###(
146    """ WAeUPId Widget"""
147    meta_type = "WAeUPId Widget"
148
149    def validate(self, datastructure, **kw):
150        """Validate datastructure and update datamodel."""
151
152        valid = CPSStringWidget.validate(self, datastructure, **kw)
153        id_pat_str = r"\S"
154        inv_id_pat = compile(r"^%s$" % id_pat_str)
155        if not valid:
156            return 0
157        else:
158            widget_id = self.getWidgetId()
159            value = datastructure[widget_id].upper()
160            err = 0
161            if len(value.split()) > 1:
162                err = 'Invalid Id (Id contains space(s)'
163            elif hasattr(self.aq_parent,value):
164                err = 'An object with the Id %s already exists in this context' % (value)
165            if err:
166                datastructure.setError(widget_id, err)
167            else:
168                datamodel = datastructure.getDataModel()
169                datamodel[self.fields[0]] = value
170
171            return not err
172
173InitializeClass(WAeUPIdWidget)
174
175widgetRegistry.register(WAeUPIdWidget)
176
177
178###)
179
180class StudentIdWidget(CPSStringWidget): ###(
181    """ StudentId Widget"""
182    meta_type = "StudentId Widget"
183    digits = 8
184    digits_str = "N"*digits
185    letters = 2
186    letters_str = "L"*letters
187
188    def validate(self, datastructure, **kw):
189        """Validate datastructure and update datamodel."""
190
191        valid = CPSStringWidget.validate(self, datastructure, **kw)
192        if not valid:
193            return 0
194        else:
195            widget_id = self.getWidgetId()
196            value = datastructure[widget_id]
197            err = 0
198            if not (len(value) == self.digits + self.letters and value[:self.digits].isdigit() and value[self.digits:].isalpha()):
199                err = 'Invalid Registration Number in the format: %s%s with N = Number, L = Letter' % (self.digits_str,self.letters_str)
200            if err:
201                datastructure.setError(widget_id, err)
202            else:
203                datamodel = datastructure.getDataModel()
204                datamodel[self.fields[0]] = value
205
206            return not err
207
208InitializeClass(StudentIdWidget)
209widgetRegistry.register(StudentIdWidget)
210
211###)
212
213##def getStudentByRegNo(self,reg_no):
214##    search = self.portal_catalog({'meta_type': 'StudentApplication',
215##                                  'jamb_reg_no': reg_no,
216##                                  })
217##    if len(search) < 1:
218##        return None
219##    return search[0].getObject().
220                   
221class JambRegNoWidget(CPSStringWidget): ###(
222    """ JambRegNo Widget"""
223    meta_type = "JambRegNo Widget"
224    digits = 8
225    digits_str = "N"*digits
226    letters = 2
227    letters_str = "L"*letters
228
229    def validate(self, datastructure, **kw):
230        """Validate datastructure and update datamodel."""
231
232        valid = CPSStringWidget.validate(self, datastructure, **kw)
233        if not valid:
234            return 0
235        else:
236            widget_id = self.getWidgetId()
237            value = datastructure[widget_id]
238            err = 0
239            if not (len(value) == self.digits + self.letters and value[:self.digits].isdigit() and value[self.digits:].isalpha()):
240                err = 'Invalid Registration Number in the format: %s%s with N = Number, L = Letter' % (self.digits_str,self.letters_str)
241            else:
242                s = getStudentByRegNo(self,value)
243                if s is None:
244                    err = 'No JAMB data for %s' % (value)
245            if err:
246                datastructure.setError(widget_id, err)
247            else:
248                datamodel = datastructure.getDataModel()
249                datamodel[self.fields[0]] = value
250
251            return not err
252
253InitializeClass(JambRegNoWidget)
254
255widgetRegistry.register(JambRegNoWidget)
256###)
257
258class ScratchCardPin: ###(
259    """the ScratchCardPin"""
260    def __init__(self,prefix,batch_no,number):
261        self.p = prefix
262        self.b = batch_no
263        self.n = number
264
265    def __str__(self):
266        return "%s-%s-%s" % (self.p,self.b,self.n)
267###)
268
269class ScratchcardPinWidget(CPSStringWidget): ###(
270    """ ScratchcardPin Widget"""
271    meta_type = "Scratchcard Pin Widget"
272    _properties = CPSWidget._properties + (
273        {'id': 'prefix', 'type': 'string', 'mode': 'w',
274         'label': 'Prefix'},
275         {'id': 'reference', 'type': 'string', 'mode': 'w',
276         'label': 'Reference Field'},
277
278        )
279    prefix = ''
280    reference = ''
281   
282    def prepare(self, datastructure, **kw): ###(
283        """Prepare datastructure from datamodel."""
284        datamodel = datastructure.getDataModel()
285        v = datamodel[self.fields[0]]
286        widget_id = self.getWidgetId()
287        if v and type(v) is StringType:
288            p,b,n = v.split('-')
289            v = ScratchCardPin(v,b,n)
290        if v:
291            b = '%s' % v.b
292            n = '%s' % v.n
293        else:
294            b = n = ''
295        datastructure[widget_id] = v
296        datastructure[widget_id+'_b'] = b
297        datastructure[widget_id+'_n'] = n
298       
299###)
300
301    def validate(self, datastructure, **kw):
302        """Validate datastructure and update datamodel."""
303        widget_id = self.getWidgetId()
304        v = datastructure[widget_id]
305        err = 0
306        p = self.prefix
307        b = datastructure[widget_id+'_b'].strip()
308        n = datastructure[widget_id+'_n'].strip()
309        pins = self.portal_pins
310        pin = "%(p)s%(b)s%(n)s" % vars()
311        ref = datastructure[self.reference]
312        #import pdb;pdb.set_trace()
313        ok = pins.searchAndSetRecord(pin,ref)
314        while 1:
315            if ok == -1:
316                err = 'invalid Pin'
317                break
318            if ok == 0:
319                err = 'Pin already used'
320                break
321            if ok >= 1:
322                student = getStudentByRegNo(self,ref)
323                if student is None:
324                    err = "Student not Found in validate SC"
325                    break
326                s_id = student.getId()
327                if ok == 2:
328                    break
329                student.getContent().makeStudentMember(s_id,password=pin[4:])
330            break
331        if err:
332            datastructure.setError(widget_id, err)
333        else:
334            datamodel = datastructure.getDataModel()
335            datamodel[self.fields[0]] = ScratchCardPin(self.prefix,b,n)
336            datastructure[widget_id] = ScratchCardPin(self.prefix,b,n)
337            datastructure[widget_id+'_b'] = b
338            datastructure[widget_id+'_n'] = n
339            datastructure['s_id'] = s_id
340        #import pdb;pdb.set_trace()
341        return not err
342
343    def render(self, mode, datastructure, **kw): ###(
344        """Render in mode from datastructure."""
345        render_method = 'widget_scratch_card_pin_render'
346        meth = getattr(self, render_method, None)
347        if meth is None:
348            raise RuntimeError("Unknown Render Method %s for widget type %s"
349                               % (render_method, self.getId()))
350
351        # XXX AT: datastructure has to be set again here, in case we're in edit
352        # or create mode, because a default value has to be provided.
353        #import pdb;pdb.set_trace()
354        datamodel = datastructure.getDataModel()
355        v = datamodel[self.fields[0]]
356        if v and type(v) is StringType:
357            p,b,n = v.split('-')
358            v = ScratchCardPin(p,b,n)
359        if v:
360            b = '%s' % v.b
361            n = '%s' % v.n
362        else:
363            b = n = ''
364        if mode in ['edit', 'create']:
365            widget_id = self.getWidgetId()
366            datastructure[widget_id] = v
367            datastructure[widget_id+'_b'] = b
368            datastructure[widget_id+'_n'] = n
369        return meth(mode=mode,
370                    datastructure=datastructure,
371                    prefix=self.prefix,
372                    )
373    ###)
374
375
376InitializeClass(ScratchcardPinWidget)
377widgetRegistry.register(ScratchcardPinWidget)
378
379
380###)
381
382class WAeUPImageWidget(CPSImageWidget):
383    """Photo widget."""
384    meta_type = 'WAeUP Image Widget'
385
386    def render(self, mode, datastructure, **kw):
387        render_method = 'widget_waeup_image_render'
388        meth = getattr(self, render_method, None)
389        if meth is None:
390            raise RuntimeError("Unknown Render Method %s for widget type %s"
391                               % (render_method, self.getId()))
392        img_info = self.getImageInfo(datastructure)
393        return meth(mode=mode, datastructure=datastructure, **img_info)
394
395
396
397InitializeClass(WAeUPImageWidget)
398
399widgetRegistry.register(WAeUPImageWidget)
400
401
402###########
403
Note: See TracBrowser for help on using the repository browser.