Se agrega zip de entregas rechazadas.

diff --git a/src/sc_suwi b/src/sc_suwi
index be472154a07a86ac261eaa50ed0c18614a9d1adb..cea388603184e5dfe948af4904a2a19a8136a929 100755 (executable)
--- a/src/sc_suwi
+++ b/src/sc_suwi
@@ -118,6 +118,11 @@ def input_zip(req, entrega_id):
     return input_entrega(req, entrega_id) \
         + '<input type="hidden" name="zip" value="1" />\n'
 
+def input_zip_rechazados(req, entrega_id):
+    return input_entrega(req, entrega_id) \
+        + '<input type="hidden" name="zip" value="1" />\n' \
+        + '<input type="hidden" name="rechazados" value="1" />\n'
+
 def input_inscripto(req, inscripto_id=None):
     if inscripto_id is not None:
         return '<input type="hidden" name="inscripto" value="%d" />\n' \
@@ -251,7 +256,7 @@ def intentos(req, entrega_id, inscripto_id):
     r += footer()
     return r
 
-def zip(req, entrega_id):
+def zip(req, entrega_id, rechazados=False):
     def zip_path(path, base, zipfd):
         paths = os.listdir(path)
         for p in paths:
@@ -263,13 +268,26 @@ def zip(req, entrega_id):
     from zipfile import ZipFile, ZIP_DEFLATED
     e = Entrega.get(entrega_id, connection=conn)
     c = e.curso
-    req.write(http_header_zip(req, 'entrega-%d.%d.%d-%d.%d.zip'
-        % (c.anio, c.cuatrimestre, c.curso, e.nroEjercicio, e.entrega)))
+    filename = 'entrega-%d.%d.%d-%d.%d' \
+        % (c.anio, c.cuatrimestre, c.curso, e.nroEjercicio, e.entrega)
+    if rechazados:
+        filename += '-rechazados'
+    filename += '.zip'
+    req.write(http_header_zip(req, filename))
     tmpfname = os.tmpnam()
     zipfd = ZipFile(tmpfname, 'w', ZIP_DEFLATED)
-    for c in e.correcciones:
-        zip_path(os.path.join(conf.get('general', 'data_dir'), c.intento.path),
-            str(c.intento.inscripto.padron), zipfd)
+    if rechazados:
+        inscriptos_ok = set([c.inscripto for c in e.correcciones])
+        inscriptos = set([i.inscripto for i in e.intentos])
+        for i in inscriptos - inscriptos_ok:
+            intento = list(Intento.selectBy(entregaID=e.id, inscriptoID=i.id,
+                connection=conn))[-1]
+            zip_path(os.path.join(conf.get('general', 'data_dir'),
+                intento.path), str(i.padron), zipfd)
+    else:
+        for c in e.correcciones:
+            zip_path(os.path.join(conf.get('general', 'data_dir'),
+                c.intento.path), str(c.intento.inscripto.padron), zipfd)
     zipfd.close()
     req.write(file(tmpfname, 'r').read())
     os.unlink(tmpfname)
@@ -334,6 +352,8 @@ def correcciones(req, entrega_id):
         r += form(req, 'Elegir curso', input_login(req))
         r += form(req, 'Elegir entrega', input_curso(req, e.curso.id))
         r += form(req, 'Bajar entrega en .zip', input_zip(req, entrega_id))
+        r += form(req, 'Bajar entrega rechazadas en .zip',
+            input_zip_rechazados(req, entrega_id))
         r += '</p>\n'
         return r
 
@@ -388,7 +408,7 @@ elif f.has_key('intento'):
 elif f.has_key('inscripto'):
     print intentos(req, int(f.getfirst('entrega')), int(f.getfirst('inscripto')))
 elif f.has_key('zip'):
-    zip(req, int(f.getfirst('entrega')))
+    zip(req, int(f.getfirst('entrega')), f.getfirst('rechazados', False))
 else:
     print correcciones(req, int(f.getfirst('entrega')))