def open_cassandra_session(): setup_env() connection.setup([CASSANDRA_CONFIG['ip']], CASSANDRA_CONFIG['keyspace'], protocol_version=3) SourceUserEntity.sync_table() RawRecentMediaEntity.sync_table() RawUserEntity.sync_table() RawUserRecentMediaEntity.sync_table()
def handle_user_info(parsed_media, user_recent_media_added, users_added, users_updated): # Find the user info user = RawUserEntity.parse(api.user(parsed_media.user_id)) user.save() # Find and parse the users recent media user_recent_media = [] recents = api.user_recent_media(user_id=parsed_media.user_id, count=64) for recent in recents[0]: parsed_user_recent_media = RawUserRecentMediaEntity.parse(recent) parsed_user_recent_media.save() user_recent_media.append(parsed_user_recent_media) user_recent_media_added += 1 source_user_model_obj = SourceUserEntity.objects( SourceUserEntity.user_id == user.user_id) if source_user_model_obj.first(): save_source_user(True, source_user_model_obj.first(), user_recent_media) users_updated += 1 else: save_source_user(False, user, user_recent_media) users_added += 1 return user_recent_media_added, users_added, users_updated
def handle_user_info(parsed_media, user_recent_media_added, users_added, users_updated): # Find the user info user = RawUserEntity.parse(api.user(parsed_media.user_id)) user.save() # Find and parse the users recent media user_recent_media = [] recents = api.user_recent_media(user_id=parsed_media.user_id, count=64) for recent in recents[0]: parsed_user_recent_media = RawUserRecentMediaEntity.parse(recent) parsed_user_recent_media.save() user_recent_media.append(parsed_user_recent_media) user_recent_media_added += 1 source_user_model_obj = SourceUserEntity.objects(SourceUserEntity.user_id == user.user_id) if source_user_model_obj.first(): save_source_user(True, source_user_model_obj.first(), user_recent_media) users_updated += 1 else: save_source_user(False, user, user_recent_media) users_added += 1 return user_recent_media_added, users_added, users_updated