improve dicom uploading

rad/views.py

@@ -600,7 +600,7 @@ class UpdateUserView(CidManagerRequiredMixin, UpdateView):
         return reverse(view_name, kwargs={"slug": self.object.username})
 
 
-class UpdateUserProfileView(CidManagerRequiredMixin, UpdateView):
+class UpdateUserProfileAdminView(CidManagerRequiredMixin, UpdateView):
     model = UserProfile
     fields = [
         "supervisor",
@@ -617,6 +617,34 @@ class UpdateUserProfileView(CidManagerRequiredMixin, UpdateView):
         # No need for reverse_lazy here, because it's called inside the method
         return reverse(view_name, kwargs={"slug": self.object.username})
 
+class UpdateUserProfileView(UpdateView):
+    model = UserProfile
+    fields = [
+        "supervisor",
+        "grade",
+        "registration_number",
+        #"peninsula_trainee",
+    ]  # Keep listing whatever fields
+    template_name = "user_update_profile.html"
+    slug_field = "user__username"
+    slug_url_kwarg = "slug"
+
+    def dispatch(self, request, *args, **kwargs):
+        if request.user.is_superuser or request.user.groups.filter(name="cid_user_manager").exists():
+            self.fields = ["supervisor", "grade", "registration_number", "peninsula_trainee"]
+        # Check permissions for the request.user here
+        elif request.user.userprofile.peninsula_trainee:
+            self.fields = ["supervisor", "grade", "registration_number"]
+
+        else:
+            self.fields = ["grade", "registration_number"]
+        return super().dispatch(request, *args, **kwargs)
+
+
+    def get_success_url(self):
+        view_name = "account_profile"
+        # No need for reverse_lazy here, because it's called inside the method
+        return reverse(view_name, kwargs={"slug": self.object.username})
 
 # class UpdateUser(TemplateView):
 #