def test_get_id_from_code(self):
code = Media.get_code_from_id('1270593720437182847')
self.assertEqual('BGiDkHAgBF_', code)
code = Media.get_code_from_id('1270593720437182847_3')
self.assertEqual('BGiDkHAgBF_', code)
code = Media.get_code_from_id(1270593720437182847)
self.assertEqual('BGiDkHAgBF_', code)
def _init_media(self, array):
self.media_count = array['count'] if 'count' in array.keys() else 0
try:
nodes = array['edges']
except:
return
if not self.media_count or isinstance(nodes, list):
return
for media_array in nodes:
media = Media(media_array['node'])
if isinstance(media, Media):
self.add_media(media)
def parse_media(media_in_json):
"""Parse comments from media_json page.
Args:
media_in_json (json): json object representing the media content
Returns: (Media)
"""
media = Media(media_in_json)
# add comment info
comments = parse_comments(media_in_json)
media.comments_count = len(comments)
media.comments = '_[COMMENT_SEP]_'.join(comments) # into a string
# add carousel info
carousel = parse_carousel(media_in_json)
media.carousel_ids = carousel['media_ids']
media.carousel_types = carousel['media_types']
media.carousel_thumbnail_urls = carousel['thumbnails']
media.carousel_image_highres_urls = carousel['image_highres_urls']
return media
def test_get_code_from_id(self):
id = Media.get_id_from_code('BGiDkHAgBF_')
self.assertEqual(1270593720437182847, id)