Fixup imports of distributor views

This commit is contained in:
Mario Hüttel 2024-11-18 21:15:05 +01:00
parent b3f8041f08
commit 1f3ed7f8ed

View File

@ -1,20 +1,12 @@
from django.shortcuts import render, redirect import uuid
from django.urls import reverse from django.shortcuts import redirect
from django.contrib.auth.forms import AuthenticationForm as AuthForm
from django.contrib.auth.forms import PasswordChangeForm
from django.contrib.auth import update_session_auth_hash
import django.forms as forms
from django.views.generic import TemplateView, DetailView from django.views.generic import TemplateView, DetailView
from django.contrib.auth.mixins import LoginRequiredMixin from django.contrib.auth.mixins import LoginRequiredMixin
from ..models import Storage, Stock, Component, Distributor, Manufacturer, Package, ComponentParameter, DistributorNum from ..models import Distributor
from django.core.paginator import Paginator from django.core.paginator import Paginator
from django.core.exceptions import ValidationError
from django.db import IntegrityError
from django.db.models import ProtectedError from django.db.models import ProtectedError
from ..forms import * from ..forms import *
from django.db.models import Q from django.db.models import Q
from django.db.models.functions import Lower
import uuid
from .generic_views import BaseTemplateMixin from .generic_views import BaseTemplateMixin
class DistributorView(LoginRequiredMixin, BaseTemplateMixin, TemplateView): class DistributorView(LoginRequiredMixin, BaseTemplateMixin, TemplateView):
@ -28,7 +20,7 @@ class DistributorView(LoginRequiredMixin, BaseTemplateMixin, TemplateView):
if not search: if not search:
return qs return qs
search_fragments = search.strip().split() search_fragments = search.strip().split()
for search in search_fragments: for search in search_fragments:
@ -39,10 +31,10 @@ class DistributorView(LoginRequiredMixin, BaseTemplateMixin, TemplateView):
context = super().get_context_data(**kwargs) context = super().get_context_data(**kwargs)
page_num = self.request.GET.get('page', default=1) page_num = self.request.GET.get('page', default=1)
search_string = self.request.GET.get('search', default=None) search_string = self.request.GET.get('search', default=None)
queryset = self.search_distributors(search_string) queryset = self.search_distributors(search_string)
paginator = Paginator(queryset, self.default_page_size) paginator = Paginator(queryset, self.default_page_size)
context['search_string'] = search_string context['search_string'] = search_string
@ -50,7 +42,7 @@ class DistributorView(LoginRequiredMixin, BaseTemplateMixin, TemplateView):
context['new_distri_form'] = DistributorForm() context['new_distri_form'] = DistributorForm()
return context return context
def handle_add_new_distributor(self, request): def handle_add_new_distributor(self, request):
form = DistributorForm(data=request.POST, files=request.FILES) form = DistributorForm(data=request.POST, files=request.FILES)
@ -77,19 +69,19 @@ class DistributorDetailView(LoginRequiredMixin, BaseTemplateMixin, DetailView):
pk_url_kwarg = 'uuid' pk_url_kwarg = 'uuid'
base_title = '' base_title = ''
navbar_selected = 'Distributors' navbar_selected = 'Distributors'
def get_context_data(self, **kwargs): def get_context_data(self, **kwargs):
self.base_title = 'Distributor / '+self.object.name self.base_title = 'Distributor / '+self.object.name
context = super().get_context_data(**kwargs) context = super().get_context_data(**kwargs)
context['distributor'] = self.object context['distributor'] = self.object
context['edit_form'] = DistributorForm(instance=self.object) context['edit_form'] = DistributorForm(instance=self.object)
return context return context
def handle_delete_distributor(self, request): def handle_delete_distributor(self, request):
delete_error = None delete_error = None
protected_objects = None protected_objects = None
# Try to delete this instance # Try to delete this instance
try: try:
self.object.delete() self.object.delete()
@ -98,7 +90,7 @@ class DistributorDetailView(LoginRequiredMixin, BaseTemplateMixin, DetailView):
protected_objects = pe.protected_objects protected_objects = pe.protected_objects
except: except:
delete_error = 'Cannot delete this distributor. Unknown error' delete_error = 'Cannot delete this distributor. Unknown error'
if delete_error: if delete_error:
context = self.get_context_data() context = self.get_context_data()
context['delete_error'] = delete_error context['delete_error'] = delete_error
@ -125,5 +117,5 @@ class DistributorDetailView(LoginRequiredMixin, BaseTemplateMixin, DetailView):
return self.handle_delete_distributor(request) return self.handle_delete_distributor(request)
elif 'submit-distri-edit' in request.POST: elif 'submit-distri-edit' in request.POST:
return self.edit_distributor(request) return self.edit_distributor(request)
return super().post(request, *args, **kwargs) return super().post(request, *args, **kwargs)