60 lines
2.4 KiB
Python
60 lines
2.4 KiB
Python
from django.views.generic import ListView, CreateView, UpdateView, DeleteView, DetailView
|
|
from django.contrib.auth.mixins import LoginRequiredMixin , PermissionRequiredMixin
|
|
from Movement.Forms import FormsStock
|
|
from Movement import models
|
|
from django.urls import reverse_lazy
|
|
from django.contrib import messages
|
|
from django.utils import timezone
|
|
|
|
class StockListView(LoginRequiredMixin, PermissionRequiredMixin, ListView):
|
|
model = models.Stock
|
|
template_name = 'Stock/List.html'
|
|
context_object_name = 'Stocks'
|
|
paginate_by = 10
|
|
permission_required = 'Base.view_professional'
|
|
def get_queryset(self):
|
|
queryset = super().get_queryset()
|
|
date = self.request.GET.get('date')
|
|
if date:
|
|
queryset = queryset.filter(date__icontains=date)
|
|
else:
|
|
queryset = queryset.filter(date__icontains=timezone.localdate())
|
|
return queryset
|
|
|
|
class StockCreateView(LoginRequiredMixin,PermissionRequiredMixin, CreateView):
|
|
model = models.Stock
|
|
template_name = 'Stock/Create.html'
|
|
form_class = FormsStock.FormsStock
|
|
success_url = reverse_lazy('MovStockListView')
|
|
permission_required = 'Base.add_professional'
|
|
|
|
def form_valid(self, form):
|
|
gross_value = form.instance.gross_value
|
|
quantity = form.instance.quantity
|
|
form.instance.unit_value = gross_value / quantity
|
|
form.instance.created_by = self.request.user
|
|
messages.success(self.request, 'Registro realizado com sucesso!')
|
|
return super().form_valid(form)
|
|
|
|
def form_invalid(self, form):
|
|
errors = form.errors.as_text()
|
|
messages.error(self.request, f'Verifique os dados: {errors}')
|
|
return super().form_invalid(form)
|
|
|
|
class StockDetailView(LoginRequiredMixin,PermissionRequiredMixin, DetailView):
|
|
model = models.Stock
|
|
template_name = 'Stock/Detail.html'
|
|
permission_required = 'Base.view_professional'
|
|
|
|
class StockUpdateView(LoginRequiredMixin,PermissionRequiredMixin, UpdateView):
|
|
model = models.Stock
|
|
template_name = 'Stock/Update.html'
|
|
form_class = FormsStock.FormsStock
|
|
success_url = reverse_lazy('MovStockListView')
|
|
permission_required = 'Base.change_professional'
|
|
|
|
class StockDeleteView(LoginRequiredMixin,PermissionRequiredMixin, DeleteView):
|
|
model = models.Stock
|
|
template_name = 'Stock/Delete.html'
|
|
success_url = reverse_lazy('MovStockListView')
|
|
permission_required = 'Base.delete_professional' |