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 from django.core.paginator import Paginator 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' default_pagination_size = 25 def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) storage_page = self.request.GET.get('storage_page') if storage_page is None: storage_page = 1 low_stock_page = self.request.GET.get('low_stock_page') if low_stock_page is None: low_stock_page = 1 storage_paginator = Paginator(Storage.objects.filter(parent_storage=None), self.default_pagination_size) low_stock_paginator = Paginator(Stock.get_under_watermark(), self.default_pagination_size) context['low_stocks'] = low_stock_paginator.get_page(low_stock_page) context['storages'] = storage_paginator.get_page(storage_page) return context class StockViewDetail(LoginRequiredMixin, BaseTemplateMixin, DetailView): template_name = 'parts/stocks-detail.html' model = Storage pk_url_kwarg = 'uuid' base_title = '' navbar_selected = 'Stocks' default_pagination_size = 8 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() storage_page = self.request.GET.get('storage_page') if storage_page is None: storage_page = 1 storage_paginator = Paginator(Storage.objects.filter(parent_storage=self.object), self.default_pagination_size) context['storages'] = storage_paginator.get_page(storage_page) return context