Fix Authorized User bugs (#3146)

Co-authored-by: Maffooch <[email protected]>

Author: Maffooch

dojo/product_type/views.py

@@ -87,12 +87,14 @@ def add_product_type(request):
 @user_passes_test(lambda u: u.is_staff)
 def edit_product_type(request, ptid):
     pt = get_object_or_404(Product_Type, pk=ptid)
-    pt_form = Product_TypeForm(instance=pt)
+    authed_users = pt.authorized_users.all()
+    pt_form = Product_TypeForm(instance=pt, initial={'authorized_users': authed_users})
     delete_pt_form = Delete_Product_TypeForm(instance=pt)
     if request.method == "POST" and request.POST.get('edit_product_type'):
         pt_form = Product_TypeForm(request.POST, instance=pt)
         if pt_form.is_valid():
             pt = pt_form.save()
+            pt.authorized_users.set(pt_form.cleaned_data['authorized_users'])
             messages.add_message(
                 request,
                 messages.SUCCESS,

dojo/user/views.py

@@ -310,6 +310,12 @@ def edit_user(request, uid):
 
         if form.is_valid() and contact_form.is_valid():
             form.save()
+            for init_auth_prods in authed_products:
+                init_auth_prods.authorized_users.remove(user)
+                init_auth_prods.save()
+            for init_auth_prod_types in authed_product_types:
+                init_auth_prod_types.authorized_users.remove(user)
+                init_auth_prod_types.save()
             if 'authorized_products' in form.cleaned_data and len(form.cleaned_data['authorized_products']) > 0:
                 for p in form.cleaned_data['authorized_products']:
                     p.authorized_users.add(user)