ajeitando a edição de dados tanto de palestrante quanto de participante.

[cascardo/eventmanager.git] / views.py

diff --git a/views.py b/views.py
index e64e704..245b0bd 100644 (file)
--- a/views.py
+++ b/views.py
@@ -354,22 +354,31 @@ def editar_trabalho(request, codigo):
 
 @login_required
 def meus_dados(request):
-    form = EditarPalestrante(request.POST or None)
-    palestrante = request.user.palestrante_set.get()
+    try:
+        entity = request.user.palestrante_set.get()
+    except Palestrante.DoesNotExist:
+        entity = request.user.participante_set.get()
+
+    FormKlass = form_for_instance(entity)
+
+    # ugly hammer to hide some fields...
+    del FormKlass.base_fields['usuario']
+
+    form = FormKlass(request.POST or None)
     ok = False
 
     for name, field in form.fields.items():
-        field.initial = getattr(palestrante, name)
+        field.initial = getattr(entity, name)
 
     if request.POST and form.is_valid():
         cd = form.cleaned_data
         for name, field in form.fields.items():
-            setattr(palestrante, name, cd[name])
-        palestrante.save()
+            setattr(entity, name, cd[name])
+        entity.save()
         ok = True
 
-    c = {'form': form, 'ok': ok}
-    return build_response(request, 'editar_palestrante.html', c)
+    c = {'form': form, 'ok': ok, 'title': entity.__class__.__name__}
+    return build_response(request, 'editar_usuario.html', c)
 
 
 @enable_login_form