def update_band(cls, user_id, follow_count, transaction):
band_data = ArangoCore.get_vertex_in_collection(
ArangoVertex.BAND, user_id)
band_data['follow_count'] = follow_count
ArangoCore.update_vertex_in_collection(ArangoVertex.BAND, band_data,
transaction)
band_data['_id'] = '{0}/{1}'.format('rv_users', user_id)
ArangoCore.update_vertex_in_collection(ArangoVertex.USER, band_data,
transaction)
def save_user(cls, user_id, user_data=None, is_new=True):
try:
user_data = cls.prepare_data_to_save(user_id, user_data, is_new)
database = ArangoCore.get_database()
collections = [ArangoVertex.USER]
with database.transaction(write=collections,
commit_on_error=False) as transaction:
if is_new:
ArangoCore.add_vertex_to_collection(
ArangoVertex.USER, user_data, transaction)
else:
ArangoCore.update_vertex_in_collection(
ArangoVertex.USER, user_data, transaction)
user_data['_id'] = '{0}/{1}'.format(
'rv_users', user_data['id'])
transaction.commit()
return True
except Exception as exception:
raise exception
