ソースを参照

finalmente la parte azienda funziona

mauro 1 年間 前
コミット
eda7343a92
1 ファイル変更17 行追加15 行削除
  1. 17 15
      azienda/views.py

+ 17 - 15
azienda/views.py

@@ -57,16 +57,16 @@ def welcome(request):
         return HttpResponseRedirect(reverse('azienda:welcome'))
 
       print("sceltaazienda = ",request.POST.get('sceltaazienda'))
-<<<<<<< HEAD
-      if sceltaazienda < 1 : #richiesta una nuova azienda
-        if 'AziendaId' in data:
-          data['AziendaId'] = none
-        return HttpResponseRedirect(reverse('azienda:edit'))
-=======
-      if sceltaazienda < 1: #richiesta una nuova azienda
-        data['AziendaId']=None
+
+      # attività su nuova/edit azienda
+      if sceltaazienda == -1: #richiesta una nuova azienda
+        print('sceltaazienda -1:',sceltaazienda)
+        #return HttpResponseRedirect(reverse('azienda:editAzienda'))
+      elif sceltaazienda == 0:
+        print('sceltaazienda 0:',sceltaazienda)
+        request.session['AziendaIdEdit'] = 0
+        print('AziendaId',request.session['AziendaId'])
         return HttpResponseRedirect(reverse('azienda:editAzienda'))
->>>>>>> origin/Sede
       else:
         try:
           azienda = Azienda.objects.get(pk=sceltaazienda)
@@ -115,6 +115,7 @@ def welcome(request):
       print('azienda passata',request.POST.get('parazienda'))
       azienda = Azienda.objects.get(pk=azid)
       request.session['AziendaId'] = azienda.id
+      request.session['AziendaIdEdit'] = azienda.id
       return HttpResponseRedirect(reverse('azienda:editAzienda'))
 
   tmplist = list()
@@ -141,13 +142,13 @@ def editAzienda(request):
   user = data['admin'] = User.objects.get(pk=AdminId)
 
   azienda = None
-  if 'AziendaId' in request.session:
-    print('trovato riferimento AziendaId',request.session.get('AziendaId'))
-    azid = request.session.get('AziendaId')
+  if 'AziendaIdEdit' in request.session:
+    print('trovato riferimento AziendaIdEdit',request.session.get('AziendaIdEdit'))
+    azid = request.session.get('AziendaIdEdit')
     try:
       azienda = Azienda.objects.get(pk=azid)
       print('azienda trovata',azienda.nome)
-    except Azienda.DoesNotFound as dnf:
+    except Azienda.DoesNotExist as dnf:
         print(dnf)
 
   if request.method == "POST":
@@ -161,16 +162,17 @@ def editAzienda(request):
       azienda.mail = nuovaazienda.cleaned_data.get('mail')
       azienda.partitaiva = nuovaazienda.cleaned_data.get('partitaiva')
       azienda.save()
-      request.session['AziendaId'] = azienda.id
+      request.session['AziendaIdEdit'] = azienda.id
       print('azienda:',azienda.id)
 
       assegnazione = Assegnazione()
       if azienda:
-        assegnazione = Assegnazione.objects.get(azienda=azienda,user=user)
+        assegnazione = Assegnazione.objects.get(azienda=azienda.id,user=user.id)
       assegnazione.azienda=azienda
       assegnazione.user=User.objects.get(pk=AdminId)
 
       assegnazione.save()
+      del request.session['AziendaIdEdit']
       return HttpResponseRedirect(reverse('azienda:welcome'))
     else:
       print("form non valido")