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')
 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')
 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(SQLObject): #{{{
 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