Start content

shimatta_kenkyusho/parts/views.py

@@ -1,20 +1,128 @@
 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
 
-def main_view(request):
+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('Main', request.user),
-        'title': NavBar.get_brand()+' / '+'Main',
+        '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/main.html', context)
+
+    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