129 lines
4.0 KiB
Python
129 lines
4.0 KiB
Python
from django.shortcuts import render, redirect
|
|
from django.urls import resolve, reverse
|
|
from django.contrib.auth import logout, login
|
|
from django.http import HttpResponse
|
|
from .navbar import NavBar
|
|
from django.contrib.auth.forms import AuthenticationForm as AuthForm
|
|
from django.views import View
|
|
import django.forms as forms
|
|
from django.views.generic import TemplateView, DetailView
|
|
from django.contrib.auth.mixins import LoginRequiredMixin, PermissionRequiredMixin
|
|
from .models import Storage, Stock
|
|
from .qr_parser import QrCodeValidator
|
|
|
|
class QrSearchForm(forms.Form):
|
|
my_qr_validator = QrCodeValidator()
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
|
|
|
|
qr_search = forms.CharField(label='qr_search', validators=[my_qr_validator])
|
|
|
|
class BaseTemplateMixin(object):
|
|
navbar_selected = ''
|
|
base_title = ''
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
base_context = {
|
|
'navbar': NavBar.get_navbar(self.navbar_selected, self.request.user),
|
|
'title': NavBar.get_brand()+' / '+ self.base_title,
|
|
'login_active': False,
|
|
}
|
|
context['base'] = base_context
|
|
return context
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
data = request.POST
|
|
if 'qr_search' not in data:
|
|
super().post(request, *args, **kwargs)
|
|
|
|
print('QR',data['qr_search'])
|
|
f = QrSearchForm(data)
|
|
if f.is_valid():
|
|
return redirect(f.my_qr_validator.get_redirect_url(f.cleaned_data['qr_search']))
|
|
|
|
return self.get(request)
|
|
|
|
class MainView(BaseTemplateMixin, TemplateView):
|
|
template_name = 'parts/main.html'
|
|
navbar_selected = 'Main'
|
|
base_title = 'Main'
|
|
|
|
def logout_view(request):
|
|
logout(request)
|
|
return redirect('parts-main')
|
|
|
|
def login_view(request):
|
|
base_context = {
|
|
'navbar': NavBar.get_navbar('Login', request.user),
|
|
'title': NavBar.get_brand()+' / '+'Login',
|
|
'login_active': True,
|
|
}
|
|
|
|
if request.user.is_authenticated:
|
|
next_param = request.GET.get('next')
|
|
if next_param is not None:
|
|
return redirect(next_param)
|
|
else:
|
|
return redirect('parts-main')
|
|
|
|
if request.method == 'POST':
|
|
form = AuthForm(data=request.POST)
|
|
if form.is_valid():
|
|
valid_user = form.get_user()
|
|
login(request, valid_user)
|
|
next_param = request.GET.get('next')
|
|
if next_param is not None:
|
|
return redirect(next_param)
|
|
else:
|
|
return redirect('parts-main')
|
|
else:
|
|
form = AuthForm()
|
|
|
|
|
|
context = {
|
|
'base': base_context,
|
|
'form': form,
|
|
}
|
|
|
|
return render(request, 'parts/login.html', context)
|
|
|
|
# Create your views here.
|
|
|
|
class ComponentView(LoginRequiredMixin, BaseTemplateMixin, TemplateView):
|
|
template_name = 'parts/components.html'
|
|
base_title = 'Components'
|
|
navbar_selected = 'Components'
|
|
|
|
class StockView(LoginRequiredMixin, BaseTemplateMixin, TemplateView):
|
|
template_name = 'parts/stocks.html'
|
|
base_title = 'Stocks'
|
|
navbar_selected = 'Stocks'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context['low_stocks'] = Stock.get_under_watermark()
|
|
context['storages'] = Storage.objects.filter(parent_storage=None)
|
|
return context
|
|
|
|
class StockViewDetail(LoginRequiredMixin, BaseTemplateMixin, DetailView):
|
|
template_name = 'parts/stocks-detail.html'
|
|
model = Storage
|
|
pk_url_kwarg = 'uuid'
|
|
base_title = ''
|
|
navbar_selected = 'Stocks'
|
|
|
|
def get_breadcrumbs(self):
|
|
crumbs = self.object.get_path_components()
|
|
# Reverse list and drop the last element of the reversed list
|
|
crumbs = crumbs[::-1][:-1]
|
|
return crumbs
|
|
|
|
def get_context_data(self, **kwargs):
|
|
self.base_title = 'Stocks / ' + self.object.name
|
|
context = super().get_context_data(**kwargs)
|
|
context['breadcrumbs'] = self.get_breadcrumbs()
|
|
return context
|