X-Git-Url: https://git.llucax.com/software/sercom-old.git/blobdiff_plain/9c348742728b33f40e9fe55936741c544dbd7050..606019a00431114a45885c910f86ed74b2513e14:/src/sercom/sqlo.py

diff --git a/src/sercom/sqlo.py b/src/sercom/sqlo.py
index f4824df..de2347c 100644
--- a/src/sercom/sqlo.py
+++ b/src/sercom/sqlo.py
@@ -1,6 +1,9 @@
 from sqlobject import *
 from sqlobject.sqlbuilder import *
+import errno
+import os
 from os import path
+from datetime import datetime
 
 __all__ = ('Curso', 'Inscripto', 'Docente', 'Ejercicio', 'Entrega',
            'CasoDePrueba', 'Intento', 'Correccion', 'Prueba')
@@ -11,9 +14,9 @@ dir_base = '.'
 class BaseSQLObject(SQLObject):
 
     @classmethod
-    def by(cls, connection = None, **kw):
+    def by(cls, **kw):
         try:
-            return cls.selectBy(connection = connection, **kw)[0]
+            return cls.selectBy(limit=1, **kw)[0]
         except IndexError:
             raise SQLObjectNotFound, "The object %s with columns %s does not exist" % (cls.__name__, kw)
 
@@ -32,18 +35,18 @@ class Curso(BaseSQLObject):
 
 class Inscripto(BaseSQLObject):
     # Clave
-    padron          = IntCol(alternateID = True)
+    padron          = IntCol(alternateID=True)
     # Campos
     curso           = ForeignKey('Curso')
     mail            = StringCol()
-    activo          = BoolCol(default = True)
+    activo          = BoolCol(default=True)
     # Joins
     intentos        = MultipleJoin('Intento')
     correcciones    = MultipleJoin('Correccion')
 
 class Docente(BaseSQLObject):
     # Clave
-    nombre          = StringCol(alternateID = True)
+    nombre          = StringCol(alternateID=True)
     # Campos
     mail            = StringCol()
     corrige         = BoolCol()
@@ -59,7 +62,7 @@ class Ejercicio(BaseSQLObject):
     numero          = Col()
     docente         = ForeignKey('Docente')
     # Joins
-    casosDePrueba   = MultipleJoin('CasoDePrueba', joinMethodName='casosDePrueba') # XXX hack
+    casosDePrueba   = MultipleJoin('CasoDePrueba')
     entregas        = MultipleJoin('Entrega')
     cursos          = RelatedJoin('Curso', intermediateTable = 'entrega')
 
@@ -75,38 +78,57 @@ class Entrega(BaseSQLObject):
     ejercicio       = ForeignKey('Ejercicio')
     desde           = DateTimeCol()
     hasta           = DateTimeCol()
+    finalizada      = BoolCol(default=False)
     # Joins
     cursos          = MultipleJoin('Curso')
     correcciones    = MultipleJoin('Correccion')
     intentos        = MultipleJoin('Intento')
 
+    @classmethod
+    def getPendientes(cls, connection=None):
+        return cls.select((cls.q.finalizada == False)
+            & (cls.q.hasta <= datetime.now()),
+            orderBy=cls.q.hasta, connection=connection)
+
 class CasoDePrueba(BaseSQLObject):
     # Clave
     ejercicio       = ForeignKey('Ejercicio')
     nombre          = StringCol()
     # Campos
     privado         = BoolCol()
-    activo          = BoolCol(default = True)
-    parametros      = StringCol(default = None)
-    codigoRetorno   = IntCol(default = False)
-    tiempoCpu       = FloatCol(default = None)
+    activo          = BoolCol(default=True)
+    parametros      = StringCol(default=None)
+    codigoRetorno   = IntCol(default=False)
+    tiempoCpu       = FloatCol(default=None)
     # Joins
     pruebas         = MultipleJoin('Prueba')
 
     def _get_path(self):
         return path.join(self.ejercicio.path, 'casos_de_prueba', self.nombre)
 
