prije gemini provjere
This commit is contained in:
mariomitte
@@ -0,0 +1,161 @@
|
||||
import random
|
||||
import os
|
||||
from django.core.management.base import BaseCommand
|
||||
from django.contrib.auth import get_user_model
|
||||
from django.utils import timezone
|
||||
from django.core.files import File
|
||||
from django.db import transaction
|
||||
from operations.models import RadniNalog, RadniNalogSlika, PutniNalog, BrojacSekvence
|
||||
from fleet.models import Vozilo, Stroj
|
||||
from kupci.models import Kupac
|
||||
from kalendar.services import kreiraj_kalendarski_unos
|
||||
|
||||
User = get_user_model()
|
||||
|
||||
class Command(BaseCommand):
|
||||
help = 'Popunjava bazu s Putnim nalozima, Radnim nalozima i slikama sinkronizirano sa sekvencama'
|
||||
|
||||
def add_arguments(self, parser):
|
||||
parser.add_argument('total_putovanja', type=int, nargs='?', default=3, help='Broj putnih naloga')
|
||||
parser.add_argument('--img', type=str, default='test_assets/kvar.jpg')
|
||||
|
||||
def handle(self, *args, **kwargs):
|
||||
total_putovanja = kwargs['total_putovanja']
|
||||
test_img_path = kwargs['img']
|
||||
|
||||
kupci = Kupac.objects.prefetch_related('strojevi').all()
|
||||
vozila = list(Vozilo.objects.all())
|
||||
serviseri = list(User.objects.filter(is_staff=True))
|
||||
|
||||
if not kupci.exists() or not vozila or not serviseri:
|
||||
self.stdout.write(self.style.ERROR("Baza mora imati kupce, vozila i servisere (is_staff=True)."))
|
||||
return
|
||||
|
||||
# Provjera slike
|
||||
if not os.path.exists(test_img_path):
|
||||
self.stdout.write(self.style.WARNING(f"Testna slika {test_img_path} nije pronađena. Generiram bez slika."))
|
||||
test_img_path = None
|
||||
|
||||
opisi_kvarova = [
|
||||
"Godišnji periodični servis dizalice.",
|
||||
"Zamjena hidrauličnog crijeva i dopuna ulja.",
|
||||
"Kalibracija senzora opterećenja.",
|
||||
"Popravak daljinskog upravljača.",
|
||||
"Hitna intervencija: Stroj u kvaru."
|
||||
]
|
||||
|
||||
self.stdout.write(f'🚀 Generiranje {total_putovanja} putovanja sinkronizirano sa sekvencama...')
|
||||
success_count_rn = 0
|
||||
trenutna_godina = timezone.now().year
|
||||
|
||||
# Definiramo ključ i prefikse za putne naloge
|
||||
prefix_pn = f"PN-{trenutna_godina}-"
|
||||
kljuc_sekvence_rn = f"radni_nalog_{trenutna_godina}"
|
||||
|
||||
for i in range(total_putovanja):
|
||||
serviser = random.choice(serviseri)
|
||||
vozilo = random.choice(vozila)
|
||||
|
||||
# 1. GENERIRANJE NEUNIŠTIVOG BROJA PUTNOG NALOGA
|
||||
# Da skripta ne bi udarila u unique=True, tražimo stvarni zadnji PN u bazi
|
||||
with transaction.atomic():
|
||||
zadnji_pn = PutniNalog.objects.select_for_update().filter(
|
||||
broj_naloga__startswith=prefix_pn
|
||||
).order_by('-broj_naloga').first()
|
||||
|
||||
if zadnji_pn:
|
||||
try:
|
||||
zadnji_broj_pn = int(zadnji_pn.broj_naloga.split('-')[-1])
|
||||
novi_broj_pn = zadnji_broj_pn + 1
|
||||
except (ValueError, IndexError):
|
||||
novi_broj_pn = PutniNalog.objects.count() + 1
|
||||
else:
|
||||
novi_broj_pn = 1
|
||||
|
||||
broj_pn = f"{prefix_pn}{novi_broj_pn:04d}"
|
||||
|
||||
# Kreiraj PUTNI NALOG
|
||||
putni_nalog = PutniNalog.objects.create(
|
||||
broj_naloga=broj_pn,
|
||||
vozilo=vozilo,
|
||||
korisnik=serviser,
|
||||
relacija="Zagreb - Teren - Zagreb",
|
||||
mjesto_odredista="Teren",
|
||||
vrijeme_polaska=timezone.now()
|
||||
)
|
||||
|
||||
self.stdout.write(self.style.MIGRATE_LABEL(f"Otvoren {broj_pn} (Vozilo: {vozilo.registracija})"))
|
||||
|
||||
# 2. Kreiraj 1 do 3 RADNA NALOGA za ovaj putni nalog
|
||||
za_odraditi = random.randint(1, 3)
|
||||
kreirani_radni_nalozi = []
|
||||
|
||||
for j in range(za_odraditi):
|
||||
klijenti_sa_strojima = [k for k in kupci if k.strojevi.exists()]
|
||||
if not klijenti_sa_strojima:
|
||||
continue
|
||||
klijent = random.choice(klijenti_sa_strojima)
|
||||
stroj = random.choice(list(klijent.strojevi.all()))
|
||||
|
||||
try:
|
||||
# Inicijaliziramo objekt bez broja_naloga kako bi se aktivirala
|
||||
# naša neuništiva sekvencijska logika unutar save() metode modela!
|
||||
nalog = RadniNalog(
|
||||
klijent=klijent,
|
||||
stroj=stroj,
|
||||
putni_nalog=putni_nalog,
|
||||
opis_kvara=random.choice(opisi_kvarova),
|
||||
izvrsitelj=serviser,
|
||||
status=RadniNalog.StatusRadaChoices.PLANIRANO
|
||||
)
|
||||
|
||||
# Prvo moramo izvršiti osnovni save da model kroz BrojacSekvence dodijeli čisti broj
|
||||
nalog.save()
|
||||
|
||||
# Ako slika postoji, dodajemo je NAKON što je broj kreiran (da putanja_slike_naloga ima ispravan folder)
|
||||
if test_img_path:
|
||||
with open(test_img_path, 'rb') as f:
|
||||
nalog.slika_kvara.save(f'rn_{nalog.broj_naloga}.jpg', File(f), save=True)
|
||||
|
||||
kreirani_radni_nalozi.append(nalog)
|
||||
|
||||
# Dodatne slike u galeriju
|
||||
if test_img_path:
|
||||
for k in range(random.randint(1, 2)):
|
||||
with open(test_img_path, 'rb') as f:
|
||||
RadniNalogSlika.objects.create(
|
||||
radni_nalog=nalog,
|
||||
slika=File(f, name=f'gal_{nalog.broj_naloga}_{k}.jpg'),
|
||||
opis=f"Detalj kvara {k+1}"
|
||||
)
|
||||
|
||||
# 5. Upis u kalendar
|
||||
try:
|
||||
kreiraj_kalendarski_unos(nalog)
|
||||
except Exception as cal_err:
|
||||
self.stdout.write(self.style.WARNING(f" ⚠️ Kalendar unos preskočen: {cal_err}"))
|
||||
|
||||
success_count_rn += 1
|
||||
self.stdout.write(f" ✅ {nalog.broj_naloga} uspješno povezan u {broj_pn}")
|
||||
|
||||
except Exception as e:
|
||||
self.stderr.write(f" ❌ Greška kod kreiranja RN: {e}")
|
||||
|
||||
# 3. NASUMIČNO ZATVARANJE (Simulacija završenog posla)
|
||||
if random.random() > 0.5 and kreirani_radni_nalozi:
|
||||
try:
|
||||
# Prvo moramo zatvoriti sve radne naloge da bi PN prošao validaciju
|
||||
for rn in kreirani_radni_nalozi:
|
||||
rn.status = RadniNalog.StatusRadaChoices.ZAVRSENO
|
||||
rn.save()
|
||||
|
||||
# Dodajemo završne kilometre (uvijek više od početnih)
|
||||
putni_nalog.zavrsna_km = putni_nalog.pocetna_km + random.randint(50, 500)
|
||||
putni_nalog.status = PutniNalog.StatusChoices.ZAVRSEN
|
||||
putni_nalog.save()
|
||||
|
||||
self.stdout.write(self.style.SUCCESS(f" 🏁 {broj_pn} automatski ZATVOREN. Kilometri vozila ažurirani."))
|
||||
except Exception as e:
|
||||
self.stdout.write(self.style.WARNING(f" ⚠️ PN-{broj_pn} nije zatvoren: {e}"))
|
||||
|
||||
self.stdout.write(self.style.SUCCESS(f'\nGotovo! Generirano {total_putovanja} PN i {success_count_rn} RN.'))
|
||||
Reference in New Issue
Block a user