From c157dcba28f14f3849d981c4c8c3a335eb36878f Mon Sep 17 00:00:00 2001 From: Chris Smith Date: Wed, 4 Mar 2026 12:52:19 -0500 Subject: [PATCH] Add Mega Millions table to the database --- data/database.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/data/database.py b/data/database.py index c6c5917..5eec763 100644 --- a/data/database.py +++ b/data/database.py @@ -77,3 +77,30 @@ class PowerballDraw(Base): self.main_ball4 == other.main_ball4 and self.main_ball5 == other.main_ball5 and \ self.powerball == other.powerball and self.power_play == other.power_play + +class MegaMillionsDraw(Base): + __tablename__ = 'mega_millions_draws' + + draw_date: Mapped[datetime.date] = mapped_column(Date, primary_key=True) + main_ball1: Mapped[int] = mapped_column(Integer) + main_ball2: Mapped[int] = mapped_column(Integer) + main_ball3: Mapped[int] = mapped_column(Integer) + main_ball4: Mapped[int] = mapped_column(Integer) + main_ball5: Mapped[int] = mapped_column(Integer) + mega_ball: Mapped[int] = mapped_column(Integer) + megaplier: Mapped[int] = mapped_column(Integer) + + + def __repr__(self): + return f"MegaMillionsDraw({self.draw_date=}, {self.main_ball1=}, {self.main_ball2=}, {self.main_ball3=}, " \ + f"{self.main_ball4=}, {self.main_ball5=}, {self.mega_ball=}, {self.megaplier=})" + + def __str__(self): + return f"{self.draw_date}, [{self.main_ball1}, {self.main_ball2}, {self.main_ball3}, {self.main_ball4}, " \ + f"{self.main_ball5}], {self.mega_ball}, {self.megaplier if self.megaplier else 1}x" + + def __eq__(self, other): + return self.draw_date == other.draw_date and self.main_ball1 == other.main_ball1 and \ + self.main_ball2 == other.main_ball2 and self.main_ball3 == other.main_ball3 and \ + self.main_ball4 == other.main_ball4 and self.main_ball5 == other.main_ball5 and \ + self.mega_ball == other.mega_ball and self.megaplier == other.megaplier