sercom/subcontrollers/tarea_prueba/comandos/__init__.py

   1 # vim: set et sw=4 sts=4 encoding=utf-8 foldmethod=marker :
   2
   3 #{{{ Imports
   4 import cherrypy
   5 from turbogears import controllers, expose, redirect
   6 from turbogears import validate, flash, error_handler
   7 from turbogears import validators as V
   8 from turbogears import widgets as W
   9 from turbogears import identity
  10 from turbogears import paginate
  11 from docutils.core import publish_parts
  12 from sercom.subcontrollers import validate as val
  13 from sercom.model import TareaPrueba, ComandoPrueba, Comando
  14 from sqlobject import *
  15 #}}}
  16
  17 #{{{ Configuración
  18 cls = ComandoPrueba
  19 name = 'comando prueba'
  20 namepl = 'comandos prueba'
  21 #}}}
  22
  23 #{{{ Validación
  24 def validate_get(id):
  25     return val.validate_get(cls, name, id)
  26
  27 def validate_set(id, data):
  28     return val.validate_set(cls, name, id, data)
  29
  30 def validate_new(data):
  31     return val.validate_new(cls, name, data)
  32 #}}}
  33
  34 #{{{ Formulario
  35 class ComandoPruebaForm(W.TableForm):
  36     class Fields(W.WidgetsList):
  37         tareaID = W.HiddenField()
  38         orden = W.TextField(label=_(u'Orden'), validator=V.Int(not_empty=True))
  39         comando = W.TextField(label=_(u'Comando'), validator=V.UnicodeString(max=255, strip=True))
  40         descripcion = W.TextField(label=_(u'Descripcion'), validator=V.UnicodeString(min=5, max=255, strip=True))
  41         retorno = W.TextField(label=_(u'Retorno'), help_text=u"Codigo de retorno esperado",validator=V.Int(if_empty=ComandoPrueba.RET_PRUEBA))
  42         max_tiempo_cpu = W.TextField(label=_(u'CPU'), help_text=u"Maximo tiempo de CPU que puede utilizar [seg]",validator=V.Int())
  43         max_memoria = W.TextField(label=_(u'Memoria'), help_text=u"Maximo cantidad de memoria que puede utilizar [MB]",validator=V.Int())
  44         max_tam_archivo = W.TextField(label=_(u'Tam. Archivo'), help_text=u"Maximo tamanio de archivo a crear [MB]",validator=V.Int())
  45         max_cant_archivos = W.TextField(label=_(u'Archivos'),validator=V.Int())
  46         max_cant_procesos = W.TextField(label=_(u'Procesos'),validator=V.Int())
  47         max_locks_memoria = W.TextField(label=_(u'Mem. Locks'),validator=V.Int())
  48         terminar_si_falla = W.CheckBox(label=_(u'Terminar si falla?'), default=1, validator=V.Bool(if_empty=1))
  49         rechazar_si_falla = W.CheckBox(label=_(u'Rechazar si falla?'), default=1, validator=V.Bool(if_empty=1))
  50         publico = W.CheckBox(label=_(u'Es público?'), default=1, validator=V.Bool(if_empty=1))
  51         los_archivos_entrada = W.FileField(label=_(u'Archivos Entrada'))
  52         los_archivos_a_comparar = W.FileField(label=_(u'Archivos a Comparar'))
  53         archivos_guardar = W.TextField(label=_(u'Archivos a Guardar'))
  54         activo = W.CheckBox(label=_(u'Activo'), default=1, validator=V.Bool(if_empty=1))
  55     fields = Fields()
  56     javascript = [W.JSSource("MochiKit.DOM.focusOnLoad('form_orden');")]
  57
  58 form = ComandoPruebaForm()
  59 #}}}
  60
  61 #{{{ Controlador
  62 class ComandoPruebaController(controllers.Controller, identity.SecureResource):
  63     """Basic model admin interface"""
  64     require = identity.has_permission('admin')
  65
  66     @expose(template='kid:%s.templates.list' % __name__)
  67     @paginate('records')
  68     def list(self, tareaID):
  69         """List records in model"""
  70         r = cls.select(cls.q.tareaID == tareaID)
  71         return dict(records=r, name=name, tareaID=int(tareaID),namepl=namepl)
  72
  73     @expose(template='kid:%s.templates.new' % __name__)
  74     def new(self, tareaID,**kw):
  75         """Create new records in model"""
  76         form.fields[0].attrs['value'] = tareaID
  77         return dict(name=name, namepl=namepl, form=form, values=kw)
  78
  79     @validate(form=form)
  80     @error_handler(new)
  81     @expose()
  82     def create(self, **kw):
  83         """Save or create record to model"""
  84         t = TareaPrueba.get(kw['tareaID'])
  85         orden = kw['orden']
  86         del kw['orden']
  87         del kw['tareaID']
  88         if kw['los_archivos_entrada'].filename:
  89             kw['archivos_entrada'] = kw['los_archivos_entrada'].file.read()
  90         del kw['los_archivos_entrada']
  91         if kw['los_archivos_a_comparar'].filename:
  92             kw['archivos_a_comparar'] = kw['los_archivos_a_comparar'].file.read()
  93         del kw['los_archivos_a_comparar']
  94         # TODO : Hacer ventanita mas amigable para cargar esto.
  95         try:
  96             kw['archivos_a_guardar'] = tuple(kw['archivos_guardar'].split(','))
  97         except AttributeError:
  98             pass
  99         del kw['archivos_guardar']
 100         t.add_comando(orden, **kw)
 101         flash(_(u'Se creó un nuevo %s.') % name)
 102         raise redirect('list/%d' % t.id)
 103
 104     @expose(template='kid:%s.templates.edit' % __name__)
 105     def edit(self, id, **kw):
 106         """Edit record in model"""
 107         r = validate_get(id)
 108         r.archivos_guardar = ",".join(r.archivos_a_guardar)
 109         return dict(name=name, namepl=namepl, record=r, form=form)
 110
 111     @validate(form=form)
 112     @error_handler(edit)
 113     @expose()
 114     def update(self, id, **kw):
 115         """Save or create record to model"""
 116         orden = kw['orden']
 117         del kw['orden']
 118         del kw['tareaID']
 119         if kw['los_archivos_entrada'].filename:
 120             kw['archivos_entrada'] = kw['los_archivos_entrada'].file.read()
 121         del kw['los_archivos_entrada']
 122         if kw['los_archivos_a_comparar'].filename:
 123             kw['archivos_a_comparar'] = kw['los_archivos_a_comparar'].file.read()
 124         del kw['los_archivos_a_comparar']
 125         # TODO : Hacer ventanita mas amigable para cargar esto.
 126         try:
 127             kw['archivos_a_guardar'] = tuple(kw['archivos_guardar'].split(','))
 128         except AttributeError:
 129             pass
 130         del kw['archivos_guardar']
 131         r = validate_set(id, kw)
 132         flash(_(u'El %s fue actualizado.') % name)
 133         raise redirect('../list/%d' % r.tarea.id)
 134
 135     @expose(template='kid:%s.templates.show' % __name__)
 136     def show(self,id, **kw):
 137         """Show record in model"""
 138         r = validate_get(id)
 139         return dict(name=name, namepl=namepl, record=r)
 140
 141     @expose()
 142     def delete(self, id):
 143         """Destroy record in model"""
 144         r = validate_get(id)
 145         tareaID = r.tarea.id
 146         r.destroySelf()
 147         flash(_(u'El %s fue eliminado permanentemente.') % name)
 148         raise redirect('../list/%d' % tareaID)
 149
 150     @expose()
 151     def file(self, name, id):
 152         from cherrypy import request, response
 153         r = validate_get(id)
 154         response.headers["Content-Type"] = "application/zip"
 155         response.headers["Content-disposition"] = "attachment;filename=%s_%d.zip" % (name, r.id)
 156         if name == "archivos_entrada":
 157             ret = r.archivos_entrada
 158         elif name == "archivos_a_comparar":
 159             ret = r.archivos_a_comparar
 160         else:
 161             raise NotFound
 162         return ret
 163 #}}}
 164