ail-framework/bin/export/AILObjects.py

#!/usr/bin/env python3
# -*-coding:UTF-8 -*

import os
import sys
import uuid

sys.path.append(os.environ['AIL_BIN'])
from lib.objects import ail_objects
from lib import ConfigLoader

config_loader = ConfigLoader.ConfigLoader()
r_serv_objects = config_loader.get_redis_conn("ARDB_Objects")
config_loader = None

def create_map_obj_uuid_golbal_id(obj_uuid, global_id):
    r_serv_objects.sadd('all_object:uuid', obj_uuid)
    r_serv_objects.sadd('all_object:global_id', global_id)
    r_serv_objects.sadd('object:map:uuid_id:{}'.format(obj_uuid), global_id)
    r_serv_objects.sadd('object:map:id_uuid:{}'.format(global_id), obj_uuid)

def create_map_obj_event_uuid(event_uuid, global_id):
    r_serv_objects.sadd('export:all_object:event_uuid', event_uuid)
    r_serv_objects.sadd('export:all_object:global_id', global_id)
    r_serv_objects.sadd('object:map:event_id:{}'.format(event_uuid), global_id)
    r_serv_objects.sadd('object:map:id_event:{}'.format(global_id), event_uuid)

# def get_user_list_of_obj_to_export(user_id, add_uuid=False):
#     objs_to_export = []
#     res = r_serv_objects.hgetall('user:all_objs_to_export:{}'.format(user_id))
#     for global_id in res:
#         dict_obj = Correlate_object.get_global_id_from_id(global_id)
#         dict_obj['lvl'] = int(res[global_id])
#         if add_uuid:
#             obj_dict['uuid'] = str(uuid.uuid4())
#         objs_to_export.append(dict_obj)
#     return objs_to_export

def add_user_object_to_export(user_id, obj_type, obj_id, lvl, obj_subtype=None):
    ## TODO: check if user exist
    # # TODO: check if valid object
    # # TODO: check lvl
    global_id = ail_objects.get_obj_global_id(obj_type, obj_subtype, obj_id)
    return r_serv_objects.hset('user:all_objs_to_export:{}'.format(user_id), global_id, lvl)

def delete_user_object_to_export(user_id, obj_type, obj_id, obj_subtype=None):
    ## TODO: check if user exist
    global_id = ail_objects.get_obj_global_id(obj_type, obj_subtype, obj_id)
    r_serv_objects.hdel('user:all_objs_to_export:{}'.format(user_id), global_id)

def delete_all_user_object_to_export(user_id):
    ## TODO: check if user exist
    r_serv_objects.delete('user:all_objs_to_export:{}'.format(user_id))