216 lines
8.5 KiB
Python
216 lines
8.5 KiB
Python
from rest_framework.decorators import action
|
|
from django.utils import timezone
|
|
from rest_framework import viewsets, filters, permissions, status
|
|
from django_filters.rest_framework import DjangoFilterBackend
|
|
from .models import RadniNalog, RadniNalogSlika, PutniNalog
|
|
from rest_framework.response import Response
|
|
from .serializers import (
|
|
RadniNalogSerializer,
|
|
RadniNalogListaSerializer,
|
|
RadniNalogDetaljiSerializer,
|
|
KreirajPutniNalogSerializer,
|
|
RadniNalogSlikaSerializer
|
|
)
|
|
from kalendar.services import kreiraj_kalendarski_unos
|
|
|
|
class RadniNalogViewSet(viewsets.ModelViewSet):
|
|
"""
|
|
API endpoint za upravljanje radnim nalozima.
|
|
Vozilo se sada filtrira i pretražuje preko Putnog Naloga.
|
|
"""
|
|
queryset = RadniNalog.objects.all().select_related(
|
|
'klijent', 'izvrsitelj', 'putni_nalog__vozilo'
|
|
)
|
|
# permission_classes = [permissions.IsAuthenticated]
|
|
# permission_classes = [permissions.AllowAny]
|
|
|
|
serializer_class = RadniNalogSerializer
|
|
|
|
filter_backends = [
|
|
DjangoFilterBackend,
|
|
filters.SearchFilter,
|
|
filters.OrderingFilter
|
|
]
|
|
|
|
filterset_fields = ['status', 'izvrsitelj', 'klijent', 'putni_nalog__vozilo']
|
|
|
|
search_fields = [
|
|
'broj_naloga',
|
|
'klijent__naziv',
|
|
'putni_nalog__vozilo__registracija',
|
|
'putni_nalog__vozilo__naziv'
|
|
]
|
|
|
|
ordering_fields = ['datum_kreiranja', 'status', 'broj_naloga']
|
|
ordering = ['-datum_kreiranja']
|
|
|
|
def partial_update(self, request, *args, **kwargs):
|
|
# 1. Izvrši standardni update
|
|
response = super().partial_update(request, *args, **kwargs)
|
|
|
|
# 2. Dohvati svježu instancu nakon spremanja
|
|
instance = self.get_object()
|
|
|
|
# 3. Koristi detaljni serializer kako bi Astro dobio sve podatke (slike, vozilo, klijent)
|
|
serializer = RadniNalogDetaljiSerializer(instance, context={'request': request})
|
|
|
|
return Response(serializer.data)
|
|
|
|
def get_serializer_class(self):
|
|
if self.action == 'list':
|
|
return RadniNalogListaSerializer
|
|
if self.action == 'retrieve':
|
|
return RadniNalogDetaljiSerializer
|
|
return RadniNalogSerializer
|
|
|
|
def get_queryset(self):
|
|
queryset = self.queryset
|
|
if self.action == 'retrieve':
|
|
queryset = queryset.prefetch_related('slike')
|
|
return queryset
|
|
|
|
def perform_create(self, serializer):
|
|
if not serializer.validated_data.get('izvrsitelj'):
|
|
radni_nalog = serializer.save(izvrsitelj=self.request.user)
|
|
else:
|
|
radni_nalog = serializer.save()
|
|
|
|
try:
|
|
kreiraj_kalendarski_unos(radni_nalog)
|
|
except Exception as e:
|
|
print(f"Greška pri kreiranju kalendarskog zapisa: {e}")
|
|
|
|
# Uredi isključivo ovu metodu unutar RadniNalogViewSet u tvom views.py
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
serializer = self.get_serializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
|
|
if request.data.get('kreiraj_putni') == 'true':
|
|
# Kreiraj novi putni nalog (moraš imati vozilo definirano!)
|
|
# Pretpostavka: 'stroj' je poslan u requestu, pa iz njega dobijemo vozilo ako postoji
|
|
# Ovdje prilagodi logiku prema tome kako tvoj model PutniNalog povezuje vozilo
|
|
novi_putni = PutniNalog.objects.create(
|
|
broj_naloga=f"PN-{timezone.now().year}-AUTO", # Ili tvoja logika za broj
|
|
vozilo=serializer.validated_data['stroj'].vozilo, # Primjer povezivanja
|
|
korisnik=request.user,
|
|
relacija="Automatski kreirano iz radnog naloga"
|
|
)
|
|
# Spremi nalog s ovim novim putnim nalogom
|
|
instance = serializer.save(putni_nalog=novi_putni)
|
|
else:
|
|
# Standardni put
|
|
self.perform_create(serializer)
|
|
instance = serializer.instance
|
|
|
|
instance = serializer.instance
|
|
|
|
# Obrada višestrukih slika
|
|
files = request.FILES.getlist('slike')
|
|
for f in files:
|
|
RadniNalogSlika.objects.create(radni_nalog=instance, slika=f)
|
|
|
|
# KLJUČNI POPRAVAK: Ponovno osvježavamo instancu iz baze podataka
|
|
# Ovo povlači čisti 'broj_naloga' nakon što ga je save() u models.py očistio od RN-RN-
|
|
instance.refresh_from_db()
|
|
|
|
# Za povratni JSON koristimo detaljniji prikaz kako bi Astro odmah dobio ispravno stanje
|
|
izlazni_serializer = RadniNalogDetaljiSerializer(instance, context={'request': request})
|
|
|
|
headers = self.get_success_headers(izlazni_serializer.data)
|
|
return Response(izlazni_serializer.data, status=status.HTTP_201_CREATED, headers=headers)
|
|
|
|
@action(detail=False, methods=['get'], url_path='sljedeci-broj', permission_classes=[permissions.AllowAny])
|
|
def dohvati_sljedeci_broj(self, request):
|
|
"""
|
|
Dohvaća točan sljedeći broj radnog naloga iz neuništive sekvencijske tablice.
|
|
"""
|
|
godina = timezone.now().year
|
|
kljuc_sekvence = f"radni_nalog_{godina}"
|
|
prefix = f"RN-{godina}-"
|
|
|
|
# Pogledaj trenutačno stanje u tablici sekvenci (bez uvećavanja)
|
|
from .models import BrojacSekvence # Lokalni uvoz ako zatreba
|
|
|
|
brojac = BrojacSekvence.objects.filter(naziv_sekvence=kljuc_sekvence).first()
|
|
|
|
if brojac:
|
|
sljedeci_broj = brojac.zadnji_broj + 1
|
|
else:
|
|
# Ako brojač još ne postoji u sekvencama, baci oko na bazu za prvu inicijalizaciju
|
|
zadnji_nalog = RadniNalog.objects.filter(broj_naloga__startswith=prefix).order_by('-broj_naloga').first()
|
|
if zadnji_nalog:
|
|
try:
|
|
sljedeci_broj = int(zadnji_nalog.broj_naloga.split('-')[-1]) + 1
|
|
except (ValueError, IndexError):
|
|
sljedeci_broj = 1
|
|
else:
|
|
sljedeci_broj = 1
|
|
|
|
sljedeci_broj_formatiran = f"{prefix}{sljedeci_broj:04d}"
|
|
return Response({"broj_naloga": sljedeci_broj_formatiran})
|
|
|
|
class PutniNalogViewSet(viewsets.ModelViewSet):
|
|
"""
|
|
ViewSet za upravljanje putnim nalozima.
|
|
Podržava izradu novog putnog naloga iz radnog naloga (POST) i pregled (GET).
|
|
"""
|
|
queryset = PutniNalog.objects.all().order_by('-datum_izdavanja', '-id')
|
|
# permission_classes = [permissions.IsAuthenticated] # Preporučeno: Otkomentarisano radi getAuthHeaders
|
|
permission_classes = [permissions.AllowAny]
|
|
|
|
def get_serializer_class(self):
|
|
if self.action == 'create':
|
|
return KreirajPutniNalogSerializer
|
|
return KreirajPutniNalogSerializer
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
serializer = self.get_serializer_class()(data=request.data, context={'request': request})
|
|
|
|
if serializer.is_valid():
|
|
putni_nalog = serializer.save()
|
|
return Response(
|
|
{
|
|
"id": putni_nalog.id,
|
|
"broj_naloga": putni_nalog.broj_naloga,
|
|
"status": putni_nalog.status,
|
|
"pocetna_km": putni_nalog.pocetna_km,
|
|
"message": "Putni nalog uspješno kreiran!"
|
|
},
|
|
status=status.HTTP_201_CREATED
|
|
)
|
|
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
|
|
class RadniNalogSlikaViewSet(viewsets.ModelViewSet):
|
|
queryset = RadniNalogSlika.objects.all()
|
|
serializer_class = RadniNalogSlikaSerializer
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
# permission_classes = [permissions.AllowAny]
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
# 1. Dohvati listu datoteka (mora se podudarati s imenom u FormData)
|
|
files = request.FILES.getlist('slika')
|
|
|
|
# 2. Dohvati nalog_id iz request.data
|
|
radni_nalog_id = request.data.get('radni_nalog')
|
|
|
|
if not files:
|
|
# Ako nema datoteka, pokušaj standardno kreiranje
|
|
return super().create(request, *args, **kwargs)
|
|
|
|
created_instances = []
|
|
for file in files:
|
|
# 3. Ručno kreiraj instancu za svaku sliku
|
|
# Napomena: 'slika' i 'radni_nalog' su imena polja u tvom modelu
|
|
instance = RadniNalogSlika.objects.create(
|
|
radni_nalog_id=radni_nalog_id,
|
|
slika=file
|
|
)
|
|
created_instances.append(instance)
|
|
|
|
return Response(
|
|
{"status": f"Uspješno dodano {len(created_instances)} slika"},
|
|
status=status.HTTP_201_CREATED
|
|
) |