Fixup imports of distributor views
This commit is contained in:
parent
b3f8041f08
commit
1f3ed7f8ed
@ -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)
|
||||||
|
Loading…
Reference in New Issue
Block a user