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

add comments, add section index to Quote, making printing optional on Section.clear(), remove scrapeURL

Browse Source
This commit is contained in:
Xevion
2020-01-20 17:12:01 -06:00
parent 81365da041
commit c12eb65eb2
1 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
app/models.py
View File

@@ -91,6 +91,7 @@ class Episode(db.Model):
db.session.commit() db.session.commit()
def clear(self): def clear(self):
"""delete all sections relevant to this episode in order to reprocess"""
sections = Section.query.filter_by(episode_id=self.id).all() sections = Section.query.filter_by(episode_id=self.id).all()
print(f'Clearing {len(sections)} Sections of Ep {self.number} Season {self.season_id}') print(f'Clearing {len(sections)} Sections of Ep {self.number} Season {self.season_id}')
for section in sections: for section in sections:
@@ -98,10 +99,6 @@ class Episode(db.Model):
db.session.delete(section) db.session.delete(section)
db.session.commit() db.session.commit()
@property
def scrapeURL(self):
return f'http://officequotes.net/no{self.season_id}-{str(self.number).zfill(2)}.php'
@staticmethod @staticmethod
def clear_all(): def clear_all():
"""runs clear() on every episode in the database""" """runs clear() on every episode in the database"""
@@ -121,20 +118,21 @@ class Section(db.Model):
def build(self, quotes, commit=False, reset=False): def build(self, quotes, commit=False, reset=False):
"""given an List of unformatted script quotes, automatically creates Quotes assigned to this Section""" """given an List of unformatted script quotes, automatically creates Quotes assigned to this Section"""
for quote in quotes: for i, quote in enumerate(quotes):
if quote.lower().startswith('deleted scene'): if quote.lower().startswith('deleted scene'):
raise Exception(f'Deleted Scene Quote passed to Section Builder: "{quote}"') raise Exception(f'Deleted Scene Quote passed to Section Builder: "{quote}"')
# match = re.match(quotePattern, quote) # match = re.match(quotePattern, quote)
# assert match != None, f"Quote '{quote}' could not be processed." # assert match != None, f"Quote '{quote}' could not be processed."
# q = Quote(section=self, speaker=match[1].strip(), text=match[2].strip()) # q = Quote(section=self, speaker=match[1].strip(), text=match[2].strip())
mark = quote.find(':') mark = quote.find(':')
q = Quote(section=self, speaker=quote[:mark], text=quote[mark + 1:]) q = Quote(section=self, speaker=quote[:mark], text=quote[mark + 1:], section_index=i)
db.session.add(q) db.session.add(q)
if commit: db.session.commit() if commit: db.session.commit()
def clear(self, commit=True): def clear(self, doprint=True, commit=True):
"""delete all quotes relevant to this section"""
quotes = Quote.query.filter_by(section_id=self.id).all() quotes = Quote.query.filter_by(section_id=self.id).all()
print(f'Clearing {len(quotes)} quotes from Section ID {self.id}') if doprint: print(f'Clearing {len(quotes)} quotes from Section ID {self.id}')
for quote in quotes: for quote in quotes:
db.session.delete(quote) db.session.delete(quote)
if commit: db.session.commit() if commit: db.session.commit()