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     def add_miembro(self, alumno, **kw):
 633         if isinstance(alumno, AlumnoInscripto):
 634             alumno = alumno.id
 635         return Miembro(grupo=self, alumnoID=alumno, **kw)
 636
 637     def remove_miembro(self, alumno):
 638         if isinstance(alumno, AlumnoInscripto):
 639             alumno = alumno.id
 640         # FIXME self.id
 641         Miembro.pk.get(self.id, alumno).destroySelf()
 642
 643     def add_tutor(self, docente, **kw):
 644         if isinstance(docente, DocenteInscripto):
 645             docente = docente.id
 646         return Tutor(grupo=self, docenteID=docente, **kw)
 647
 648     def remove_tutor(self, docente):
 649         if isinstance(docente, DocenteInscripto):
 650             docente = docente.id
 651         # FIXME self.id
 652         Tutor.pk.get(self.id, docente).destroySelf()
 653
 654     def __repr__(self):
 655         return 'Grupo(id=%s, nombre=%s, responsable=%s, nota=%s, ' \
 656             'nota_cursada=%s, observaciones=%s, activo=%s)' \
 657                 % (self.id, self.nombre, srepr(self.responsable), self.nota,
 658                     self.nota_cursada, self.observaciones, self.activo)
 659
 660     def shortrepr(self):
 661         return 'grupo:' + self.nombre
 662 #}}}
 663
 664 class AlumnoInscripto(Entregador): #{{{
 665     _inheritable = False
 666     # Clave
 667     curso               = ForeignKey('Curso', notNone=True, cascade=True)
 668     alumno              = ForeignKey('Alumno', notNone=True, cascade=True)
 669     pk                  = DatabaseIndex(curso, alumno, unique=True)
 670     # Campos
 671     condicional         = BoolCol(notNone=True, default=False)
 672     tutor               = ForeignKey('DocenteInscripto', default=None, cascade='null')
 673     # Joins
 674     responsabilidades   = MultipleJoin('Grupo', joinColumn='responsable_id')
 675     membresias          = MultipleJoin('Miembro', joinColumn='alumno_id')
 676     entregas            = MultipleJoin('Entrega', joinColumn='alumno_id')
 677     correcciones        = MultipleJoin('Correccion', joinColumn='alumno_id')
 678
 679     def _get_nombre(self):
 680         return self.alumno.padron
 681
 682     def __repr__(self):
 683         return 'AlumnoInscripto(id=%s, alumno=%s, condicional=%s, nota=%s, ' \
 684             'nota_cursada=%s, tutor=%s, observaciones=%s, activo=%s)' \
 685                 % (self.id, self.alumno.shortrepr(), self.condicional,
 686                     self.nota, self.nota_cursada, srepr(self.tutor),
 687                     self.observaciones, self.activo)
 688
 689     def shortrepr(self):
 690         return self.alumno.shortrepr()
 691 #}}}
 692
 693 class Tutor(SQLObject): #{{{
 694     # Clave
 695     grupo           = ForeignKey('Grupo', notNone=True, cascade=True)
 696     docente         = ForeignKey('DocenteInscripto', notNone=True, cascade=True)
 697     pk              = DatabaseIndex(grupo, docente, unique=True)
 698     # Campos
 699     alta            = DateTimeCol(notNone=True, default=DateTimeCol.now)
 700     baja            = DateTimeCol(default=None)
 701
 702     def __repr__(self):
 703         return 'Tutor(docente=%s, grupo=%s, alta=%s, baja=%s)' \
 704                 % (self.docente.shortrepr(), self.grupo.shortrepr(),
 705                     self.alta, self.baja)
 706
 707     def shortrepr(self):
 708         return '%s-%s' % (self.docente.shortrepr(), self.grupo.shortrepr())
 709 #}}}
 710
 711 class Miembro(SQLObject): #{{{
 712     # Clave
 713     grupo           = ForeignKey('Grupo', notNone=True, cascade=True)
 714     alumno          = ForeignKey('AlumnoInscripto', notNone=True, cascade=True)
 715     pk              = DatabaseIndex(grupo, alumno, unique=True)
 716     # Campos
 717     nota            = DecimalCol(size=3, precision=1, default=None)
 718     alta            = DateTimeCol(notNone=True, default=DateTimeCol.now)
 719     baja            = DateTimeCol(default=None)
 720
 721     def __repr__(self):
 722         return 'Miembro(alumno=%s, grupo=%s, nota=%s, alta=%s, baja=%s)' \
 723                 % (self.alumno.shortrepr(), self.grupo.shortrepr(),
 724                     self.nota, self.alta, self.baja)
 725
 726     def shortrepr(self):
 727         return '%s-%s' % (self.alumno.shortrepr(), self.grupo.shortrepr())
 728 #}}}
 729
 730 class Entrega(SQLObject): #{{{
 731     # Clave
 732     instancia           = ForeignKey('InstanciaDeEntrega', notNone=True, cascade=False)
 733     entregador          = ForeignKey('Entregador', default=None, cascade=False) # Si es None era un Docente
 734     fecha               = DateTimeCol(notNone=True, default=DateTimeCol.now)
 735     pk                  = DatabaseIndex(instancia, entregador, fecha, unique=True)
 736     # Campos
 737     correcta            = BoolCol(default=None)
 738     inicio_tareas       = DateTimeCol(default=None)
 739     fin_tareas          = DateTimeCol(default=None)
 740     observaciones       = UnicodeCol(default=None)
 741     # Joins
 742     comandos_ejecutados = MultipleJoin('ComandoFuenteEjecutado')
 743     pruebas             = MultipleJoin('Prueba')
 744     # Para generar código
 745     codigo_dict         = r'0123456789abcdefghijklmnopqrstuvwxyz_.,*@#+'
 746     codigo_format       = r'%m%d%H%M%S'
 747
 748     def add_comando_ejecutado(self, comando, **kw):
 749         return ComandoFuenteEjecutado(entrega=self, comando=comando, **kw)
 750
 751     def remove_comando_ejecutado(self, comando):
 752         if isinstance(comando, ComandoFuente):
 753             comando = comando.id
 754         # FIXME self.id
 755         ComandoFuenteEjecutado.pk.get(self.id, comando).destroySelf()
 756
 757     def add_prueba(self, caso_de_prueba, **kw):
 758         return Prueba(entrega=self, caso_de_prueba=caso_de_prueba, **kw)
 759
 760     def remove_prueba(self, caso_de_prueba):
 761         if isinstance(caso_de_prueba, CasoDePrueba):
 762             caso_de_prueba = caso_de_prueba.id
 763         # FIXME self.id, caso_de_prueba
 764         Prueba.pk.get(self.id, caso_de_prueba).destroySelf()
 765
 766     def _get_codigo(self):
 767         if not hasattr(self, '_codigo'): # cache
 768             n = long(self.fecha.strftime(Entrega.codigo_format))
 769             d = Entrega.codigo_dict
 770             l = len(d)
 771             res = ''
 772             while n:
 773                     res += d[n % l]
 774                     n /= l
 775             self._codigo = res
 776         return self._codigo
 777
 778     def _set_fecha(self, fecha):
 779         self._SO_set_fecha(fecha)
 780         if hasattr(self, '_codigo'): del self._codigo # bye, bye cache!
 781
 782     def _get_path(self):
 783         import os.path
 784         def path_join(*args):
 785             return os.path.join(*[unicode(p) for p in args])
 786         curso = self.entregador.curso
 787         instancia = self.instancia
 788         ejercicio = instancia.ejercicio
 789         fecha = self.fecha.strftime(r'%Y-%m-%d_%H.%M.%S')
 790         print ejercicio
 791         return path_join(curso.anio, curso.cuatrimestre, curso.numero,
 792             ejercicio.numero, instancia.numero, self.entregador.nombre, fecha)
 793             # FIXME un grupo con nombre tipo "../../lala" puede romper todo.
 794             #       Hacer que el nombre del grupo sea numérico (o validar que
 795             #       sean solo caracteres inofensivos: letras ASCII, espacio
 796             #       -traducirlos a underscores- y números). Creo que un numero
 797             #       que se autoasigne es lo más cómodo.
 798
 799     def __repr__(self):
 800         return 'Entrega(instancia=%s, entregador=%s, codigo=%s, fecha=%s, ' \
 801             'correcta=%s, inicio_tareas=%s, fin_tareas=%s, observaciones=%s)' \
 802                 % (self.instancia.shortrepr(), srepr(self.entregador),
 803                     self.codigo, self.fecha, self.inicio_tareas,
 804                     self.fin_tareas, self.correcta, self.observaciones)
 805
 806     def shortrepr(self):
 807         return '%s-%s-%s' % (self.instancia.shortrepr(), srepr(self.entregador),
 808             self.codigo)
 809 #}}}
 810
 811 class Correccion(SQLObject): #{{{
 812     # Clave
 813     instancia       = ForeignKey('InstanciaDeEntrega', notNone=True, cascade=False)
 814     entregador      = ForeignKey('Entregador', notNone=True, cascade=False) # Docente no tiene
 815     pk              = DatabaseIndex(instancia, entregador, unique=True)
 816     # Campos
 817     entrega         = ForeignKey('Entrega', notNone=True, cascade=False)
 818     corrector       = ForeignKey('DocenteInscripto', default=None, cascade='null')
 819     asignado        = DateTimeCol(notNone=True, default=DateTimeCol.now)
 820     corregido       = DateTimeCol(default=None)
 821     nota            = DecimalCol(size=3, precision=1, default=None)
 822     observaciones   = UnicodeCol(default=None)
 823
 824     def _get_entregas(self):
 825         return list(Entrega.selectBy(instancia=self.instancia, entregador=self.entregador))
 826
 827     def __repr__(self):
 828         return 'Correccion(instancia=%s, entregador=%s, entrega=%s, ' \
 829             'corrector=%s, asignado=%s, corregido=%s, nota=%s, ' \
 830             'observaciones=%s)' \
 831                 % (self.instancia.shortrepr(), self.entregador.shortrepr(),
 832                     self.entrega.shortrepr(), self.corrector, self.asignado,
 833                     self.corregido, self.nota, self.observaciones)
 834
 835     def shortrepr(self):
 836         if not self.corrector:
 837             return '%s' % self.entrega.shortrepr()
 838         return '%s,%s' % (self.entrega.shortrepr(), self.corrector.shortrepr())
 839 #}}}
 840
 841 class ComandoEjecutado(InheritableSQLObject): #{{{
 842     # Campos
 843     inicio          = DateTimeCol(notNone=True, default=DateTimeCol.now)
 844     fin             = DateTimeCol(default=None)
 845     exito           = IntCol(default=None)
 846     observaciones   = UnicodeCol(default=None)
 847
 848     def __repr__(self):
 849         raise NotImplementedError('ComandoEjecutado es una clase abstracta')
 850 #}}}
 851
 852 class ComandoFuenteEjecutado(ComandoEjecutado): #{{{
 853     # Clave
 854     comando = ForeignKey('ComandoFuente', notNone=True, cascade=False)
 855     entrega = ForeignKey('Entrega', notNone=True, cascade=False)
 856     pk      = DatabaseIndex(comando, entrega, unique=True)
 857
 858     def __repr__(self):
 859         return 'ComandoFuenteEjecutado(comando=%s, entrega=%s, inicio=%s, ' \
 860             'fin=%s, exito=%s, observaciones=%s)' \
 861                 % (self.comando.shortrepr(), self.entrega.shortrepr(),
 862                     self.inicio, self.fin, self.exito, self.observaciones)
 863
 864     def shortrepr(self):
 865         return '%s-%s' % (self.tarea.shortrepr(), self.entrega.shortrepr())
 866 #}}}
 867
 868 class ComandoPruebaEjecutado(SQLObject): #{{{
 869     # Clave
 870     comando = ForeignKey('ComandoPrueba', notNone=True, cascade=False)
 871     prueba  = ForeignKey('Prueba', notNone=True, cascade=False)
 872     pk      = DatabaseIndex(comando, prueba, unique=True)
 873
 874     def __repr__(self):
 875         return 'ComandoPruebaEjecutado(comando=%s, prueba=%s, inicio=%s, ' \
 876             'fin=%s, exito=%s, observaciones=%s)' \
 877                 % (self.comando.shortrepr(), self.prueba.shortrepr(),
 878                     self.inicio, self.fin, self.exito, self.observaciones)
 879
 880     def shortrepr(self):
 881         return '%s:%s:%s' % (self.tarea.shortrepr(), self.entrega.shortrerp(),
 882             self.caso_de_prueba.shortrerp())
 883 #}}}
 884
 885 class Prueba(SQLObject): #{{{
 886     # Clave
 887     entrega             = ForeignKey('Entrega', notNone=True, cascade=False)
 888     caso_de_prueba      = ForeignKey('CasoDePrueba', notNone=True, cascade=False)
 889     pk                  = DatabaseIndex(entrega, caso_de_prueba, unique=True)
 890     # Campos
 891     inicio              = DateTimeCol(notNone=True, default=DateTimeCol.now)
 892     fin                 = DateTimeCol(default=None)
 893     pasada              = IntCol(default=None)
 894     observaciones       = UnicodeCol(default=None)
 895     # Joins
 896     comandos_ejecutados = MultipleJoin('ComandoPruebaEjecutado')
 897
 898     def add_comando_ejecutado(self, comando, **kw):
 899         if isinstance(comando, ComandoPrueba):
 900             comando = comando.id
 901         return ComandoPruebaEjecutado(prueba=self, comandoID=comando, **kw)
 902
 903     def remove_comando_ejecutado(self, comando):
 904         if isinstance(comando, ComandoPrueba):
 905             comando = comando.id
 906         # FIXME self.id, comando.id
 907         ComandoPruebaEjecutado.pk.get(self.id, comando).destroySelf()
 908
 909     def __repr__(self):
 910         return 'Prueba(entrega=%s, caso_de_prueba=%s, inicio=%s, fin=%s, ' \
 911             'pasada=%s, observaciones=%s)' \
 912                 % (self.entrega.shortrepr(), self.caso_de_prueba.shortrepr(),
 913                 self.inicio, self.fin, self.pasada, self.observaciones)
 914
 915     def shortrepr(self):
 916         return '%s:%s' % (self.entrega.shortrepr(),
 917             self.caso_de_prueba.shortrerp())
 918 #}}}
 919
 920 #{{{ Específico de Identity
 921
 922 class Visita(SQLObject): #{{{
 923     visit_key   = StringCol(length=40, alternateID=True,
 924                     alternateMethodName="by_visit_key")
 925     created     = DateTimeCol(notNone=True, default=datetime.now)
 926     expiry      = DateTimeCol()
 927
 928     @classmethod
 929     def lookup_visit(cls, visit_key):
 930         try:
 931             return cls.by_visit_key(visit_key)
 932         except SQLObjectNotFound:
 933             return None
 934 #}}}
 935
 936 class VisitaUsuario(SQLObject): #{{{
 937     # Clave
 938     visit_key   = StringCol(length=40, alternateID=True,
 939                           alternateMethodName="by_visit_key")
 940     # Campos
 941     user_id     = IntCol() # Negrada de identity
 942 #}}}
 943
 944 class Rol(SQLObject): #{{{
 945     # Clave
 946     nombre      = UnicodeCol(length=255, alternateID=True,
 947                     alternateMethodName='by_nombre')
 948     # Campos
 949     descripcion = UnicodeCol(length=255, default=None)
 950     creado      = DateTimeCol(notNone=True, default=datetime.now)
 951     permisos    = TupleCol(notNone=True)
 952     # Joins
 953     usuarios    = RelatedJoin('Usuario', addRemoveName='_usuario')
 954
 955     def by_group_name(self, name): # para identity
 956         return self.by_nombre(name)
 957 #}}}
 958
 959 # No es un SQLObject porque no tiene sentido agregar/sacar permisos, están
 960 # hardcodeados en el código
 961 class Permiso(object): #{{{
 962     max_valor = 1
 963     def __init__(self, nombre, descripcion):
 964         self.valor = Permiso.max_valor
 965         Permiso.max_valor <<= 1
 966         self.nombre = nombre
 967         self.descripcion = descripcion
 968
 969     @classmethod
 970     def createTable(cls, ifNotExists): # para identity
 971         pass
 972
 973     @property
 974     def permission_name(self): # para identity
 975         return self.nombre
 976
 977     def __and__(self, other):
 978         return self.valor & other.valor
 979
 980     def __or__(self, other):
 981         return self.valor | other.valor
 982
 983     def __repr__(self):
 984         return self.nombre
 985 #}}}
 986
 987 # TODO ejemplos
 988 entregar_tp = Permiso(u'entregar', u'Permite entregar trabajos prácticos')
 989 admin = Permiso(u'admin', u'Permite hacer ABMs arbitrarios')
 990
 991 #}}} Identity
 992
 993 #}}} Clases
 994