Add create episode function.

This commit is contained in:
luxick
2018-03-12 20:09:04 +01:00
parent 7034321889
commit 437f418f49
7 changed files with 1172 additions and 1157 deletions

View File

@@ -18,3 +18,19 @@ class WriteFunctions:
sql.Season.start_date: season.start_date,
sql.Season.end_date: season.end_date})
.execute())
@staticmethod
def update_episode(episode: 'models.Episode', *_):
players = list(sql.Player.select().where(sql.Player.id << [player.id for player in episode.players]))
new_ep_id = (sql.Episode
.insert(id=episode.id, number=episode.number, seq_number=episode.number, name=episode.name,
date=episode.date, season=episode.season)
.on_conflict(update={sql.Episode.name: episode.name,
sql.Episode.seq_number: episode.seq_number,
sql.Episode.number: episode.number,
sql.Episode.date: episode.date,
sql.Episode.season: episode.season})
.execute())
db_episode = sql.Episode.get(sql.Episode.id == new_ep_id)
db_episode.players = players
db_episode.save()