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

newpeat quote check

Browse Source
This commit is contained in:
Xevion
2020-01-20 19:28:05 -06:00
parent a15a2b3344
commit 8a1b8060c0
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
app/models.py
View File

@@ -78,14 +78,18 @@ class Episode(db.Model):
for section in sections: for section in sections:
try: try:
quotes = [quote.string + quote.next_sibling.string for quote in section.find_all('b')] isNewpeat = False
except BaseException as e: quotes = []
print(section) for quote in section.find_all('b'):
raise e print(quote.string.lower())
if 'newpeat' in quote.string.lower():
isNewPeat = True
else:
quotes.append(quote.string + quote.next_sibling.string)
isDeletedScene = quotes[0].lower().startswith('deleted scene') isDeletedScene = quotes[0].lower().startswith('deleted scene')
if isDeletedScene: if isDeletedScene:
deleted += 1 deleted += 1
s = Section(episode_id=self.id, deleted=deleted if isDeletedScene else -1) s = Section(episode_id=self.id, deleted=deleted if isDeletedScene else -1, newpeat=isNewpeat)
s.build(quotes[1:] if isDeletedScene else quotes) s.build(quotes[1:] if isDeletedScene else quotes)
db.session.add(s) db.session.add(s)
db.session.commit() db.session.commit()
@@ -114,6 +118,7 @@ class Section(db.Model):
id = db.Column(db.Integer, primary_key=True) id = db.Column(db.Integer, primary_key=True)
episode_id = db.Column(db.Integer, db.ForeignKey('episode.id')) episode_id = db.Column(db.Integer, db.ForeignKey('episode.id'))
deleted = db.Column(db.Integer, default=-1) deleted = db.Column(db.Integer, default=-1)
newpeat = db.Column(db.Boolean, default=False)
quotes = db.relationship('Quote', backref='section', lazy='dynamic') quotes = db.relationship('Quote', backref='section', lazy='dynamic')
def build(self, quotes, commit=False, reset=False): def build(self, quotes, commit=False, reset=False):