# /opt/docker/dev/service_finder/backend/app/models/service.py import uuid from datetime import datetime from typing import Any, List, Optional from sqlalchemy import Integer, String, Boolean, DateTime, ForeignKey, text, Text, Float, Index, Numeric from sqlalchemy.orm import Mapped, mapped_column, relationship from sqlalchemy.dialects.postgresql import UUID as PG_UUID, JSONB from geoalchemy2 import Geometry from sqlalchemy.sql import func # MB 2.0: Központi aszinkron adatbázis motorból húzzuk be a Base-t from app.database import Base class ServiceProfile(Base): """ Szerviz szolgáltató adatai (v1.3.1). """ __tablename__ = "service_profiles" __table_args__ = ( Index('idx_service_fingerprint', 'fingerprint', unique=True), {"schema": "data"} ) id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True) organization_id: Mapped[Optional[int]] = mapped_column(Integer, ForeignKey("data.organizations.id"), unique=True) parent_id: Mapped[Optional[int]] = mapped_column(Integer, ForeignKey("data.service_profiles.id")) fingerprint: Mapped[str] = mapped_column(String(255), index=True, nullable=False) location: Mapped[Any] = mapped_column(Geometry(geometry_type='POINT', srid=4326, spatial_index=False), index=True) status: Mapped[str] = mapped_column(String(20), server_default=text("'ghost'"), index=True) last_audit_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now()) google_place_id: Mapped[Optional[str]] = mapped_column(String(100), unique=True) rating: Mapped[Optional[float]] = mapped_column(Float) user_ratings_total: Mapped[Optional[int]] = mapped_column(Integer) vibe_analysis: Mapped[Any] = mapped_column(JSONB, server_default=text("'{}'::jsonb")) social_links: Mapped[Any] = mapped_column(JSONB, server_default=text("'{}'::jsonb")) specialization_tags: Mapped[Any] = mapped_column(JSONB, server_default=text("'{}'::jsonb")) trust_score: Mapped[int] = mapped_column(Integer, default=30) is_verified: Mapped[bool] = mapped_column(Boolean, default=False) verification_log: Mapped[Any] = mapped_column(JSONB, server_default=text("'{}'::jsonb")) opening_hours: Mapped[Any] = mapped_column(JSONB, server_default=text("'{}'::jsonb")) contact_phone: Mapped[Optional[str]] = mapped_column(String) contact_email: Mapped[Optional[str]] = mapped_column(String) website: Mapped[Optional[str]] = mapped_column(String) bio: Mapped[Optional[str]] = mapped_column(Text) # Kapcsolatok organization: Mapped["Organization"] = relationship("Organization", back_populates="service_profile") expertises: Mapped[List["ServiceExpertise"]] = relationship("ServiceExpertise", back_populates="service") created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now()) updated_at: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True), onupdate=func.now()) class ExpertiseTag(Base): __tablename__ = "expertise_tags" __table_args__ = {"schema": "data"} id: Mapped[int] = mapped_column(Integer, primary_key=True) key: Mapped[str] = mapped_column(String(50), unique=True, index=True) name_hu: Mapped[Optional[str]] = mapped_column(String(100)) category: Mapped[Optional[str]] = mapped_column(String(30)) class ServiceExpertise(Base): __tablename__ = "service_expertises" __table_args__ = {"schema": "data"} service_id: Mapped[int] = mapped_column(Integer, ForeignKey("data.service_profiles.id"), primary_key=True) expertise_id: Mapped[int] = mapped_column(Integer, ForeignKey("data.expertise_tags.id"), primary_key=True) validation_level: Mapped[int] = mapped_column(Integer, default=0) service: Mapped["ServiceProfile"] = relationship("ServiceProfile", back_populates="expertises") expertise: Mapped["ExpertiseTag"] = relationship("ExpertiseTag") class ServiceStaging(Base): """ Hunter (robot) adatok tárolója. """ __tablename__ = "service_staging" __table_args__ = ( Index('idx_staging_fingerprint', 'fingerprint', unique=True), {"schema": "data"} ) id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True) name: Mapped[str] = mapped_column(String, index=True, nullable=False) postal_code: Mapped[Optional[str]] = mapped_column(String(10), index=True) city: Mapped[Optional[str]] = mapped_column(String(100), index=True) full_address: Mapped[Optional[str]] = mapped_column(String) fingerprint: Mapped[str] = mapped_column(String(255), nullable=False) raw_data: Mapped[Any] = mapped_column(JSONB, server_default=text("'{}'::jsonb")) status: Mapped[str] = mapped_column(String(20), server_default=text("'pending'"), index=True) created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now()) class DiscoveryParameter(Base): """ Robot vezérlési paraméterek adminból. """ __tablename__ = "discovery_parameters" __table_args__ = {"schema": "data"} id: Mapped[int] = mapped_column(Integer, primary_key=True) city: Mapped[str] = mapped_column(String(100)) keyword: Mapped[str] = mapped_column(String(100)) is_active: Mapped[bool] = mapped_column(Boolean, default=True) last_run_at: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True))