adicionando patch do gabriel (brigaduxo =) e arrumando um probleminha no form de...

[cascardo/eventmanager.git] / views.py

diff --git a/views.py b/views.py
index e75b7e1..bf5db99 100644 (file)
--- a/views.py
+++ b/views.py
@@ -17,10 +17,11 @@ License along with this program; if not, write to the
 Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 Boston, MA 02111-1307, USA.
 """
-from django.shortcuts import render_to_response
+from django.shortcuts import render_to_response, get_object_or_404
 from django.template import RequestContext, Context, loader
 from django.contrib.auth.decorators import login_required, user_passes_test
 from django.contrib.auth.models import Group, User
+from django.newforms import form_for_instance
 from django.core.mail import EmailMessage
 from django.db import transaction
 from django.http import get_host
@@ -152,7 +153,7 @@ def inscricao(request):
         form = InscricaoEstudante(post2)
 
     # inscrição normal (sem ser estudante)
-    elif not 'estudante' in post or 'empresa' in post:
+    elif not 'estudante' in post and ('first_step' in post or 'empresa' in post):
         form = InscricaoNormal(post2)
 
     # primeiro passo...
@@ -205,6 +206,30 @@ def meus_trabalhos(request):
     c = {'trabalhos': t, 'palestrante': 1}
     return build_response(request, 'meus_trabalhos.html', c)
 
+@login_required
+@user_passes_test(lambda u:u.palestrante_set.count() == 1, login_url='/')
+def editar_trabalho(request,codigo):
+    try:
+        p = Palestrante.objects.get(usuario=request.user)
+    except Palestrante.DoesNotExist:
+        # não palestrante...
+        c = {'palestrante': 0}
+        return build_response(request, 'meus_trabalhos.html', c)
+    trabalho = get_object_or_404(Trabalho, id=codigo,palestrante=p)
+    Formulario = form_for_instance(trabalho)
+    if request.method == 'POST':
+        form = Formulario(request.POST)
+        if form.is_valid():
+            form.save()
+            t = Trabalho.objects.filter(palestrante=p)
+            c = {'trabalhos': t, 'palestrante': 1}
+            c['editado_sucesso']=trabalho.titulo
+            return build_response(request, 'meus_trabalhos.html', c)
+    else:
+        form = Formulario()
+    
+    c = {'formulario':form}
+    return build_response(request, 'editar_trabalho.html', c)
 
 @login_required
 def meus_dados(request):