102 lines
3.9 KiB
Python
102 lines
3.9 KiB
Python
from django.views.generic import ListView, CreateView, UpdateView, DeleteView, DetailView
|
|
from django.contrib.auth.mixins import LoginRequiredMixin , PermissionRequiredMixin
|
|
from Movement.Forms import FormsExpense
|
|
from Movement import models
|
|
from Base.models import ChartOfAccount
|
|
from django.urls import reverse_lazy
|
|
from django.contrib import messages
|
|
from django.utils import timezone
|
|
|
|
class ExpenseListView(LoginRequiredMixin, PermissionRequiredMixin, ListView):
|
|
model = models.Expense
|
|
template_name = 'Expense/List.html'
|
|
context_object_name = 'Expenses'
|
|
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.localtime(timezone.now()).month
|
|
)
|
|
return queryset
|
|
|
|
class ExpenseCreateView(LoginRequiredMixin,PermissionRequiredMixin, CreateView):
|
|
model = models.Expense
|
|
template_name = 'Expense/Create.html'
|
|
form_class = FormsExpense.FormsExpense
|
|
success_url = reverse_lazy('MovExpenseListView')
|
|
permission_required = 'Base.add_professional'
|
|
|
|
class ExpenseDetailView(LoginRequiredMixin,PermissionRequiredMixin, DetailView):
|
|
model = models.Expense
|
|
template_name = 'Expense/Detail.html'
|
|
permission_required = 'Base.view_professional'
|
|
|
|
class ExpenseUpdateView(LoginRequiredMixin,PermissionRequiredMixin, UpdateView):
|
|
model = models.Expense
|
|
template_name = 'Expense/Update.html'
|
|
form_class = FormsExpense.FormsExpense
|
|
success_url = reverse_lazy('ExpenseListView')
|
|
permission_required = 'Base.change_professional'
|
|
|
|
class ExpenseDeleteView(LoginRequiredMixin,PermissionRequiredMixin, DeleteView):
|
|
model = models.Expense
|
|
template_name = 'Expense/Delete.html'
|
|
success_url = reverse_lazy('MovExpenseListView')
|
|
permission_required = 'Base.delete_professional'
|
|
|
|
class ExpenseFirmCreateView(LoginRequiredMixin,PermissionRequiredMixin, CreateView):
|
|
model = models.Expense
|
|
template_name = 'Expense/CreateFirm.html'
|
|
form_class = FormsExpense.FormsExpense
|
|
success_url = reverse_lazy('MovExpenseListView')
|
|
permission_required = 'Base.add_professional'
|
|
|
|
def get_form(self, form_class=None):
|
|
form = super().get_form(form_class)
|
|
form.fields['chart_of_account'].queryset = ChartOfAccount.objects.filter(
|
|
firm=True,
|
|
debit=True,
|
|
)
|
|
return form
|
|
|
|
def form_valid(self, form):
|
|
form.instance.firm = True
|
|
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 ExpenseProfCreateView(LoginRequiredMixin,PermissionRequiredMixin, CreateView):
|
|
model = models.Expense
|
|
template_name = 'Expense/CreateProf.html'
|
|
form_class = FormsExpense.FormsExpense
|
|
success_url = reverse_lazy('MovExpenseListView')
|
|
permission_required = 'Base.add_professional'
|
|
|
|
def get_form(self, form_class=None):
|
|
form = super().get_form(form_class)
|
|
form.fields['chart_of_account'].queryset = ChartOfAccount.objects.filter(
|
|
firm=False,
|
|
debit=True,
|
|
)
|
|
return form
|
|
|
|
def form_valid(self, form):
|
|
form.instance.firm = False
|
|
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) |