xevion/the-office
1
0
Fork 0
mirror of https://github.com/Xevion/the-office.git synced 2025-12-10 06:08:51 -06:00
Code Issues Packages Projects Releases Wiki Activity

fix up episode not having a number, add some more comments, fix episode count meta, fix Season having "sections" not "episodes" relationship, add build() to Season

Browse Source
This commit is contained in:
Xevion
2020-01-19 18:53:28 -06:00
parent ed7e9052d9
commit 3e73d9f1f8
1 changed files with 23 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
app/models.py
View File

@@ -1,20 +1,35 @@
from app import db, login from app import db, login
episodes = [0, 6, 22, 23, 14, 26, 24, 24, 24, 23] episodes = [5, 6, 22, 23, 14, 26, 24, 24, 24, 23]
class Season(db.Model): class Season(db.Model):
id = db.Column(db.Integer, primary_key=True) id = db.Column(db.Integer, primary_key=True)
episode = db.Column(db.Integer) episodes = db.relationship('Episode', backref='season', lazy='dynamic')
sections = db.relationship('Episode', backref='season', lazy='dynamic')
def __init__(self, **kwargs): def __init__(self, **kwargs):
assert 0 >= kwargs.get('id') <= 9, "Season ID must be 0-9 inclusive" assert 0 >= kwargs.get('id') <= 9, "Season ID must be 0-9 inclusive"
super(Season, self).__init__(**kwargs) super(Season, self).__init__(**kwargs)
def build(self):
"""runs build operations on every Episode under this season"""
for episode in range(1, episodes[self.id - 1] + 1):
ep = Episode.query.filter_by(season=self, number=episode).first()
if ep is None:
# Add the episode, then build
print(f'Creating new Episode, Season {self.id}, Episode {episode}')
ep = Episode(season=self, number=episode)
db.session.add(ep)
# I'm commiting early, which is a bit taboo, but I'm more worried about what the Episode object will need while building.
db.session.commit()
else:
# Regardless of whether it existended before hand, the episode will be built.
pass
ep.build()
@property @property
def episodes(self): def episodes(self):
"""returns a List of Episodes under this Season""" """returns a List of Episodes under this Season"""
pass return Episode.query.filter_by(season=self).all()
@property @property
def characters(self, sort): def characters(self, sort):
@@ -23,9 +38,10 @@ class Season(db.Model):
class Episode(db.Model): class Episode(db.Model):
"""represents a Episode with underlying Sections (representing a specific cutscene or area""" """represents a Episode with underlying Sections (representing a specific cutscene or area"""
id = db.Column(db.Integer, primary_key=True) id = db.Column(db.Integer, primary_key=True) # arbitrary ID, should NOT be relied on to determine episode number or correlating season
season_id = db.Column(db.Integer, db.ForeignKey('season.id')) number = db.Column(db.Integer) # episode number
sections = db.relationship('Section', backref='episode', lazy='dynamic') season_id = db.Column(db.Integer, db.ForeignKey('season.id')) # correlating season number
sections = db.relationship('Section', backref='episode', lazy='dynamic') # sections of quotes under this episode
class Section(db.Model): class Section(db.Model):
"""represents a Section of Quotes, a specific scene with relevant dialog""" """represents a Section of Quotes, a specific scene with relevant dialog"""