add entities

alembic/versions/fc681e60e977_add_entities.py

@@ -0,0 +1,46 @@
+"""add entities
+
+Revision ID: fc681e60e977
+Revises: 315d7604a82e
+Create Date: 2025-04-17 23:17:58.236176
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'fc681e60e977'
+down_revision: Union[str, None] = '315d7604a82e'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint('caracteristics_fk_player_id_fkey', 'caracteristics', schema='data', type_='foreignkey')
+    op.create_foreign_key('caracteristics_fk_player_id_fkey', 'caracteristics', 'player', ['fk_player_id'], ['id'], source_schema='data', referent_schema='data', ondelete='CASCADE')
+    op.create_unique_constraint('uq_date_fk_winner_id_fk_loser_id', 'match', ['date', 'fk_winner_id', 'fk_loser_id'], schema='data')
+    op.drop_constraint('match_fk_winner_id_fkey', 'match', schema='data', type_='foreignkey')
+    op.drop_constraint('match_fk_loser_id_fkey', 'match', schema='data', type_='foreignkey')
+    op.create_foreign_key('match_fk_winner_id_fkey', 'match', 'player', ['fk_winner_id'], ['id'], source_schema='data', referent_schema='data', ondelete='CASCADE')
+    op.create_foreign_key('match_fk_loser_id_fkey', 'match', 'player', ['fk_loser_id'], ['id'], source_schema='data', referent_schema='data', ondelete='CASCADE')
+    op.create_foreign_key('odds_fk_match_id_fkey', 'odds', 'match', ['fk_match_id'], ['id'], source_schema='data', referent_schema='data', ondelete='CASCADE')
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint('odds_fk_match_id_fkey', 'odds', schema='data', type_='foreignkey')
+    op.drop_constraint('match_fk_loser_id_fkey', 'match', schema='data', type_='foreignkey')
+    op.drop_constraint('match_fk_winner_id_fkey', 'match', schema='data', type_='foreignkey')
+    op.create_foreign_key('match_fk_loser_id_fkey', 'match', 'player', ['fk_loser_id'], ['id'], source_schema='data', referent_schema='data')
+    op.create_foreign_key('match_fk_winner_id_fkey', 'match', 'player', ['fk_winner_id'], ['id'], source_schema='data', referent_schema='data')
+    op.drop_constraint('uq_date_fk_winner_id_fk_loser_id', 'match', schema='data', type_='unique')
+    op.drop_constraint('caracteristics_fk_player_id_fkey', 'caracteristics', schema='data', type_='foreignkey')
+    op.create_foreign_key('caracteristics_fk_player_id_fkey', 'caracteristics', 'player', ['fk_player_id'], ['id'], source_schema='data', referent_schema='data')
+    # ### end Alembic commands ###

src/entity/match.py

@@ -0,0 +1,41 @@
+
+from sqlalchemy import Integer, String, DateTime, ForeignKey, UniqueConstraint
+from sqlalchemy.orm import relationship, mapped_column, Mapped
+from src.entity.odds import Odds
+from src.entity.player import Player
+
+from . import Base
+
+class Match(Base):
+    """
+    Match table
+    """
+    __tablename__ = "match"
+    __table_args__ = (
+        UniqueConstraint('date', 'fk_winner_id', 'fk_loser_id', name='uq_date_fk_winner_id_fk_loser_id'),
+        {'schema': 'data'},
+    )
+
+    # Match table columns
+    id: Mapped[int] = mapped_column(primary_key=True)
+    date: Mapped[DateTime] = mapped_column(DateTime, nullable=True)
+    comment: Mapped[str] = mapped_column(String, nullable=True)
+    winner_rank: Mapped[int] = mapped_column(Integer, nullable=True)
+    winner_points: Mapped[int] = mapped_column(Integer, nullable=True)
+    loser_rank: Mapped[int] = mapped_column(Integer, nullable=True)
+    loser_points: Mapped[int] = mapped_column(Integer, nullable=True)
+    tournament_name: Mapped[str] = mapped_column(String, nullable=True)
+    tournament_series: Mapped[str] = mapped_column(String, nullable=True)
+    tournament_surface: Mapped[str] = mapped_column(String, nullable=True)
+    tournament_court: Mapped[str] = mapped_column(String, nullable=True)
+    tournament_round: Mapped[str] = mapped_column(String, nullable=True)
+    tournament_location: Mapped[str] = mapped_column(String, nullable=True)
+
+    # Dependent table
+    odds: Mapped[list["Odds"]] = relationship("Odds", back_populates="match", cascade="all, delete-orphan", passive_deletes=True)
+
+    fk_winner_id: Mapped[int] = mapped_column(ForeignKey("data.player.id", ondelete="CASCADE", name='match_fk_winner_id_fkey'), nullable=False)
+    fk_loser_id: Mapped[int] = mapped_column(ForeignKey("data.player.id", ondelete="CASCADE", name='match_fk_loser_id_fkey'), nullable=False)
+
+    winner: Mapped["Player"] = relationship("Player", foreign_keys=[fk_winner_id])
+    loser: Mapped["Player"] = relationship("Player", foreign_keys=[fk_loser_id])

src/entity/odds.py

@@ -0,0 +1,24 @@
+
+from sqlalchemy import String, Float, ForeignKey, UniqueConstraint
+from sqlalchemy.orm import relationship, mapped_column, Mapped
+
+from . import Base
+
+class Odds(Base):
+    """
+    Odds table
+    """
+    __tablename__ = "odds"
+    __table_args__ = (
+        UniqueConstraint('fk_match_id', 'bookmaker', name='uq_odds_match_bookmaker'),
+        {'schema': 'data', 'extend_existing': True}
+    )
+
+    # Odds table columns
+    id: Mapped[int] = mapped_column(primary_key=True)
+    bookmaker: Mapped[str] = mapped_column(String, nullable=True)
+    winner: Mapped[float] = mapped_column(Float, nullable=True)
+    loser: Mapped[float] = mapped_column(Float, nullable=True)
+
+    fk_match_id: Mapped[int] = mapped_column(ForeignKey("data.match.id", ondelete="CASCADE", name='odds_fk_match_id_fkey'), nullable=False)
+    match: Mapped["Match"] = relationship("Match", back_populates="odds")  # type: ignore # noqa: F821

src/entity/player.py

@@ -0,0 +1,42 @@
+
+from datetime import date
+from sqlalchemy import String, Float, ForeignKey
+from sqlalchemy.orm import relationship, mapped_column, Mapped
+from typing import List
+
+from . import Base
+
+class Player(Base):
+    """
+    Player table
+    """
+    __tablename__ = "player"
+    __table_args__ = {'schema': 'data'}
+
+    # Player table columns
+    id: Mapped[int] = mapped_column(primary_key=True)
+    name: Mapped[str] = mapped_column(String, nullable=True)
+    tennis_id: Mapped[str] = mapped_column(String, nullable=True)
+
+    caracteristics: Mapped["Caracteristics"] = relationship("Caracteristics", back_populates="player", cascade="all, delete-orphan")
+
+    matches_won: Mapped[List["Match"]] = relationship("Match", back_populates="winner", foreign_keys='Match.fk_winner_id')
+    matches_lost: Mapped[List["Match"]] = relationship("Match", back_populates="loser", foreign_keys='Match.fk_loser_id')
+
+class Caracteristics(Base):
+    """
+    Caracteristics table
+    """
+    __tablename__ = "caracteristics"
+    __table_args__ = {'schema': 'data'}
+
+    # Caracteristics table columns
+    id: Mapped[int] = mapped_column(primary_key=True)
+    height: Mapped[float] = mapped_column(Float, nullable=True)
+    weight: Mapped[float] = mapped_column(Float, nullable=True)
+    handedness: Mapped[str] = mapped_column(String, nullable=True)
+    date_of_birth: Mapped[date] = mapped_column(String, nullable=True)
+    date_turned_pro: Mapped[date] = mapped_column(String, nullable=True)
+
+    fk_player_id: Mapped[int] = mapped_column(ForeignKey("data.player.id", ondelete="CASCADE", name='caracteristics_fk_player_id_fkey'), nullable=False)
+    player: Mapped["Player"] = relationship("Player", back_populates="caracteristics")