Agregar funcionalidad de edición a Comandos.

[software/sercom.git] / sercom / subcontrollers / tarea_prueba / comandos / __init__.py

diff --git a/sercom/subcontrollers/tarea_prueba/comandos/__init__.py b/sercom/subcontrollers/tarea_prueba/comandos/__init__.py
index 393c780581f4593edec8e1eb50367d32d69f5772..13f1959caeb83d09ec2e41f8dc59be4d43970967 100644 (file)
--- a/sercom/subcontrollers/tarea_prueba/comandos/__init__.py
+++ b/sercom/subcontrollers/tarea_prueba/comandos/__init__.py
@@ -36,19 +36,20 @@ class ComandoPruebaForm(W.TableForm):
     class Fields(W.WidgetsList):
         tareaID = W.HiddenField()
         orden = W.TextField(label=_(u'Orden'), validator=V.Int(not_empty=True))
-        comando = W.TextField(label=_(u'Comando'), validator=V.UnicodeString(min=3, max=255, strip=True))
-        descripcion = W.TextField(label=_(u'Descripcion'), validator=V.UnicodeString(min=3, max=255, strip=True))
-        retorno = W.TextField(label=_(u'Retorno'), help_text=u"Codigo de retorno esperado",validator=V.Int(if_empty=0))
+        comando = W.TextField(label=_(u'Comando'), validator=V.UnicodeString(max=255, strip=True))
+        descripcion = W.TextField(label=_(u'Descripcion'), validator=V.UnicodeString(min=5, max=255, strip=True))
+        retorno = W.TextField(label=_(u'Retorno'), help_text=u"Codigo de retorno esperado",validator=V.Int(if_empty=ComandoPrueba.RET_PRUEBA))
         max_tiempo_cpu = W.TextField(label=_(u'CPU'), help_text=u"Maximo tiempo de CPU que puede utilizar [seg]",validator=V.Int())
         max_memoria = W.TextField(label=_(u'Memoria'), help_text=u"Maximo cantidad de memoria que puede utilizar [MB]",validator=V.Int())
         max_tam_archivo = W.TextField(label=_(u'Tam. Archivo'), help_text=u"Maximo tamanio de archivo a crear [MB]",validator=V.Int())
         max_cant_archivos = W.TextField(label=_(u'Archivos'),validator=V.Int())
         max_cant_procesos = W.TextField(label=_(u'Procesos'),validator=V.Int())
         max_locks_memoria = W.TextField(label=_(u'Mem. Locks'),validator=V.Int())
-        terminar_si_falla = W.CheckBox(label=_(u'Terminar si falla'), default=1, validator=V.Bool(if_empty=1))
-        rechazar_si_falla = W.CheckBox(label=_(u'Terminar si falla'), default=1, validator=V.Bool(if_empty=1))
-        archivos_entrada = W.FileField(label=_(u'Archivos Entrada'))
-        archivos_a_comparar = W.FileField(label=_(u'Archivos a Comparar'))
+        terminar_si_falla = W.CheckBox(label=_(u'Terminar si falla?'), default=1, validator=V.Bool(if_empty=1))
+        rechazar_si_falla = W.CheckBox(label=_(u'Rechazar si falla?'), default=1, validator=V.Bool(if_empty=1))
+        publico = W.CheckBox(label=_(u'Es público?'), default=1, validator=V.Bool(if_empty=1))
+        los_archivos_entrada = W.FileField(label=_(u'Archivos Entrada'))
+        los_archivos_a_comparar = W.FileField(label=_(u'Archivos a Comparar'))
         archivos_guardar = W.TextField(label=_(u'Archivos a Guardar'))
         activo = W.CheckBox(label=_(u'Activo'), default=1, validator=V.Bool(if_empty=1))
     fields = Fields()
@@ -82,21 +83,20 @@ class ComandoPruebaController(controllers.Controller, identity.SecureResource):
         """Save or create record to model"""
         t = TareaPrueba.get(kw['tareaID'])
         orden = kw['orden']
-        del(kw['orden'])
-        del(kw['tareaID'])
-        if kw['archivos_entrada'].filename:
-            kw['archivos_entrada'] = kw['archivos_entrada'].file.read()
-        else:
-            kw['archivos_entrada'] =  None
-        if kw['archivos_a_comparar'].filename:
-            kw['archivos_a_comparar'] = kw['archivos_a_comparar'].file.read()
-        else:
-            kw['archivos_a_comparar'] =  None
+        del kw['orden']
+        del kw['tareaID']
+        if kw['los_archivos_entrada'].filename:
+            kw['archivos_entrada'] = kw['los_archivos_entrada'].file.read()
+        del kw['los_archivos_entrada']
+        if kw['los_archivos_a_comparar'].filename:
+            kw['archivos_a_comparar'] = kw['los_archivos_a_comparar'].file.read()
+        del kw['los_archivos_a_comparar']
         # TODO : Hacer ventanita mas amigable para cargar esto.
         try:
             kw['archivos_a_guardar'] = tuple(kw['archivos_guardar'].split(','))
-        except:
-            del(kw['archivos_guardar'])
+        except AttributeError:
+            pass
+        del kw['archivos_guardar']
         t.add_comando(orden, **kw)
         flash(_(u'Se creó un nuevo %s.') % name)
         raise redirect('list/%d' % t.id)
@@ -113,6 +113,21 @@ class ComandoPruebaController(controllers.Controller, identity.SecureResource):
     @expose()
     def update(self, id, **kw):
         """Save or create record to model"""
+        orden = kw['orden']
+        del kw['orden']
+        del kw['tareaID']
+        if kw['los_archivos_entrada'].filename:
+            kw['archivos_entrada'] = kw['los_archivos_entrada'].file.read()
+        del kw['los_archivos_entrada']
+        if kw['los_archivos_a_comparar'].filename:
+            kw['archivos_a_comparar'] = kw['los_archivos_a_comparar'].file.read()
+        del kw['los_archivos_a_comparar']
+        # TODO : Hacer ventanita mas amigable para cargar esto.
+        try:
+            kw['archivos_a_guardar'] = tuple(kw['archivos_guardar'].split(','))
+        except AttributeError:
+            pass
+        del kw['archivos_guardar']
         r = validate_set(id, kw)
         flash(_(u'El %s fue actualizado.') % name)
         raise redirect('../list/%d' % r.tarea.id)
@@ -136,5 +151,20 @@ class ComandoPruebaController(controllers.Controller, identity.SecureResource):
         flash(_(u'El %s fue eliminado permanentemente.') % name)
         raise redirect('../list/%d' % tareaID)
 
+    @expose()
+    def get_archivos_entrada(self, id):
+        from cherrypy import request, response
+        r = validate_get(id)
+        response.headers["Content-Type"] = "application/zip"
+        response.headers["Content-disposition"] = "attachment;filename=archivos_entrada.zip"
+        return r.archivos_entrada
+
+    @expose()
+    def get_archivos_a_comparar(self, id):
+        from cherrypy import request, response
+        r = validate_get(id)
+        response.headers["Content-Type"] = "application/zip"
+        response.headers["Content-disposition"] = "attachment;filename=archivos_a_comparar.zip"
+        return r.archivos_a_comparar
 #}}}