fix ambiguous foreign key relationships, Period.state default, fixing back populates

This commit is contained in:
Xevion
2021-02-13 08:31:09 -06:00
parent 7fb9f961b5
commit 00f5b64f3f

View File

@@ -36,24 +36,41 @@ class Guild(Base):
prefix = Column(Text, default='$') # The command prefix used by this particular guild.
submission_channel = Column(Integer, nullable=True) # The channel being scanned for messages by this particular guild.
current_period = Column(Integer, ForeignKey('period.id'), nullable=True) # The period currently active for this guild.
periods = relationship("Period", back_populates="guild") # All periods ever started inside this guild
current_period_id = Column(Integer, ForeignKey('period.id'), nullable=True) # The period currently active for this guild.
current_period = relationship("Period", foreign_keys=current_period_id)
periods = relationship("Period", back_populates="guild", foreign_keys="Period.guild_id") # All periods ever started inside this guild
active = Column(Boolean, default=True) # Whether or not the bot is active inside the given Guild. Used for better querying.
joined = Column(DateTime, default=datetime.datetime.utcnow) # The initial join time for this bot to a particular Discord.
last_joined = Column(DateTime, nullable=True) # The last time the bot joined this server.
def check_not_finished(func):
"""
Throws `FinishedPeriod` if the period has already completed, is inactive, or is in it's Finished State.
"""
@functools.wraps(func)
def wrapper(self, *args, **kwargs):
if self.state is PeriodStates.FINISHED: raise FinishedPeriod(f"Period is in it's Finished state.")
elif not self.active: raise FinishedPeriod("Period is no longer active.")
elif self.completed: raise FinishedPeriod("Period is already completed.")
func(self, *args, **kwargs)
return wrapper
class Submission(Base):
"""Represents a Message the bot has seen and remembered as a valid active submission."""
__tablename__ = 'submission'
id = Column(Integer, primary_key=True) # Doubles as the ID this Guild has in Discord
user = Column(Integer) # The ID of the user who submitted it.
period_id = Column(Integer, ForeignKey("period.id")) # The id of the period this Submission relates to.
period = relationship("Period", back_populates="period") # The period this submission was made in.
timestamp = Column(DateTime) # When the Submission was posted
period_id = Column(Integer, ForeignKey("period.id")) # The id of the period this Submission relates to.
period = relationship("Period", back_populates="submissions") # The period this submission was made in.
class Period(Base):
"""Represents a particular period of submissions and voting for a given"""
@@ -61,11 +78,12 @@ class Period(Base):
id = Column(Integer, primary_key=True)
guild_id = Column(Integer, ForeignKey("guild.id")) # The guild this period was created in.
guild = relationship("Guild", back_populates="guild")
guild = relationship("Guild", back_populates="periods", foreign_keys=guild_id)
state = Column(Enum(PeriodStates)) # The current state of the period.
state = Column(Enum(PeriodStates), default=PeriodStates.READY) # The current state of the period.
active = Column(Boolean, default=True) # Whether this Period is currently running. State will not necessarily be FINISHED.
completed = Column(Boolean, default=False) # Whether this Period was completed to the end, properly.
submissions = relationship("Submission", back_populates="period") # All the submissions submitted during this Period's active state.
start_time = Column(DateTime, default=datetime.datetime.utcnow()) # When this period was created/started (Ready state).
@@ -74,19 +92,6 @@ class Period(Base):
voting_time = Column(DateTime, nullable=True) # When this period switched to the Voting state.
finished_time = Column(DateTime, nullable=True) # When this period switched to the Finished state.
def check_not_finished(self, func):
"""
Throws `FinishedPeriod` if the period has already completed, is inactive, or is in it's Finished State.
"""
@functools.wraps(func)
def wrapper(*args, **kwargs):
if self.state is PeriodStates.FINISHED: raise FinishedPeriod(f"Period is in it's Finished state.")
elif not self.active: raise FinishedPeriod("Period is no longer active.")
elif self.completed: raise FinishedPeriod("Period is already completed.")
func(*args, **kwargs)
return wrapper
@check_not_finished
def advance_state(self) -> PeriodStates:
"""