service-finder/backend/app/models/asset.py

import uuid
from sqlalchemy import Column, Integer, String, Boolean, DateTime, ForeignKey, JSON, Numeric, text, Text
from sqlalchemy.orm import relationship
from sqlalchemy.dialects.postgresql import UUID as PG_UUID
from sqlalchemy.sql import func
from app.db.base_class import Base

class AssetCatalog(Base):
    __tablename__ = "vehicle_catalog"
    __table_args__ = {"schema": "data"}
    id = Column(Integer, primary_key=True, index=True)
    make = Column(String, index=True, nullable=False)
    model = Column(String, index=True, nullable=False)
    generation = Column(String)
    year_from = Column(Integer)
    year_to = Column(Integer)
    vehicle_class = Column(String) 
    fuel_type = Column(String)
    engine_code = Column(String)
    factory_data = Column(JSON, server_default=text("'{}'::jsonb"))
    assets = relationship("Asset", back_populates="catalog")

class Asset(Base):
    __tablename__ = "assets"
    __table_args__ = {"schema": "data"}
    id = Column(PG_UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    vin = Column(String(17), unique=True, index=True, nullable=False)
    license_plate = Column(String(20), index=True)
    name = Column(String) 
    year_of_manufacture = Column(Integer)
    catalog_id = Column(Integer, ForeignKey("data.vehicle_catalog.id"))
    is_verified = Column(Boolean, default=False)
    status = Column(String(20), default="active")
    created_at = Column(DateTime(timezone=True), server_default=func.now())
    updated_at = Column(DateTime(timezone=True), onupdate=func.now())

    catalog = relationship("AssetCatalog", back_populates="assets")
    financials = relationship("AssetFinancials", back_populates="asset", uselist=False)
    telemetry = relationship("AssetTelemetry", back_populates="asset", uselist=False)
    assignments = relationship("AssetAssignment", back_populates="asset")
    events = relationship("AssetEvent", back_populates="asset")
    costs = relationship("AssetCost", back_populates="asset")
    reviews = relationship("AssetReview", back_populates="asset")
    ownership_history = relationship("VehicleOwnership", back_populates="vehicle")

class AssetFinancials(Base):
    __tablename__ = "asset_financials"
    __table_args__ = {"schema": "data"}
    id = Column(Integer, primary_key=True)
    asset_id = Column(PG_UUID(as_uuid=True), ForeignKey("data.assets.id"), unique=True)
    acquisition_price = Column(Numeric(18, 2))
    acquisition_date = Column(DateTime)
    financing_type = Column(String) 
    residual_value_estimate = Column(Numeric(18, 2))
    asset = relationship("Asset", back_populates="financials")

class AssetTelemetry(Base):
    __tablename__ = "asset_telemetry"
    __table_args__ = {"schema": "data"}
    id = Column(Integer, primary_key=True)
    asset_id = Column(PG_UUID(as_uuid=True), ForeignKey("data.assets.id"), unique=True)
    current_mileage = Column(Integer, default=0)
    mileage_unit = Column(String(10), default="km")
    vqi_score = Column(Numeric(5, 2), default=100.00)
    dbs_score = Column(Numeric(5, 2), default=100.00)
    asset = relationship("Asset", back_populates="telemetry")

class AssetReview(Base):
    __tablename__ = "asset_reviews"
    __table_args__ = {"schema": "data"}
    id = Column(Integer, primary_key=True)
    asset_id = Column(PG_UUID(as_uuid=True), ForeignKey("data.assets.id"), nullable=False)
    user_id = Column(Integer, ForeignKey("data.users.id"), nullable=False)
    overall_rating = Column(Integer)
    criteria_scores = Column(JSON, server_default=text("'{}'::jsonb")) 
    comment = Column(Text)
    created_at = Column(DateTime(timezone=True), server_default=func.now())
    
    asset = relationship("Asset", back_populates="reviews")
    user = relationship("User") # <--- JAVÍTÁS: Hozzáadva

class AssetAssignment(Base):
    __tablename__ = "asset_assignments"
    __table_args__ = {"schema": "data"}
    id = Column(PG_UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    asset_id = Column(PG_UUID(as_uuid=True), ForeignKey("data.assets.id"), nullable=False)
    organization_id = Column(Integer, ForeignKey("data.organizations.id"), nullable=False)
    assigned_at = Column(DateTime(timezone=True), server_default=func.now())
    released_at = Column(DateTime(timezone=True), nullable=True)
    status = Column(String(30), default="active")
    
    asset = relationship("Asset", back_populates="assignments")
    organization = relationship("Organization") # <--- KRITIKUS JAVÍTÁS: Ez okozta a login hibát

class AssetEvent(Base):
    __tablename__ = "asset_events"
    __table_args__ = {"schema": "data"}
    id = Column(PG_UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    asset_id = Column(PG_UUID(as_uuid=True), ForeignKey("data.assets.id"), nullable=False)
    event_type = Column(String(50), nullable=False)
    recorded_mileage = Column(Integer)
    data = Column(JSON, server_default=text("'{}'::jsonb"))
    asset = relationship("Asset", back_populates="events")

class AssetCost(Base):
    __tablename__ = "asset_costs"
    __table_args__ = {"schema": "data"}
    id = Column(PG_UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    asset_id = Column(PG_UUID(as_uuid=True), ForeignKey("data.assets.id"), nullable=False)
    organization_id = Column(Integer, ForeignKey("data.organizations.id"), nullable=False)
    driver_id = Column(Integer, ForeignKey("data.users.id"), nullable=True)
    cost_type = Column(String(50), nullable=False)
    amount_local = Column(Numeric(18, 2), nullable=False)
    currency_local = Column(String(3), nullable=False)
    amount_eur = Column(Numeric(18, 2), nullable=True)
    net_amount_local = Column(Numeric(18, 2))
    vat_rate = Column(Numeric(5, 2))
    exchange_rate_used = Column(Numeric(18, 6))
    date = Column(DateTime(timezone=True), server_default=func.now())
    mileage_at_cost = Column(Integer)
    data = Column(JSON, server_default=text("'{}'::jsonb"))
    
    asset = relationship("Asset", back_populates="costs")
    organization = relationship("Organization") # <--- JAVÍTÁS: Hozzáadva
    driver = relationship("User") # <--- JAVÍTÁS: Hozzáadva

class ExchangeRate(Base):
    __tablename__ = "exchange_rates"
    __table_args__ = {"schema": "data"}
    id = Column(Integer, primary_key=True)
    base_currency = Column(String(3), default="EUR")
    target_currency = Column(String(3), unique=True)
    rate = Column(Numeric(18, 6), nullable=False)
    rate_date = Column(DateTime, server_default=func.now())
    updated_at = Column(DateTime(timezone=True), onupdate=func.now())