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