def poll_refresh():
try:
response = requests.get(
f'{app.config["STORIES_ENDPOINT"]}/stories/stats/refresh')
except Timeout:
return
to_update = response.json()
for author in to_update:
auth_stats = Stats.query.get(int(author))
add = auth_stats is None
if auth_stats is None:
auth_stats = Stats()
auth_stats.n_dice = 0
auth_stats.stories_written = 0
auth_stats.likes = 0
auth_stats.dislikes = 0
for story in to_update[author]:
auth_stats.n_dice += story['dice']
auth_stats.stories_written += 1
auth_stats.likes += story['likes']
auth_stats.dislikes += story['dislikes']
if add:
db.session.add(auth_stats)
db.session.commit()
def init_database(database):
example = Stats()
example.author_id = 1
example.likes = 5
example.dislikes = 3
example.stories_written = 15
example.n_dice = 4
database.session.add(example)
database.session.commit()
