57 lines
2.8 KiB
Python
57 lines
2.8 KiB
Python
from django.shortcuts import redirect
|
||
from django.urls import reverse_lazy
|
||
from django.views.generic import TemplateView, ListView, UpdateView
|
||
from django.contrib.auth.mixins import LoginRequiredMixin
|
||
from .models import Item # Проверь, как точно называется твоя модель деталей/заказов
|
||
|
||
# Класс главной страницы (роутер)
|
||
class IndexView(TemplateView):
|
||
template_name = 'shiftflow/landing.html'
|
||
|
||
def get(self, request, *args, **kwargs):
|
||
# Если юзер авторизован — сразу отправляем его в реестр
|
||
if request.user.is_authenticated:
|
||
return redirect('registry')
|
||
# Если нет — показываем кнопку "Войти"
|
||
return super().get(request, *args, **kwargs)
|
||
|
||
# Класс реестра деталей (защищен LoginRequiredMixin)
|
||
class RegistryView(LoginRequiredMixin, ListView):
|
||
model = Item
|
||
template_name = 'shiftflow/registry.html'
|
||
context_object_name = 'items'
|
||
|
||
def get_queryset(self):
|
||
# Позже здесь добавим: .filter(machine__in=request.user.profile.machines.all())
|
||
# Сортируем: сначала статус (по алфавиту или логике choices),
|
||
# затем по дате (свежие сверху), по станку и по номеру сделки
|
||
return Item.objects.all().order_by('status', '-date', 'machine__name', 'deal__number')
|
||
|
||
def get_context_data(self, **kwargs):
|
||
context = super().get_context_data(**kwargs)
|
||
# Передаем роль в шаблон, чтобы скрывать/показывать кнопки
|
||
if hasattr(self.request.user, 'profile'):
|
||
context['user_role'] = self.request.user.profile.role
|
||
return context
|
||
|
||
# Вьюха детального вида и редактирования
|
||
class ItemUpdateView(LoginRequiredMixin, UpdateView):
|
||
model = Item
|
||
template_name = 'shiftflow/item_detail.html'
|
||
# Перечисляем поля, которые можно редактировать (укажи нужные)
|
||
fields = [
|
||
'drawing_name', 'machine', 'quantity_plan', 'quantity_fact',
|
||
'material', 'size_value', 'status', 'is_synced_1c', 'extra_drawing'
|
||
]
|
||
context_object_name = 'item'
|
||
|
||
def get_context_data(self, **kwargs):
|
||
context = super().get_context_data(**kwargs)
|
||
# Обязательно добавляем роль в контекст этого шаблона!
|
||
if hasattr(self.request.user, 'profile'):
|
||
context['user_role'] = self.request.user.profile.role
|
||
return context
|
||
|
||
def get_success_url(self):
|
||
# После сохранения возвращаемся в реестр
|
||
return reverse_lazy('registry') |