121 lines
4.7 KiB
Python
Executable File
121 lines
4.7 KiB
Python
Executable File
from fastapi import FastAPI, HTTPException
|
|
from fastapi.staticfiles import StaticFiles
|
|
from fastapi.responses import FileResponse
|
|
from pydantic import BaseModel, validator
|
|
from typing import Optional
|
|
from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
|
|
from sqlalchemy.orm import sessionmaker
|
|
from sqlalchemy import text
|
|
import os
|
|
from datetime import date
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
# --- 1. ADATBÁZIS KAPCSOLAT ---
|
|
raw_url = os.getenv("DATABASE_URL")
|
|
if not raw_url:
|
|
raw_url = "postgresql://admin:PASSWORD_111@postgres-db:5432/service_finder"
|
|
fixed_url = raw_url.replace("postgresql://", "postgresql+asyncpg://").replace("/service_finder_db", "/service_finder")
|
|
|
|
engine = create_async_engine(fixed_url)
|
|
AsyncSessionLocal = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
|
|
|
|
app = FastAPI(title="Service Finder API")
|
|
|
|
# --- 2. ADATMODELLEK (VALIDÁCIÓVAL) ---
|
|
class VehicleRegister(BaseModel):
|
|
model_id: int
|
|
vin: str # Alvázszám (KÖTELEZŐ!)
|
|
plate: str # Rendszám
|
|
mileage: int # Km óra
|
|
purchase_date: date
|
|
role: str = "OWNER" # Alapértelmezett: Tulajdonos
|
|
|
|
# Adatőr: Automatikus nagybetűsítés és tisztítás
|
|
@validator('vin')
|
|
def clean_vin(cls, v):
|
|
if len(v) < 5: raise ValueError('Az alvázszám túl rövid!')
|
|
return v.upper().replace("-", "").replace(" ", "")
|
|
|
|
@validator('plate')
|
|
def clean_plate(cls, v):
|
|
return v.upper().replace("-", "").replace(" ", "")
|
|
|
|
@validator('mileage')
|
|
def positive_mileage(cls, v):
|
|
if v < 0: raise ValueError('A kilométer nem lehet negatív!')
|
|
return v
|
|
|
|
# --- 3. VÉGPONTOK ---
|
|
|
|
# Lista lekérése (Katalógus)
|
|
@app.get("/api/vehicles")
|
|
async def get_vehicles():
|
|
async with AsyncSessionLocal() as session:
|
|
result = await session.execute(text("""
|
|
SELECT vm.id, m.name as brand, vm.model_name, vm.category
|
|
FROM ref.vehicle_models vm
|
|
JOIN ref.vehicle_makes m ON vm.make_id = m.id
|
|
ORDER BY m.name, vm.model_name
|
|
"""))
|
|
return [{"id": r.id, "brand": r.brand, "model": r.model_name, "category": r.category} for r in result.fetchall()]
|
|
|
|
# ÚJ: JÁRMŰ REGISZTRÁCIÓ (A Nagy Logika)
|
|
@app.post("/api/register")
|
|
async def register_vehicle(data: VehicleRegister):
|
|
async with AsyncSessionLocal() as session:
|
|
async with session.begin(): # Tranzakció indítása
|
|
try:
|
|
# 1. Megnézzük, létezik-e már ez a VAS (Alvázszám alapján)?
|
|
check_query = text("SELECT id FROM data.vehicles WHERE vin = :vin")
|
|
result = await session.execute(check_query, {"vin": data.vin})
|
|
existing_car_id = result.scalar()
|
|
|
|
vehicle_id = existing_car_id
|
|
|
|
# 2. HA NEM LÉTEZIK -> Létrehozzuk a VASAT
|
|
if not vehicle_id:
|
|
print(f"🆕 Új autó az adatbázisban: {data.vin}")
|
|
insert_car = text("""
|
|
INSERT INTO data.vehicles (model_id, vin, current_plate)
|
|
VALUES (:mid, :vin, :plt)
|
|
RETURNING id
|
|
""")
|
|
res = await session.execute(insert_car, {
|
|
"mid": data.model_id, "vin": data.vin, "plt": data.plate
|
|
})
|
|
vehicle_id = res.scalar()
|
|
else:
|
|
print(f"♻️ Létező autó átvétele: {data.vin} (ID: {vehicle_id})")
|
|
# Itt később lezárhatjuk az előző tulajdonos history-ját!
|
|
|
|
# 3. BEJEGYZÉS A TÖRTÉNELEMBE (HISTORY)
|
|
# Ez köti össze a User-t (most fixen ID=1) az Autóval
|
|
insert_history = text("""
|
|
INSERT INTO data.vehicle_history
|
|
(vehicle_id, user_id, role, start_date, start_mileage)
|
|
VALUES (:vid, 1, :role, :s_date, :s_mil)
|
|
""")
|
|
|
|
await session.execute(insert_history, {
|
|
"vid": vehicle_id,
|
|
"role": data.role,
|
|
"s_date": data.purchase_date,
|
|
"s_mil": data.mileage
|
|
})
|
|
|
|
return {"status": "success", "message": "Jármű sikeresen rögzítve a flottában!"}
|
|
|
|
except Exception as e:
|
|
print(f"❌ Hiba: {e}")
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
# Frontend kiszolgálása
|
|
@app.get("/")
|
|
async def serve_frontend():
|
|
if os.path.exists("/app/frontend/index.html"):
|
|
return FileResponse("/app/frontend/index.html")
|
|
return {"error": "Frontend hiányzik"}
|
|
|