]> git.llucax.com Git - software/sercom.git/blob - sercom/model.py
Bugfix. Corregir un par de MultipleJoin y herencia incorrecta.
[software/sercom.git] / sercom / model.py
1 # vim: set et sw=4 sts=4 encoding=utf-8 foldmethod=marker :
2
3 from datetime import datetime
4 from turbogears.database import PackageHub
5 from sqlobject import *
6 from sqlobject.sqlbuilder import *
7 from sqlobject.inheritance import InheritableSQLObject
8 from sqlobject.col import PickleValidator, UnicodeStringValidator
9 from turbogears import identity
10 from turbogears.identity import encrypt_password as encryptpw
11 from sercom.validators import params_to_list, ParseError
12 from formencode import Invalid
13
14 hub = PackageHub("sercom")
15 __connection__ = hub
16
17 __all__ = ('Curso', 'Usuario', 'Docente', 'Alumno', 'CasoDePrueba')
18
19 #{{{ Custom Columns
20
21 class TupleValidator(PickleValidator):
22     """
23     Validator for tuple types.  A tuple type is simply a pickle type
24     that validates that the represented type is a tuple.
25     """
26     def to_python(self, value, state):
27         value = super(TupleValidator, self).to_python(value, state)
28         if value is None:
29             return None
30         if isinstance(value, tuple):
31             return value
32         raise Invalid("expected a tuple in the TupleCol '%s', got %s %r instead" % \
33             (self.name, type(value), value), value, state)
34     def from_python(self, value, state):
35         if value is None:
36             return None
37         if not isinstance(value, tuple):
38             raise Invalid("expected a tuple in the TupleCol '%s', got %s %r instead" % \
39                 (self.name, type(value), value), value, state)
40         return super(TupleValidator, self).from_python(value, state)
41
42 class SOTupleCol(SOPickleCol):
43     def createValidators(self):
44         return [TupleValidator(name=self.name)] \
45             + super(SOPickleCol, self).createValidators()
46
47 class TupleCol(PickleCol):
48     baseClass = SOTupleCol
49
50 class ParamsValidator(UnicodeStringValidator):
51     def to_python(self, value, state):
52         if isinstance(value, basestring) or value is None:
53             value = super(ParamsValidator, self).to_python(value, state)
54             try:
55                 value = params_to_list(value)
56             except ParseError, e:
57                 raise Invalid("invalid parameters in the ParamsCol '%s', parse "
58                     "error: %s" % (self.name, e), value, state)
59         elif not isinstance(value, (list, tuple)):
60             raise Invalid("expected a tuple, list or valid string in the "
61                 "ParamsCol '%s', got %s %r instead"
62                     % (self.name, type(value), value), value, state)
63         return value
64     def from_python(self, value, state):
65         if isinstance(value, (list, tuple)):
66             value = ' '.join([repr(p) for p in value])
67         elif isinstance(value, basestring) or value is None:
68             value = super(ParamsValidator, self).to_python(value, state)
69             try:
70                 params_to_list(value)
71             except ParseError, e:
72                 raise Invalid("invalid parameters in the ParamsCol '%s', parse "
73                     "error: %s" % (self.name, e), value, state)
74         else:
75             raise Invalid("expected a tuple, list or valid string in the "
76                 "ParamsCol '%s', got %s %r instead"
77                     % (self.name, type(value), value), value, state)
78         return value
79
80 class SOParamsCol(SOUnicodeCol):
81     def createValidators(self):
82         return [ParamsValidator(db_encoding=self.dbEncoding, name=self.name)] \
83             + super(SOParamsCol, self).createValidators()
84
85 class ParamsCol(UnicodeCol):
86     baseClass = SOParamsCol
87
88 #}}}
89
90 #{{{ Clases
91
92 def srepr(obj): #{{{
93     if obj is not None:
94         return obj.shortrepr()
95     return obj
96 #}}}
97
98 class Curso(SQLObject): #{{{
99     # Clave
100     anio            = IntCol(notNone=True)
101     cuatrimestre    = IntCol(notNone=True)
102     numero          = IntCol(notNone=True)
103     pk              = DatabaseIndex(anio, cuatrimestre, numero, unique=True)
104     # Campos
105     descripcion     = UnicodeCol(length=255, default=None)
106     # Joins
107     docentes        = MultipleJoin('DocenteInscripto')
108     alumnos         = MultipleJoin('AlumnoInscripto')
109     grupos          = MultipleJoin('Grupo')
110     ejercicios      = MultipleJoin('Ejercicio', orderBy='numero')
111
112     def __init__(self, docentes=[], ejercicios=[], alumnos=[], **kw):
113         super(Curso, self).__init__(**kw)
114         for d in docentes:
115             self.add_docente(d)
116         for (n, e) in enumerate(ejercicios):
117             self.add_ejercicio(n, e)
118         for a in alumnos:
119             self.add_alumno(a)
120
121     def set(self, docentes=None, ejercicios=None, alumnos=None, **kw):
122         super(Curso, self).set(**kw)
123         if docentes is not None:
124             for d in DocenteInscripto.selectBy(curso=self):
125                 d.destroySelf()
126             for d in docentes:
127                 self.add_docente(d)
128         if ejercicios is not None:
129             for e in Ejercicio.selectBy(curso=self):
130                 e.destroySelf()
131             for (n, e) in enumerate(ejercicios):
132                 self.add_ejercicio(n+1, e)
133         if alumnos is not None:
134             for a in AlumnoInscripto.selectBy(curso=self):
135                 a.destroySelf()
136             for a in alumnos:
137                 self.add_alumno(a)
138
139     def add_docente(self, docente, **kw):
140         if isinstance(docente, Docente):
141             kw['docente'] = docente
142         else:
143             kw['docenteID'] = docente
144         return DocenteInscripto(curso=self, **kw)
145
146     def remove_docente(self, docente):
147         if isinstance(docente, Docente):
148             docente = docente.id
149         # FIXME esto deberian arreglarlo en SQLObject y debería ser
150         # DocenteInscripto.pk.get(self, docente).destroySelf()
151         DocenteInscripto.pk.get(self.id, docente).destroySelf()
152
153     def add_alumno(self, alumno, **kw):
154         if isinstance(alumno, Alumno):
155             kw['alumno'] = alumno
156         else:
157             kw['alumnoID'] = alumno
158         return AlumnoInscripto(curso=self, **kw)
159
160     def remove_alumno(self, alumno):
161         if isinstance(alumno, Alumno):
162             alumno = alumno.id
163         # FIXME esto deberian arreglarlo en SQLObject
164         AlumnoInscripto.pk.get(self.id, alumno).destroySelf()
165
166     def add_grupo(self, nombre, **kw):
167         return Grupo(curso=self, nombre=unicode(nombre), **kw)
168
169     def remove_grupo(self, nombre):
170         # FIXME esto deberian arreglarlo en SQLObject
171         Grupo.pk.get(self.id, nombre).destroySelf()
172
173     def add_ejercicio(self, numero, enunciado, **kw):
174         if isinstance(enunciado, Enunciado):
175             kw['enunciado'] = enunciado
176         else:
177             kw['enunciadoID'] = enunciado
178         return Ejercicio(curso=self, numero=numero, **kw)
179
180     def remove_ejercicio(self, numero):
181         # FIXME esto deberian arreglarlo en SQLObject
182         Ejercicio.pk.get(self.id, numero).destroySelf()
183
184     def __repr__(self):
185         return 'Curso(id=%s, anio=%s, cuatrimestre=%s, numero=%s, ' \
186             'descripcion=%s)' \
187                 % (self.id, self.anio, self.cuatrimestre, self.numero,
188                     self.descripcion)
189
190     def shortrepr(self):
191         return '%s.%s.%s' \
192             % (self.anio, self.cuatrimestre, self.numero)
193 #}}}
194
195 class Usuario(InheritableSQLObject): #{{{
196     # Clave (para docentes puede ser un nombre de usuario arbitrario)
197     usuario         = UnicodeCol(length=10, alternateID=True)
198     # Campos
199     contrasenia     = UnicodeCol(length=255, default=None)
200     nombre          = UnicodeCol(length=255, notNone=True)
201     email           = UnicodeCol(length=255, default=None)
202     telefono        = UnicodeCol(length=255, default=None)
203     creado          = DateTimeCol(notNone=True, default=DateTimeCol.now)
204     observaciones   = UnicodeCol(default=None)
205     activo          = BoolCol(notNone=True, default=True)
206     # Joins
207     roles           = RelatedJoin('Rol', addRemoveName='_rol')
208
209     def __init__(self, password=None, roles=[], **kw):
210         if password is not None:
211             kw['contrasenia'] = encryptpw(password)
212         super(Usuario, self).__init__(**kw)
213         for r in roles:
214             self.add_rol(r)
215
216     def set(self, password=None, roles=None, **kw):
217         if password is not None:
218             kw['contrasenia'] = encryptpw(password)
219         super(Usuario, self).set(**kw)
220         if roles is not None:
221             for r in self.roles:
222                 self.remove_rol(r)
223             for r in roles:
224                 self.add_rol(r)
225
226     def _get_user_name(self): # para identity
227         return self.usuario
228
229     @classmethod
230     def by_user_name(cls, user_name): # para identity
231         user = cls.byUsuario(user_name)
232         if not user.activo:
233             raise SQLObjectNotFound(_(u'El %s está inactivo' % cls.__name__))
234         return user
235
236     def _get_groups(self): # para identity
237         return self.roles
238
239     def _get_permissions(self): # para identity
240         perms = set()
241         for r in self.roles:
242             perms.update(r.permisos)
243         return perms
244
245     _get_permisos = _get_permissions
246
247     def _set_password(self, cleartext_password): # para identity
248         self.contrasenia = encryptpw(cleartext_password)
249
250     def _get_password(self): # para identity
251         return self.contrasenia
252
253     def __repr__(self):
254         raise NotImplementedError(_(u'Clase abstracta!'))
255
256     def shortrepr(self):
257         return '%s (%s)' % (self.usuario, self.nombre)
258 #}}}
259
260 class Docente(Usuario): #{{{
261     _inheritable = False
262     # Campos
263     nombrado    = BoolCol(notNone=True, default=True)
264     # Joins
265     enunciados  = MultipleJoin('Enunciado', joinColumn='autor_id')
266     cursos      = MultipleJoin('DocenteInscripto')
267
268     def add_entrega(self, instancia, **kw):
269         return Entrega(instancia=instancia, **kw)
270
271     def add_enunciado(self, nombre, anio, cuatrimestre, **kw):
272         return Enunciado(nombre=nombre, anio=anio, cuatrimestre=cuatrimestre,
273             autor=self, **kw)
274
275     def remove_enunciado(self, nombre, anio, cuatrimestre):
276         Enunciado.pk.get(nombre, anio, cuatrimestre).destroySelf()
277
278     def __repr__(self):
279         return 'Docente(id=%s, usuario=%s, nombre=%s, password=%s, email=%s, ' \
280             'telefono=%s, activo=%s, creado=%s, observaciones=%s)' \
281                 % (self.id, self.usuario, self.nombre, self.password,
282                     self.email, self.telefono, self.activo, self.creado,
283                     self.observaciones)
284 #}}}
285
286 class Alumno(Usuario): #{{{
287     _inheritable = False
288     # Campos
289     nota            = DecimalCol(size=3, precision=1, default=None)
290     # Joins
291     inscripciones   = MultipleJoin('AlumnoInscripto')
292
293     def __init__(self, padron=None, **kw):
294         if padron: kw['usuario'] = padron
295         super(Alumno, self).__init__(**kw)
296
297     def set(self, padron=None, **kw):
298         if padron: kw['usuario'] = padron
299         super(Alumno, self).set(**kw)
300
301     def _get_padron(self): # alias para poder referirse al alumno por padron
302         return self.usuario
303
304     def _set_padron(self, padron):
305         self.usuario = padron
306
307     @classmethod
308     def byPadron(cls, padron):
309         return cls.byUsuario(unicode(padron))
310
311     def __repr__(self):
312         return 'Alumno(id=%s, padron=%s, nombre=%s, password=%s, email=%s, ' \
313             'telefono=%s, activo=%s, creado=%s, observaciones=%s)' \
314                 % (self.id, self.padron, self.nombre, self.password, self.email,
315                     self.telefono, self.activo, self.creado, self.observaciones)
316 #}}}
317
318 class Tarea(InheritableSQLObject): #{{{
319     # Clave
320     nombre              = UnicodeCol(length=30, alternateID=True)
321     # Campos
322     descripcion         = UnicodeCol(length=255, default=None)
323     terminar_si_falla   = BoolCol(notNone=True, default=True)
324     rechazar_si_falla   = BoolCol(notNone=True, default=True)
325     # Joins
326     enunciados          = RelatedJoin('Enunciado', addRemoveName='_enunciado')
327
328     def __repr__(self):
329         raise NotImplementedError('Tarea es una clase abstracta')
330
331     def shortrepr(self):
332         return self.nombre
333 #}}}
334
335 class TareaFuente(Tarea): #{{{
336     # Joins
337     comandos    = MultipleJoin('ComandoFuente', joinColumn='tarea_id')
338
339     def add_comando(self, orden, comando, **kw):
340         return ComandoFuente(tarea=self, orden=orden, comando=comando, **kw)
341
342     def remove_comando(self, orden):
343         ComandoFuente.pk.get(self.id, orden).destroySelf()
344
345     def __repr__(self):
346         return 'TareaFuente(id=%s, nombre=%s, descripcion=%s)' \
347                 % (self.id, self.nombre, self.descripcion)
348 #}}}
349
350 class TareaPrueba(Tarea): #{{{
351     # Joins
352     comandos    = MultipleJoin('ComandoPrueba', joinColumn='tarea_id')
353
354     def add_comando(self, orden, comando, **kw):
355         return ComandoPrueba(tarea=self, orden=orden, comando=comando, **kw)
356
357     def remove_comando(self, orden):
358         ComandoPrueba.pk.get(self.id, orden).destroySelf()
359
360     def __repr__(self):
361         return 'TareaPrueba(id=%s, nombre=%s, descripcion=%s)' \
362                 % (self.id, self.nombre, self.descripcion)
363 #}}}
364
365 class Comando(SQLObject): #{{{
366     # Campos
367     comando             = ParamsCol(length=255, notNone=True)
368     descripcion         = UnicodeCol(length=255, default=None)
369     retorno             = IntCol(default=None)
370     terminar_si_falla   = BoolCol(notNone=True, default=True)
371     rechazar_si_falla   = BoolCol(notNone=True, default=True)
372 #    archivos_entrada    = list(ArchivoEntrada) #TODO
373 #    archivos_salida     = list(ArchivoSalida)  #TODO
374
375     def ejecutar(self): pass # TODO
376
377     def __repr__(self):
378         raise NotImplementedError('Comando es una clase abstracta')
379
380     def shortrepr(self):
381         return self.nombre
382 #}}}
383
384 class ComandoFuente(Comando): #{{{
385     # Clave
386     tarea       = ForeignKey('TareaFuente', notNone=True, cascade=True)
387     orden       = IntCol(notNone=True)
388     pk          = DatabaseIndex(tarea, orden, unique=True)
389     # Campos
390     tiempo_cpu  = FloatCol(default=None)
391
392     def ejecutar(self): pass # TODO
393
394     def __repr__(self):
395         return 'ComandoFuente(tarea=%s, orden=%s, comando=%s, descripcion=%s, ' \
396             'retorno=%s, tiempo_cpu=%s, terminar_si_falla=%s, ' \
397             'rechazar_si_falla=%s)' \
398                 % (srepr(self.tarea), self.orden, self.comando, self.descripcion,
399                     self.retorno, self.tiempo_cpu, self.terminar_si_falla,
400                     self.rechazar_si_falla)
401 #}}}
402
403 class ComandoPrueba(Comando): #{{{
404     # Clave
405     tarea               = ForeignKey('TareaPrueba', notNone=True, cascade=True)
406     orden               = IntCol(notNone=True)
407     pk                  = DatabaseIndex(tarea, orden, unique=True)
408     # Campos
409     multipl_tiempo_cpu  = FloatCol(notNone=True, default=1.0)
410
411     def ejecutar(self): pass # TODO
412
413     def __repr__(self):
414         return 'ComandoPrueba(tarea=%s, orden=%s, comando=%s, descripcion=%s, ' \
415             'retorno=%s, tiempo_cpu=%s, terminar_si_falla=%s, ' \
416             'rechazar_si_falla=%s)' \
417                 % (srepr(self.tarea), self.orden, self.comando, self.descripcion,
418                     self.retorno, self.tiempo_cpu, self.terminar_si_falla,
419                     self.rechazar_si_falla)
420 #}}}
421
422 class Enunciado(SQLObject): #{{{
423     # Clave
424     nombre          = UnicodeCol(length=60)
425     anio            = IntCol(notNone=True)
426     cuatrimestre    = IntCol(notNone=True)
427     pk              = DatabaseIndex(nombre, anio, cuatrimestre, unique=True)
428     # Campos
429     descripcion     = UnicodeCol(length=255, default=None)
430     autor           = ForeignKey('Docente', cascade='null')
431     creado          = DateTimeCol(notNone=True, default=DateTimeCol.now)
432     archivo         = BLOBCol(default=None)
433     archivo_name    = UnicodeCol(length=255, default=None)
434     archivo_type    = UnicodeCol(length=255, default=None)
435     # Joins
436     ejercicios      = MultipleJoin('Ejercicio')
437     casos_de_prueba = MultipleJoin('CasoDePrueba')
438     tareas          = RelatedJoin('Tarea', addRemoveName='_tarea')
439
440     def __init__(self, tareas=[], **kw):
441         super(Enunciado, self).__init__(**kw)
442         for tarea in tareas:
443             self.add_tarea(tarea)
444
445     def set(self, tareas=None, **kw):
446         super(Enunciado, self).set(**kw)
447         if tareas is not None:
448             for tarea in self.tareas:
449                 self.remove_tarea(tarea)
450             for tarea in tareas:
451                 self.add_tarea(tarea)
452
453     @classmethod
454     def selectByCurso(self, curso):
455         return Enunciado.selectBy(cuatrimestre=curso.cuatrimestre, anio=curso.anio)
456
457     def add_caso_de_prueba(self, nombre, **kw):
458         return CasoDePrueba(enunciado=self, nombre=nombre, **kw)
459
460     def __repr__(self):
461         return 'Enunciado(id=%s, autor=%s, nombre=%s, descripcion=%s, ' \
462             'creado=%s)' \
463                 % (self.id, srepr(self.autor), self.nombre, self.descripcion, \
464                     self.creado)
465
466     def shortrepr(self):
467         return self.nombre
468 #}}}
469
470 class CasoDePrueba(SQLObject): #{{{
471     # Clave
472     enunciado           = ForeignKey('Enunciado', cascade=True)
473     nombre              = UnicodeCol(length=40, notNone=True)
474     pk                  = DatabaseIndex(enunciado, nombre, unique=True)
475     # Campos
476     descripcion         = UnicodeCol(length=255, default=None)
477     terminar_si_falla   = BoolCol(notNone=True, default=False)
478     rechazar_si_falla   = BoolCol(notNone=True, default=True)
479     parametros          = ParamsCol(length=255, default=None)
480     retorno             = IntCol(default=None)
481     tiempo_cpu          = FloatCol(default=None)
482     activo              = BoolCol(notNone=True, default=True)
483     # Joins
484     pruebas             = MultipleJoin('Prueba')
485
486     def __repr__(self):
487         return 'CasoDePrueba(enunciado=%s, nombre=%s, parametros=%s, ' \
488             'retorno=%s, tiempo_cpu=%s, descripcion=%s)' \
489                 % (srepr(self.enunciado), self.nombre, self.parametros,
490                     self.retorno, self.tiempo_cpu, self.descripcion)
491
492     def shortrepr(self):
493         return '%s:%s' % (self.enunciado.shortrepr(), self.nombre)
494 #}}}
495
496 class Ejercicio(SQLObject): #{{{
497     # Clave
498     curso           = ForeignKey('Curso', notNone=True, cascade=True)
499     numero          = IntCol(notNone=True)
500     pk              = DatabaseIndex(curso, numero, unique=True)
501     # Campos
502     enunciado       = ForeignKey('Enunciado', notNone=True, cascade=False)
503     grupal          = BoolCol(default=False) # None es grupal o individual
504     # Joins
505     instancias      = MultipleJoin('InstanciaDeEntrega')
506
507     def add_instancia(self, numero, inicio, fin, **kw):
508         return InstanciaDeEntrega(ejercicio=self, numero=numero, inicio=inicio,
509             fin=fin, **kw)
510
511     def remove_instancia(self, numero):
512         # FIXME self.id
513         InstanciaDeEntrega.pk.get(self.id, numero).destroySelf()
514
515     def __repr__(self):
516         return 'Ejercicio(id=%s, curso=%s, numero=%s, enunciado=%s, ' \
517             'grupal=%s)' \
518                 % (self.id, self.curso.shortrepr(), self.numero,
519                     self.enunciado.shortrepr(), self.grupal)
520
521     def shortrepr(self):
522         return '(%s, %s, %s)' \
523             % (self.curso.shortrepr(), str(self.numero), \
524                 self.enunciado.shortrepr())
525 #}}}
526
527 class InstanciaDeEntrega(SQLObject): #{{{
528     # Clave
529     ejercicio       = ForeignKey('Ejercicio', notNone=True, cascade=True)
530     numero          = IntCol(notNone=True)
531     pk              = DatabaseIndex(ejercicio, numero, unique=True)
532     # Campos
533     inicio          = DateTimeCol(notNone=True)
534     fin             = DateTimeCol(notNone=True)
535     procesada       = BoolCol(notNone=True, default=False)
536     observaciones   = UnicodeCol(default=None)
537     activo          = BoolCol(notNone=True, default=True)
538     # Joins
539     entregas        = MultipleJoin('Entrega', joinColumn='instancia_id')
540     correcciones    = MultipleJoin('Correccion', joinColumn='instancia_id')
541
542     def __repr__(self):
543         return 'InstanciaDeEntrega(id=%s, numero=%s, inicio=%s, fin=%s, ' \
544             'procesada=%s, observaciones=%s, activo=%s)' \
545                 % (self.id, self.numero, self.inicio, self.fin,
546                     self.procesada, self.observaciones, self.activo)
547
548     def shortrepr(self):
549         return self.numero
550 #}}}
551
552 class DocenteInscripto(SQLObject): #{{{
553     # Clave
554     curso           = ForeignKey('Curso', notNone=True, cascade=True)
555     docente         = ForeignKey('Docente', notNone=True, cascade=True)
556     pk              = DatabaseIndex(curso, docente, unique=True)
557     # Campos
558     corrige         = BoolCol(notNone=True, default=True)
559     observaciones   = UnicodeCol(default=None)
560     # Joins
561     alumnos         = MultipleJoin('AlumnoInscripto', joinColumn='tutor_id')
562     tutorias        = MultipleJoin('Tutor', joinColumn='docente_id')
563     correcciones    = MultipleJoin('Correccion', joinColumn='corrector_id')
564
565     def add_correccion(self, entrega, **kw):
566         return Correccion(instancia=entrega.instancia, entrega=entrega,
567             entregador=entrega.entregador, corrector=self, **kw)
568
569     def remove_correccion(self, instancia, entregador):
570         # FIXME instancia.id, entregador.id
571         Correccion.pk.get(instancia.id, entregador.id).destroySelf()
572
573     def __repr__(self):
574         return 'DocenteInscripto(id=%s, docente=%s, corrige=%s, ' \
575             'observaciones=%s' \
576                 % (self.id, self.docente.shortrepr(), self.corrige,
577                     self.observaciones)
578
579     def shortrepr(self):
580         return self.docente.shortrepr()
581 #}}}
582
583 class Entregador(InheritableSQLObject): #{{{
584     # Campos
585     nota            = DecimalCol(size=3, precision=1, default=None)
586     nota_cursada    = DecimalCol(size=3, precision=1, default=None)
587     observaciones   = UnicodeCol(default=None)
588     activo          = BoolCol(notNone=True, default=True)
589     # Joins
590     entregas        = MultipleJoin('Entrega')
591     correcciones    = MultipleJoin('Correccion')
592
593     def add_entrega(self, instancia, **kw):
594         return Entrega(instancia=instancia, entregador=self, **kw)
595
596     def __repr__(self):
597         raise NotImplementedError, 'Clase abstracta!'
598 #}}}
599
600 class Grupo(Entregador): #{{{
601     _inheritable = False
602     # Clave
603     curso           = ForeignKey('Curso', notNone=True, cascade=True)
604     nombre          = UnicodeCol(length=20, notNone=True)
605     pk              = DatabaseIndex(curso, nombre, unique=True)
606     # Campos
607     responsable     = ForeignKey('AlumnoInscripto', default=None, cascade='null')
608     # Joins
609     miembros        = MultipleJoin('Miembro')
610     tutores         = MultipleJoin('Tutor')
611
612     def __init__(self, miembros=[], tutores=[], **kw):
613         super(Grupo, self).__init__(**kw)
614         for a in miembros:
615             self.add_miembro(a)
616         for d in tutores:
617             self.add_tutor(d)
618
619     def set(self, miembros=None, tutores=None, **kw):
620         super(Grupo, self).set(**kw)
621         if miembros is not None:
622             for m in Miembro.selectBy(grupo=self):
623                 m.destroySelf()
624             for m in miembros:
625                 self.add_miembro(m)
626         if tutores is not None:
627             for t in Tutor.selectBy(grupo=self):
628                 t.destroySelf()
629             for t in tutores:
630                 self.add_tutor(t)
631
632     _doc_alumnos = 'Devuelve una lista de AlumnoInscriptos **activos**.'
633     def _get_alumnos(self):
634         return list([m.alumno for m in Miembro.selectBy(grupo=self, baja=None)])
635
636     _doc_docentes = 'Devuelve una lista de DocenteInscriptos **activos**.'
637     def _get_docentes(self):
638         return list([t.docente for t in Tutor.selectBy(grupo=self, baja=None)])
639
640     def add_miembro(self, alumno, **kw):
641         if isinstance(alumno, AlumnoInscripto):
642             alumno = alumno.id
643         return Miembro(grupo=self, alumnoID=alumno, **kw)
644
645     def remove_miembro(self, alumno):
646         if isinstance(alumno, AlumnoInscripto):
647             alumno = alumno.id
648         m = Miembro.selectBy(grupo=self, alumnoID=alumno, baja=None).getOne()
649         m.baja = DateTimeCol.now()
650
651     def add_tutor(self, docente, **kw):
652         if isinstance(docente, DocenteInscripto):
653             docente = docente.id
654         return Tutor(grupo=self, docenteID=docente, **kw)
655
656     def remove_tutor(self, docente):
657         if isinstance(docente, DocenteInscripto):
658             docente = docente.id
659         t = Tutor.selectBy(grupo=self, docenteID=docente, baja=None)
660         t.baja = DateTimeCol.now()
661
662     def __repr__(self):
663         return 'Grupo(id=%s, nombre=%s, responsable=%s, nota=%s, ' \
664             'nota_cursada=%s, observaciones=%s, activo=%s)' \
665                 % (self.id, self.nombre, srepr(self.responsable), self.nota,
666                     self.nota_cursada, self.observaciones, self.activo)
667
668     def shortrepr(self):
669         return 'grupo:' + self.nombre
670 #}}}
671
672 class AlumnoInscripto(Entregador): #{{{
673     _inheritable = False
674     # Clave
675     curso               = ForeignKey('Curso', notNone=True, cascade=True)
676     alumno              = ForeignKey('Alumno', notNone=True, cascade=True)
677     pk                  = DatabaseIndex(curso, alumno, unique=True)
678     # Campos
679     condicional         = BoolCol(notNone=True, default=False)
680     tutor               = ForeignKey('DocenteInscripto', default=None, cascade='null')
681     # Joins
682     responsabilidades   = MultipleJoin('Grupo', joinColumn='responsable_id')
683     membresias          = MultipleJoin('Miembro', joinColumn='alumno_id')
684     entregas            = MultipleJoin('Entrega', joinColumn='alumno_id')
685     correcciones        = MultipleJoin('Correccion', joinColumn='alumno_id')
686
687     def _get_nombre(self):
688         return self.alumno.padron
689
690     def __repr__(self):
691         return 'AlumnoInscripto(id=%s, alumno=%s, condicional=%s, nota=%s, ' \
692             'nota_cursada=%s, tutor=%s, observaciones=%s, activo=%s)' \
693                 % (self.id, self.alumno.shortrepr(), self.condicional,
694                     self.nota, self.nota_cursada, srepr(self.tutor),
695                     self.observaciones, self.activo)
696
697     def shortrepr(self):
698         return self.alumno.shortrepr()
699 #}}}
700
701 class Tutor(SQLObject): #{{{
702     # Clave
703     grupo           = ForeignKey('Grupo', notNone=True, cascade=True)
704     docente         = ForeignKey('DocenteInscripto', notNone=True, cascade=True)
705     pk              = DatabaseIndex(grupo, docente, unique=True)
706     # Campos
707     alta            = DateTimeCol(notNone=True, default=DateTimeCol.now)
708     baja            = DateTimeCol(default=None)
709
710     def __repr__(self):
711         return 'Tutor(docente=%s, grupo=%s, alta=%s, baja=%s)' \
712                 % (self.docente.shortrepr(), self.grupo.shortrepr(),
713                     self.alta, self.baja)
714
715     def shortrepr(self):
716         return '%s-%s' % (self.docente.shortrepr(), self.grupo.shortrepr())
717 #}}}
718
719 class Miembro(SQLObject): #{{{
720     # Clave
721     grupo           = ForeignKey('Grupo', notNone=True, cascade=True)
722     alumno          = ForeignKey('AlumnoInscripto', notNone=True, cascade=True)
723     pk              = DatabaseIndex(grupo, alumno, unique=True)
724     # Campos
725     nota            = DecimalCol(size=3, precision=1, default=None)
726     alta            = DateTimeCol(notNone=True, default=DateTimeCol.now)
727     baja            = DateTimeCol(default=None)
728
729     def __repr__(self):
730         return 'Miembro(alumno=%s, grupo=%s, nota=%s, alta=%s, baja=%s)' \
731                 % (self.alumno.shortrepr(), self.grupo.shortrepr(),
732                     self.nota, self.alta, self.baja)
733
734     def shortrepr(self):
735         return '%s-%s' % (self.alumno.shortrepr(), self.grupo.shortrepr())
736 #}}}
737
738 class Entrega(SQLObject): #{{{
739     # Clave
740     instancia           = ForeignKey('InstanciaDeEntrega', notNone=True, cascade=False)
741     entregador          = ForeignKey('Entregador', default=None, cascade=False) # Si es None era un Docente
742     fecha               = DateTimeCol(notNone=True, default=DateTimeCol.now)
743     pk                  = DatabaseIndex(instancia, entregador, fecha, unique=True)
744     # Campos
745     correcta            = BoolCol(default=None)
746     inicio_tareas       = DateTimeCol(default=None)
747     fin_tareas          = DateTimeCol(default=None)
748     observaciones       = UnicodeCol(default=None)
749     # Joins
750     comandos_ejecutados = MultipleJoin('ComandoFuenteEjecutado')
751     pruebas             = MultipleJoin('Prueba')
752     # Para generar código
753     codigo_dict         = r'0123456789abcdefghijklmnopqrstuvwxyz_.,*@#+'
754     codigo_format       = r'%m%d%H%M%S'
755
756     def add_comando_ejecutado(self, comando, **kw):
757         return ComandoFuenteEjecutado(entrega=self, comando=comando, **kw)
758
759     def remove_comando_ejecutado(self, comando):
760         if isinstance(comando, ComandoFuente):
761             comando = comando.id
762         # FIXME self.id
763         ComandoFuenteEjecutado.pk.get(self.id, comando).destroySelf()
764
765     def add_prueba(self, caso_de_prueba, **kw):
766         return Prueba(entrega=self, caso_de_prueba=caso_de_prueba, **kw)
767
768     def remove_prueba(self, caso_de_prueba):
769         if isinstance(caso_de_prueba, CasoDePrueba):
770             caso_de_prueba = caso_de_prueba.id
771         # FIXME self.id, caso_de_prueba
772         Prueba.pk.get(self.id, caso_de_prueba).destroySelf()
773
774     def _get_codigo(self):
775         if not hasattr(self, '_codigo'): # cache
776             n = long(self.fecha.strftime(Entrega.codigo_format))
777             d = Entrega.codigo_dict
778             l = len(d)
779             res = ''
780             while n:
781                     res += d[n % l]
782                     n /= l
783             self._codigo = res
784         return self._codigo
785
786     def _set_fecha(self, fecha):
787         self._SO_set_fecha(fecha)
788         if hasattr(self, '_codigo'): del self._codigo # bye, bye cache!
789
790     def _get_path(self):
791         import os.path
792         def path_join(*args):
793             return os.path.join(*[unicode(p) for p in args])
794         curso = self.entregador.curso
795         instancia = self.instancia
796         ejercicio = instancia.ejercicio
797         fecha = self.fecha.strftime(r'%Y-%m-%d_%H.%M.%S')
798         print ejercicio
799         return path_join(curso.anio, curso.cuatrimestre, curso.numero,
800             ejercicio.numero, instancia.numero, self.entregador.nombre, fecha)
801             # FIXME un grupo con nombre tipo "../../lala" puede romper todo.
802             #       Hacer que el nombre del grupo sea numérico (o validar que
803             #       sean solo caracteres inofensivos: letras ASCII, espacio
804             #       -traducirlos a underscores- y números). Creo que un numero
805             #       que se autoasigne es lo más cómodo.
806
807     def __repr__(self):
808         return 'Entrega(instancia=%s, entregador=%s, codigo=%s, fecha=%s, ' \
809             'correcta=%s, inicio_tareas=%s, fin_tareas=%s, observaciones=%s)' \
810                 % (self.instancia.shortrepr(), srepr(self.entregador),
811                     self.codigo, self.fecha, self.inicio_tareas,
812                     self.fin_tareas, self.correcta, self.observaciones)
813
814     def shortrepr(self):
815         return '%s-%s-%s' % (self.instancia.shortrepr(), srepr(self.entregador),
816             self.codigo)
817 #}}}
818
819 class Correccion(SQLObject): #{{{
820     # Clave
821     instancia       = ForeignKey('InstanciaDeEntrega', notNone=True, cascade=False)
822     entregador      = ForeignKey('Entregador', notNone=True, cascade=False) # Docente no tiene
823     pk              = DatabaseIndex(instancia, entregador, unique=True)
824     # Campos
825     entrega         = ForeignKey('Entrega', notNone=True, cascade=False)
826     corrector       = ForeignKey('DocenteInscripto', default=None, cascade='null')
827     asignado        = DateTimeCol(notNone=True, default=DateTimeCol.now)
828     corregido       = DateTimeCol(default=None)
829     nota            = DecimalCol(size=3, precision=1, default=None)
830     observaciones   = UnicodeCol(default=None)
831
832     def _get_entregas(self):
833         return list(Entrega.selectBy(instancia=self.instancia, entregador=self.entregador))
834
835     def __repr__(self):
836         return 'Correccion(instancia=%s, entregador=%s, entrega=%s, ' \
837             'corrector=%s, asignado=%s, corregido=%s, nota=%s, ' \
838             'observaciones=%s)' \
839                 % (self.instancia.shortrepr(), self.entregador.shortrepr(),
840                     self.entrega.shortrepr(), self.corrector, self.asignado,
841                     self.corregido, self.nota, self.observaciones)
842
843     def shortrepr(self):
844         if not self.corrector:
845             return '%s' % self.entrega.shortrepr()
846         return '%s,%s' % (self.entrega.shortrepr(), self.corrector.shortrepr())
847 #}}}
848
849 class ComandoEjecutado(InheritableSQLObject): #{{{
850     # Campos
851     inicio          = DateTimeCol(notNone=True, default=DateTimeCol.now)
852     fin             = DateTimeCol(default=None)
853     exito           = IntCol(default=None)
854     observaciones   = UnicodeCol(default=None)
855
856     def __repr__(self):
857         raise NotImplementedError('ComandoEjecutado es una clase abstracta')
858 #}}}
859
860 class ComandoFuenteEjecutado(ComandoEjecutado): #{{{
861     # Clave
862     comando = ForeignKey('ComandoFuente', notNone=True, cascade=False)
863     entrega = ForeignKey('Entrega', notNone=True, cascade=False)
864     pk      = DatabaseIndex(comando, entrega, unique=True)
865
866     def __repr__(self):
867         return 'ComandoFuenteEjecutado(comando=%s, entrega=%s, inicio=%s, ' \
868             'fin=%s, exito=%s, observaciones=%s)' \
869                 % (self.comando.shortrepr(), self.entrega.shortrepr(),
870                     self.inicio, self.fin, self.exito, self.observaciones)
871
872     def shortrepr(self):
873         return '%s-%s' % (self.tarea.shortrepr(), self.entrega.shortrepr())
874 #}}}
875
876 class ComandoPruebaEjecutado(ComandoEjecutado): #{{{
877     # Clave
878     comando = ForeignKey('ComandoPrueba', notNone=True, cascade=False)
879     prueba  = ForeignKey('Prueba', notNone=True, cascade=False)
880     pk      = DatabaseIndex(comando, prueba, unique=True)
881
882     def __repr__(self):
883         return 'ComandoPruebaEjecutado(comando=%s, prueba=%s, inicio=%s, ' \
884             'fin=%s, exito=%s, observaciones=%s)' \
885                 % (self.comando.shortrepr(), self.prueba.shortrepr(),
886                     self.inicio, self.fin, self.exito, self.observaciones)
887
888     def shortrepr(self):
889         return '%s:%s:%s' % (self.tarea.shortrepr(), self.entrega.shortrerp(),
890             self.caso_de_prueba.shortrerp())
891 #}}}
892
893 class Prueba(SQLObject): #{{{
894     # Clave
895     entrega             = ForeignKey('Entrega', notNone=True, cascade=False)
896     caso_de_prueba      = ForeignKey('CasoDePrueba', notNone=True, cascade=False)
897     pk                  = DatabaseIndex(entrega, caso_de_prueba, unique=True)
898     # Campos
899     inicio              = DateTimeCol(notNone=True, default=DateTimeCol.now)
900     fin                 = DateTimeCol(default=None)
901     pasada              = IntCol(default=None)
902     observaciones       = UnicodeCol(default=None)
903     # Joins
904     comandos_ejecutados = MultipleJoin('ComandoPruebaEjecutado')
905
906     def add_comando_ejecutado(self, comando, **kw):
907         if isinstance(comando, ComandoPrueba):
908             comando = comando.id
909         return ComandoPruebaEjecutado(prueba=self, comandoID=comando, **kw)
910
911     def remove_comando_ejecutado(self, comando):
912         if isinstance(comando, ComandoPrueba):
913             comando = comando.id
914         # FIXME self.id, comando.id
915         ComandoPruebaEjecutado.pk.get(self.id, comando).destroySelf()
916
917     def __repr__(self):
918         return 'Prueba(entrega=%s, caso_de_prueba=%s, inicio=%s, fin=%s, ' \
919             'pasada=%s, observaciones=%s)' \
920                 % (self.entrega.shortrepr(), self.caso_de_prueba.shortrepr(),
921                 self.inicio, self.fin, self.pasada, self.observaciones)
922
923     def shortrepr(self):
924         return '%s:%s' % (self.entrega.shortrepr(),
925             self.caso_de_prueba.shortrerp())
926 #}}}
927
928 #{{{ Específico de Identity
929
930 class Visita(SQLObject): #{{{
931     visit_key   = StringCol(length=40, alternateID=True,
932                     alternateMethodName="by_visit_key")
933     created     = DateTimeCol(notNone=True, default=datetime.now)
934     expiry      = DateTimeCol()
935
936     @classmethod
937     def lookup_visit(cls, visit_key):
938         try:
939             return cls.by_visit_key(visit_key)
940         except SQLObjectNotFound:
941             return None
942 #}}}
943
944 class VisitaUsuario(SQLObject): #{{{
945     # Clave
946     visit_key   = StringCol(length=40, alternateID=True,
947                           alternateMethodName="by_visit_key")
948     # Campos
949     user_id     = IntCol() # Negrada de identity
950 #}}}
951
952 class Rol(SQLObject): #{{{
953     # Clave
954     nombre      = UnicodeCol(length=255, alternateID=True,
955                     alternateMethodName='by_nombre')
956     # Campos
957     descripcion = UnicodeCol(length=255, default=None)
958     creado      = DateTimeCol(notNone=True, default=datetime.now)
959     permisos    = TupleCol(notNone=True)
960     # Joins
961     usuarios    = RelatedJoin('Usuario', addRemoveName='_usuario')
962
963     def by_group_name(self, name): # para identity
964         return self.by_nombre(name)
965 #}}}
966
967 # No es un SQLObject porque no tiene sentido agregar/sacar permisos, están
968 # hardcodeados en el código
969 class Permiso(object): #{{{
970     max_valor = 1
971     def __init__(self, nombre, descripcion):
972         self.valor = Permiso.max_valor
973         Permiso.max_valor <<= 1
974         self.nombre = nombre
975         self.descripcion = descripcion
976
977     @classmethod
978     def createTable(cls, ifNotExists): # para identity
979         pass
980
981     @property
982     def permission_name(self): # para identity
983         return self.nombre
984
985     def __and__(self, other):
986         return self.valor & other.valor
987
988     def __or__(self, other):
989         return self.valor | other.valor
990
991     def __repr__(self):
992         return self.nombre
993 #}}}
994
995 # TODO ejemplos
996 entregar_tp = Permiso(u'entregar', u'Permite entregar trabajos prácticos')
997 admin = Permiso(u'admin', u'Permite hacer ABMs arbitrarios')
998
999 #}}} Identity
1000
1001 #}}} Clases
1002