Generalizar manejo de foreign key simple en subcontrolador.

[z.facultad/75.52/sercom.git] / sercom / subcontrollers / enunciado / __init__.py

diff --git a/sercom/subcontrollers/enunciado/__init__.py b/sercom/subcontrollers/enunciado/__init__.py
index 2ac34356be79cc11a3c8f502d67485884eb803ac..d2897a5c10ea38d39e9aa72c1dbeecfe113dc6bd 100644 (file)
--- a/sercom/subcontrollers/enunciado/__init__.py
+++ b/sercom/subcontrollers/enunciado/__init__.py
@@ -2,49 +2,54 @@
 
 from turbogears import controllers, expose, redirect
 from turbogears import validate, validators, flash, error_handler
-from sercom.model import Enunciado, Docente
 from turbogears.widgets import *
 from turbogears import identity
 from turbogears import paginate
 from docutils.core import publish_parts
 from sercom.subcontrollers import validate as val
+from sercom.model import Enunciado, Docente
 
 cls = Enunciado
 name = 'enunciado'
 namepl = name + 's'
 
-def validate_autor(data):
-    autor = data.get('autorID', None)
-    if autor == 0: autor = None
-    if autor is not None:
+fkcls = Docente
+fkname = 'autor'
+fknamepl = fkname + 'es'
+
+def validate_fk(data):
+    fk = data.get(fkname + 'ID', None)
+    if fk == 0: fk = None
+    if fk is not None:
         try:
-            autor = Docente.get(autor)
+            fk = fkcls.get(fk)
         except LookupError:
             raise redirect('new', tg_flash=_(u'No se pudo crear el nuevo ' \
-                '%s porque el autor con identificador %d no existe.'
-                    % (name, autor)), **data)
-    data.pop('autorID', None)
-    data['autor'] = autor
+                '%s porque el %s con identificador %d no existe.'
+                    % (name, fkname, fk)), **data)
+    data.pop(fkname + 'ID', None)
+    data[fkname] = fk
+    return fk
 
 def validate_get(id):
     return val.validate_get(cls, name, id)
 
 def validate_set(id, data):
-    validate_autor(data)
+    validate_fk(data)
     return val.validate_set(cls, name, id, data)
 
 def validate_new(data):
-    validate_autor(data)
+    validate_fk(data)
     return val.validate_new(cls, name, data)
 
 def get_options():
-    return [(0, _(u'--'))] + [(a.id, a.shortrepr()) for a in Docente.select()]
+    return [(0, _(u'--'))] + [(fk.id, fk.shortrepr()) for fk in fkcls.select()]
 
 form = TableForm(fields=[
     TextField(name='nombre', label=_(u'Nombre'),
         help_text=_(u'Requerido y único.'),
         validator=validators.UnicodeString(min=5, max=60, strip=True)),
-    SingleSelectField(name='autorID', label=_(u'Autor'),
+    SingleSelectField(name=fkname+'ID', label=_(fkname.capitalize()),
         options=get_options, validator=validators.Int(not_empty=False)),
     TextField(name='descripcion', label=_(u'Descripción'),
         validator=validators.UnicodeString(not_empty=False, max=255, strip=True)),
@@ -64,6 +69,7 @@ class EnunciadoController(controllers.Controller, identity.SecureResource):
         raise redirect('list')
 
     @expose(template='kid:%s.templates.list' % __name__)
+    @validate(validators=dict(autorID=validators.Int))
     @paginate('records')
     def list(self, **kw):
         """List records in model"""
@@ -105,7 +111,10 @@ class EnunciadoController(controllers.Controller, identity.SecureResource):
     def show(self,id, **kw):
         """Show record in model"""
         r = validate_get(id)
-        r.desc = publish_parts(r.descripcion, writer_name='html')['html_body']
+        if r.descripcion is None:
+            r.desc = ''
+        else:
+            r.desc = publish_parts(r.descripcion, writer_name='html')['html_body']
         return dict(name=name, namepl=namepl, record=r)
 
     @expose()