37 lines
1.6 KiB
Python
37 lines
1.6 KiB
Python
# users/services.py
|
|
from django.shortcuts import get_object_or_404
|
|
from django.core.exceptions import PermissionDenied
|
|
from .models import CustomUser
|
|
from operations.models import RadniNalog, PutniNalog
|
|
from fleet.models import Vozilo
|
|
|
|
def dohvati_operativne_podatke_servisera(user_id: int) -> dict:
|
|
"""
|
|
Prikuplja sve povezane resurse (radne naloge, putne naloge i vozila)
|
|
dodijeljene specifičnom serviseru.
|
|
"""
|
|
# 1. Dohvaćamo korisnika
|
|
korisnik = get_object_or_404(CustomUser, id=user_id)
|
|
|
|
# 2. Sigurnosni ček na razini poslovne logike
|
|
if getattr(korisnik, 'uloga', '').upper() != 'SERVISER':
|
|
raise PermissionDenied("Odabrani korisnik nema operativne ovlasti servisera.")
|
|
|
|
# 3. Dohvat povezanih entiteta prateći STVARNA polja iz tvoje baze podataka:
|
|
|
|
# Radni nalozi gdje je korisnik postavljen kao izvršitelj
|
|
radni_nalozi = RadniNalog.objects.filter(izvrsitelj=korisnik).select_related('klijent', 'stroj')
|
|
|
|
# 🚀 POPRAVAK: 'radni_nalozi__in' umjesto 'radni_nalog__in' (usklađivanje s related_name na RadniNalog modelu)
|
|
putni_nalozi = PutniNalog.objects.filter(radni_nalozi__in=radni_nalozi).select_related('vozilo')
|
|
|
|
# Izvlačimo jedinstvena vozila koja serviser vozi kroz svoje aktivne putne naloge
|
|
povezana_vozila_ids = putni_nalozi.values_list('vozilo_id', flat=True).distinct()
|
|
povezana_vozila = Vozilo.objects.filter(id__in=povezana_vozila_ids)
|
|
|
|
return {
|
|
"serviser": korisnik,
|
|
"radni_nalozi": radni_nalozi,
|
|
"putni_nalozi": putni_nalozi,
|
|
"vozila": povezana_vozila
|
|
} |