diff --git a/model.py b/model.py new file mode 100644 index 0000000..44cd5ca --- /dev/null +++ b/model.py @@ -0,0 +1,184 @@ +from flask_sqlalchemy import SQLAlchemy +from sqlalchemy import Integer, String, Boolean, DateTime, ForeignKey +from sqlalchemy.orm import Mapped, mapped_column, relationship +from datetime import datetime + +db = SQLAlchemy() + +# Modeling these entities separately should make it easier to add new guess categories, for example team guesses or GP guesses + +# This table contains metainformation about the GPs, not specific to seasons (can be reused). +class GrandPrix(db.Model): + __tablename__ = "grandprix" + + def from_csv(self, row): + self.id = int(row[0]) + self.name = str(row[1]) + self.country_code = str(row[2]) + return self + + id: Mapped[int] = mapped_column(Integer, primary_key=True) + name: Mapped[str] = mapped_column(String(64)) + country_code: Mapped[str] = mapped_column(String(2)) # alpha-2 code + +# This table contains season information +class Season(db.Model): + __tablename__ = "season" + + def from_csv(self, row): + self.id = int(row[0]) + self.active = bool(row[1]) + return self + + id: Mapped[int] = mapped_column(Integer, primary_key=True) + active: Mapped[bool] = mapped_column(Boolean) # Only allow guessing the current season + +# This table contains manifestations of GPs, with dates +class Race(db.Model): + __tablename__ = "race" + + def from_csv(self, row): + self.id = int(row[0]) + self.grandprix_id = int(row[1]) + self.season_id = int(row[2]) + self.date = datetime.strptime(row[3], "%y-%m-%d %H:%M") + return self + + id: Mapped[int] = mapped_column(Integer, primary_key=True) + grandprix_id: Mapped[int] = mapped_column(ForeignKey("grandprix.id")) + season_id: Mapped[int] = mapped_column(ForeignKey("season.id")) + date: Mapped[datetime] = mapped_column(DateTime) + + grandprix: Mapped["GrandPrix"] = relationship("GrandPrix", foreign_keys=[grandprix_id]) + season: Mapped["Season"] = relationship("Season", foreign_keys=[season_id]) + +# This table contains teams, e.g. RedBull +class Team(db.Model): + __tablename__ = "team" + + def from_csv(self, row): + self.id = int(row[0]) + self.name = str(row[1]) + self.country_code = str(row[2]) + return self + + id: Mapped[int] = mapped_column(Integer, primary_key=True) + name: Mapped[str] = mapped_column(String(64)) + country_code: Mapped[str] = mapped_column(String(2)) # alpha-2 code + +# This table contains drivers and their team associations, e.g. Max Verschtappen +class Driver(db.Model): + __tablename__ = "driver" + + def from_csv(self, row): + self.id = int(row[0]) + self.name = str(row[1]) + self.team_id = str(row[2]) + self.country_code = str(row[3]) + self.active = bool(row[4]) + return self + + id: Mapped[int] = mapped_column(Integer, primary_key=True) + name: Mapped[str] = mapped_column(String(64)) + team_id: Mapped[int] = mapped_column(ForeignKey("team.id")) + country_code: Mapped[str] = mapped_column(String(2)) # alpha-2 code + active: Mapped[bool] = mapped_column(Boolean) # Only allow guessing active drivers + + team: Mapped["Team"] = relationship("Team", foreign_keys=[team_id]) + +class RaceResult(db.Model): + __tablename__ = "raceresult" + + def from_csv(self, row): + self.id = int(row[0]) + self.race_id = int(row[1]) + self.season_id = int(row[2]) + self.p10_id = int(row[3]) + self.dnf_id = int(row[4]) + return self + + id: Mapped[int] = mapped_column(Integer, primary_key=True) + race_id: Mapped[int] = mapped_column(ForeignKey("race.id")) + season_id: Mapped[int] = mapped_column(ForeignKey("season.id")) + + # These map to drivers + # p01_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + # p02_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + # p03_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + # p04_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + # p05_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + # p06_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + # p07_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + # p08_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + # p09_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + p10_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + # p11_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + # p12_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + # p13_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + # p14_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + # p15_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + # p16_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + # p17_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + # p18_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + # p19_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + # p20_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + dnf_id: Mapped[int] = mapped_column(ForeignKey("driver.id")) + + race: Mapped["Race"] = relationship("Race", foreign_keys=[race_id]) + season: Mapped["Season"] = relationship("Season", foreign_keys=[season_id]) # Redundant but should make things easier + + # p01: Mapped["Driver"] = relationship("Driver", foreign_keys=[p01_id]) # Store all places to allow adding guesses + # p02: Mapped["Driver"] = relationship("Driver", foreign_keys=[p02_id]) + # p03: Mapped["Driver"] = relationship("Driver", foreign_keys=[p03_id]) + # p04: Mapped["Driver"] = relationship("Driver", foreign_keys=[p04_id]) + # p05: Mapped["Driver"] = relationship("Driver", foreign_keys=[p05_id]) + # p06: Mapped["Driver"] = relationship("Driver", foreign_keys=[p06_id]) + # p07: Mapped["Driver"] = relationship("Driver", foreign_keys=[p07_id]) + # p08: Mapped["Driver"] = relationship("Driver", foreign_keys=[p08_id]) + # p09: Mapped["Driver"] = relationship("Driver", foreign_keys=[p09_id]) + p10: Mapped["Driver"] = relationship("Driver", foreign_keys=[p10_id]) + # p11: Mapped["Driver"] = relationship("Driver", foreign_keys=[p11_id]) + # p12: Mapped["Driver"] = relationship("Driver", foreign_keys=[p12_id]) + # p13: Mapped["Driver"] = relationship("Driver", foreign_keys=[p13_id]) + # p14: Mapped["Driver"] = relationship("Driver", foreign_keys=[p14_id]) + # p15: Mapped["Driver"] = relationship("Driver", foreign_keys=[p15_id]) + # p16: Mapped["Driver"] = relationship("Driver", foreign_keys=[p16_id]) + # p17: Mapped["Driver"] = relationship("Driver", foreign_keys=[p17_id]) + # p18: Mapped["Driver"] = relationship("Driver", foreign_keys=[p18_id]) + # p19: Mapped["Driver"] = relationship("Driver", foreign_keys=[p19_id]) + # p20: Mapped["Driver"] = relationship("Driver", foreign_keys=[p20_id]) + dnf: Mapped["Driver"] = relationship("Driver", foreign_keys=[dnf_id]) # Only store first DNF + +# This table contains users that can guess +class User(db.Model): + __tablename__ = "user" + + def from_csv(self, row): + self.id = int(row[0]) + self.name = str(row[1]) + self.active = bool(row[2]) + return self + + id: Mapped[int] = mapped_column(Integer, primary_key=True) + name: Mapped[str] = mapped_column(String(64)) + active: Mapped[bool] = mapped_column(Boolean) # Only show active users + +# This table contains guesses made by users +class Guess(db.Model): + __tablename__ = "guess" + + def from_csv(self, row): + self.id = int(row[0]) + self.race_id = int(row[1]) + self.season_id = int(row[2]) + self.raceresult_id = int(row[3]) + return self + + id: Mapped[int] = mapped_column(Integer, primary_key=True) + race_id: Mapped[int] = mapped_column(ForeignKey("race.id")) + season_id: Mapped[int] = mapped_column(ForeignKey("season.id")) + raceresult_id: Mapped[int] = mapped_column(ForeignKey("raceresult.id")) + + race: Mapped["Race"] = relationship("Race", foreign_keys=[race_id]) + season: Mapped["Season"] = relationship("Season", foreign_keys=[season_id]) # Redundant but should make things easier + raceresult: Mapped["RaceResult"] = relationship("RaceResult", foreign_keys=[raceresult_id]) \ No newline at end of file