def add_comment_count(self, comments_url, count):
stories = StoryModel.query(comments_url)
for dummy in stories: # only want the first element...
story = dummy
story.comments = count
story.save()
statsd.increment('nb.comment_counts_added')
def unstar(self, nb_hash):
stories = StoryModel.scan(StoryModel.nb_hash == nb_hash)
story = next(stories) # only expect one result
story.starred = False
story.update(
actions=[
StoryModel.starred.set(False)
]
)
statsd.increment('nb.stars_removed')
def add_story(self, nb_hash, added, comments_url, story_url):
story = StoryModel(comments_url, nb_hash=nb_hash, added=added, url=story_url)
try:
story.save()
except Exception as err:
logger.error("Caught exception while saving Story model, wait 2 sec and retry")
statsd.event('Failed to save story', err.message, alert_type='error')
time.sleep(2)
story.save()
statsd.increment('nb.stories_added')
def list_urls(self):
stories = StoryModel.scan()
return list([{'nb_hash': s.nb_hash, 'url': s.url} for s in stories])
def list_stories_without_comment_count(self):
stories = StoryModel.scan(StoryModel.comments == -1)
return stories
def list_stories_with_comments_fewer_than(self, threshold):
stories = StoryModel.scan((StoryModel.comments < threshold) & (StoryModel.comments >= 0))
return stories
def ensure_stories_table_exists(self):
StoryModel.create_table()
ErrorModel.create_table()