Exemplo n.º 1
0
def get_user_list_of_obj_to_export(user_id, add_uuid=False):
    set_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())
        set_objs_to_export.append(dict_obj)
    return set_objs_to_export
Exemplo n.º 2
0
def import_object_file():
    error = None

    is_file = False
    if 'file' in request.files:
        file = request.files['file']
        if file:
            if file.filename:
                is_file = True

    all_imported_obj = []
    if is_file:
        filename = MispImport.sanitize_import_file_path(file.filename)
        file.save(filename)
        map_uuid_global_id = MispImport.import_objs_from_file(filename)
        os.remove(filename)
        for obj_uuid in map_uuid_global_id:
            dict_obj = Correlate_object.get_global_id_from_id(
                map_uuid_global_id[obj_uuid])
            dict_obj['uuid'] = obj_uuid
            dict_obj['url'] = Correlate_object.get_item_url(
                dict_obj['type'],
                dict_obj['id'],
                correlation_type=dict_obj['subtype'])
            dict_obj['node'] = Correlate_object.get_correlation_node_icon(
                dict_obj['type'],
                correlation_type=dict_obj['subtype'],
                value=dict_obj['id'])
            all_imported_obj.append(dict_obj)

        if not all_imported_obj:
            error = "error: Empty or invalid JSON file"

    return render_template("import_object.html",
                           all_imported_obj=all_imported_obj,
                           error=error)