mirror of https://github.com/CIRCL/AIL-framework
55 lines
2.2 KiB
Python
Executable File
55 lines
2.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*-coding:UTF-8 -*
|
|
|
|
import os
|
|
import sys
|
|
import uuid
|
|
|
|
sys.path.append(os.path.join(os.environ['AIL_BIN'], 'lib'))
|
|
sys.path.append(os.path.join(os.environ['AIL_BIN'], 'packages'))
|
|
import ConfigLoader
|
|
import Correlate_object
|
|
|
|
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 = Correlate_object.get_obj_global_id(obj_type, obj_id, obj_sub_type=obj_subtype)
|
|
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 = Correlate_object.get_obj_global_id(obj_type, obj_id, obj_sub_type=obj_subtype)
|
|
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))
|