91 lines
3.5 KiB
Python
91 lines
3.5 KiB
Python
import enum
|
|
from sqlalchemy import (Column, Integer, String, Date, Enum, Boolean,
|
|
DECIMAL, Text, TIMESTAMP, ForeignKey)
|
|
from sqlalchemy.sql import func
|
|
from sqlalchemy.orm import relationship # Aggiungi questo import
|
|
|
|
from .database import Base
|
|
|
|
# Definiamo gli Enum Python corrispondenti ai tipi custom di PostgreSQL
|
|
class GenderType(str, enum.Enum):
|
|
male = "male"
|
|
female = "female"
|
|
other = "other"
|
|
|
|
class ShoeSystemType(str, enum.Enum):
|
|
EU = "EU"
|
|
US = "US"
|
|
UK = "UK"
|
|
|
|
class BraSystemType(str, enum.Enum):
|
|
US = "US"
|
|
UK = "UK"
|
|
EU = "EU"
|
|
FR = "FR"
|
|
AU = "AU"
|
|
IT = "IT"
|
|
JP = "JP"
|
|
|
|
class Image(Base):
|
|
__tablename__ = "images"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
celebrity_id = Column(Integer, ForeignKey("celebrities.id"), nullable=False)
|
|
file_path = Column(String, nullable=False)
|
|
caption = Column(Text, nullable=True)
|
|
uploaded_at = Column(TIMESTAMP(timezone=True), server_default=func.now())
|
|
|
|
# Relazione inversa: "questa immagine appartiene a UNA celebrità".
|
|
# Specifichiamo che deve usare la colonna `celebrity_id` di QUESTA tabella.
|
|
celebrity = relationship("Celebrity", back_populates="images", foreign_keys=[celebrity_id])
|
|
|
|
|
|
class Celebrity(Base):
|
|
__tablename__ = "celebrities"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
name = Column(String, unique=True, nullable=False, index=True)
|
|
gender = Column(Enum(GenderType), nullable=False)
|
|
birth_date = Column(Date, nullable=True)
|
|
birth_place = Column(String, nullable=True)
|
|
nationality = Column(String, nullable=True)
|
|
ethnicity = Column(String, nullable=True)
|
|
sexuality = Column(String, nullable=True)
|
|
|
|
hair_color = Column(String, nullable=True)
|
|
eye_color = Column(String, nullable=True)
|
|
height_cm = Column(Integer, nullable=True)
|
|
weight_kg = Column(Integer, nullable=True)
|
|
body_type = Column(String, nullable=True)
|
|
|
|
bust_cm = Column(Integer, nullable=True)
|
|
waist_cm = Column(Integer, nullable=True)
|
|
hips_cm = Column(Integer, nullable=True)
|
|
|
|
chest_circumference_cm = Column(Integer, nullable=True)
|
|
|
|
bra_band_size = Column(Integer, nullable=True)
|
|
bra_cup_size = Column(String, nullable=True)
|
|
bra_size_system = Column(Enum(BraSystemType), nullable=True)
|
|
|
|
boobs_are_natural = Column(Boolean, nullable=True)
|
|
shoe_size = Column(DECIMAL(4, 1), nullable=True)
|
|
shoe_size_system = Column(Enum(ShoeSystemType), nullable=True)
|
|
|
|
biography = Column(Text, nullable=True)
|
|
official_website = Column(String, nullable=True)
|
|
# 1. Relazione "una celebrità ha MOLTE immagini".
|
|
# SQLAlchemy deve usare la FK che si trova nella tabella Image.
|
|
images = relationship("Image", back_populates="celebrity", foreign_keys="[Image.celebrity_id]")
|
|
# Chiave esterna per l'immagine del profilo
|
|
profile_image_id = Column(Integer, ForeignKey("images.id", ondelete="SET NULL"), nullable=True)
|
|
# 2. Relazione "una celebrità ha UNA immagine del profilo".
|
|
# SQLAlchemy deve usare la FK che si trova in QUESTA tabella (celebrities).
|
|
profile_image = relationship("Image", foreign_keys=[profile_image_id])
|
|
|
|
created_at = Column(TIMESTAMP(timezone=True), server_default=func.now())
|
|
updated_at = Column(TIMESTAMP(timezone=True), default=func.now(), onupdate=func.now())
|
|
|
|
# Nota: le altre tabelle (images, tattoos, etc.) andrebbero modellate qui
|
|
# se si volessero gestire le relazioni in modo completo con SQLAlchemy.
|
|
# Per questo esempio, ci concentriamo solo sulla tabella `celebrities`. |