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()