+    def _get_pathEntradas(self):
+        return path.join(self.path, 'entradas')
+
+    def _get_pathSalidas(self):
+        return path.join(self.path, 'salidas')
+
     def _get_archivosEntrada(self):
         try:
-            return os.listdir(path.join(self.path, 'entradas'))
-        except:
-            return []
+            return frozenset(os.listdir(self.pathEntradas))
+        except OSError, (err, msg):
+            if err == errno.ENOENT:
+                return frozenset()
+            else:
+                raise
 
     def _get_archivosSalida(self):
         try:
-            return os.listdir(path.join(self.path, 'salidas'))
-        except:
-            return []
+            return frozenset(os.listdir(self.pathSalidas))
+        except OSError, (err, msg):
+            if err == errno.ENOENT:
+                return frozenset()
+            else:
+                raise
 
 class Intento(BaseSQLObject):
     # Clave
@@ -115,20 +137,20 @@ class Intento(BaseSQLObject):
     numero          = IntCol()
     # Campos
     llegada         = DateTimeCol()
-    inicioCompila   = DateTimeCol(default = None)
-    finCompila      = DateTimeCol(default = None)
-    inicioPruebas   = DateTimeCol(default = None)
-    finPruebas      = DateTimeCol(default = None)
-    compila         = BoolCol(default = None)
+    inicioCompila   = DateTimeCol(default=None)
+    finCompila      = DateTimeCol(default=None)
+    inicioPruebas   = DateTimeCol(default=None)
+    finPruebas      = DateTimeCol(default=None)
+    compila         = BoolCol(default=None)
     mailRespuesta   = StringCol()
-    observaciones   = StringCol(default = None)
+    observaciones   = StringCol(default=None)
     # Joins
     pruebas         = MultipleJoin('Prueba')
 
     @classmethod
     def getProximoAProbar(cls, connection=None):
         try:
-            return cls.select(None == cls.q.compila, limit=1,
+            return cls.select(cls.q.compila == None, limit=1,
                 orderBy=cls.q.llegada, connection=connection)[0]
         except IndexError:
             return None
@@ -141,6 +163,9 @@ class Intento(BaseSQLObject):
             '%s.%s.%s' % (curso.curso, entrega.nroEjercicio, entrega.entrega),
             '%s.%s' % (self.inscripto.padron, self.numero))
 
+    def _get_chrootPath(self):
+        return path.join(self.path, 'chroot')
+
     def _get_pruebasPasadas(self):
         for p in self.pruebas:
             if not p.pasada:
@@ -161,13 +186,13 @@ class Intento(BaseSQLObject):
 
 class Correccion(BaseSQLObject):
     # Clave
-    entrega         = ForeignKey('Ejercicio')
+    entrega         = ForeignKey('Entrega')
     inscripto       = ForeignKey('Inscripto')
     # Campos
     intento         = ForeignKey('Intento')
     docente         = ForeignKey('Docente')
-    nota            = IntCol(default = None)
-    observaciones   = StringCol(default = None)
+    nota            = IntCol(default=None)
+    observaciones   = StringCol(default=None)
 
 class Prueba(BaseSQLObject):
     # Clave
@@ -175,8 +200,14 @@ class Prueba(BaseSQLObject):
     casoDePrueba    = ForeignKey('CasoDePrueba')
     # Campos
     inicio          = DateTimeCol()
-    fin             = DateTimeCol(default = None)
-    pasada          = BoolCol(default = None)
-    observaciones   = StringCol(default = None)
+    fin             = DateTimeCol(default=None)
+    pasada          = BoolCol(default=None)
+    observaciones   = StringCol(default=None)
+
+    def _get_archivosSalida(self):
+        ent = self.casoDePrueba.archivosEntrada
+        sal = self.casoDePrueba.archivosSalida
+        return frozenset([f for f in os.listdir(self.intento.chrootPath) \
+                    if f in sal or f not in ent and f <> 'tp'])
 
 # vim: set et sw=4 sts=4 :