diff --git a/app/itam/views/software_category.py b/app/itam/views/software_category.py index 6c416f4e..895447c0 100644 --- a/app/itam/views/software_category.py +++ b/app/itam/views/software_category.py @@ -5,6 +5,9 @@ from access.mixin import OrganizationPermission from ..models.software import Software, SoftwareCategory +from settings.models.user_settings import UserSettings + + class IndexView(PermissionRequiredMixin, OrganizationPermission, generic.ListView): model = Software @@ -68,6 +71,13 @@ class Add(PermissionRequiredMixin, OrganizationPermission, generic.CreateView): ] + def get_initial(self): + + return { + 'organization': UserSettings.objects.get(user = self.request.user).default_organization + } + + def get_success_url(self, **kwargs): return f"/settings/software_category"