def test_get_kids(self): """ Test retrieval of comment usernames from a story """ con = ApiConnector() item = con.get_item(8863) user_dict = con.get_kids(item) self.assertEqual(user_dict['noisemaker'], 'noisemaker') self.assertEqual(user_dict['jganetsk'], 'jganetsk') self.assertEqual(user_dict['vlad'], 'vlad')
def test_get_kids(self): """ Test retrieval of comment usernames from a story """ con = ApiConnector() item = con.get_item(8863) user_dict = con.get_kids(item) self.assertEqual(user_dict['noisemaker'], 'noisemaker') self.assertEqual(user_dict['jganetsk'], 'jganetsk') self.assertEqual(user_dict['vlad'], 'vlad')
def main(logger, known_users): conn = ApiConnector() csvio = CsvIo() article_list = conn.get_top() stories = [] for i in article_list: try: story = conn.get_item(i) if (not conn.is_valid_item(story)) or ( not conn.is_story_item(story)): continue logger.debug(csvio.story_to_csv(story)) stories.append(story) except NetworkError as e: logger.exception(e) csvio.write_stories_csv(stories) for story in stories: try: conn.get_kids(story) except NetworkError as e: logger.exception(e) users = [] for u in sorted(conn.user_dict.keys()): if known_users.get(u) == None: logger.debug("Skipping get_user call for %s" % u) continue try: userjson = conn.get_user(u) users.append(userjson) except NetworkError as e: logger.exception(e) except RuntimeError as e: logger.exception(e) csvio.write_users_csv(users)
def main(logger, known_users): conn = ApiConnector() csvio = CsvIo() article_list = conn.get_top() stories = [] for i in article_list: try: story = conn.get_item(i) if (not conn.is_valid_item(story)) or (not conn.is_story_item(story)): continue logger.debug(csvio.story_to_csv(story)) stories.append(story) except NetworkError as e: logger.exception(e) csvio.write_stories_csv(stories) for story in stories: try: conn.get_kids(story) except NetworkError as e: logger.exception(e) users = [] for u in sorted(conn.user_dict.keys()): if known_users.get(u) == None: logger.debug("Skipping get_user call for %s" % u) continue try: userjson = conn.get_user(u) users.append(userjson) except NetworkError as e: logger.exception(e) except RuntimeError as e: logger.exception(e) csvio.write_users_csv(users)