AIL-framework/tools/extract_cryptocurrency.py

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

import os
import sys
import json
import argparse

sys.path.append(os.path.join(os.environ['AIL_BIN'], 'packages/'))
import Cryptocurrency

sys.path.append(os.path.join(os.environ['AIL_BIN'], 'lib/'))
import ConfigLoader
import Correlate_object

mode = 'union'

def sanitise_int(page, default_value):
    try:
        page = int(page)
    except:
        page = default_value
    if page < 1:
        page = default_value
    return page

def sanitise_nb_max_nodes(nb_max_nodes):
    try:
        nb_max_nodes = int(nb_max_nodes)
        if nb_max_nodes < 2:
            nb_max_nodes = 300
    except:
        nb_max_nodes = 300
    return nb_max_nodes

def get_object_correlation_json(correlation_id, subtype, max_nodes):
    object_type = 'cryptocurrency'
    max_nodes = sanitise_nb_max_nodes(max_nodes)

    # ALL correlations
    correlation_names = Correlate_object.sanitise_correlation_names('')
    #correlation_objects = Correlate_object.sanitise_correlation_objects('')
    correlation_objects = ['domain']

    res = Correlate_object.get_graph_node_object_correlation(object_type, correlation_id, mode, correlation_names,
                                                    correlation_objects, requested_correl_type=subtype,
                                                    max_nodes=max_nodes, flask_context=False)
    return res

if __name__ == '__main__':

    parser = argparse.ArgumentParser(description='Trigger backgroud update')
    parser.add_argument('-t', '--type', help='Cryptocurrency type (bitcoin, bitcoin-cash, ethereum, litecoin, monero, dash, zcash)', type=str, dest='type', required=True, default=None)
    parser.add_argument('-a', '--address', help='Cryptocurrency addresses', type=str, dest='address', default=None, nargs="*")
    parser.add_argument('-p', '--page',help='page number, default=1' , type=int, default=1, dest='page')
    parser.add_argument('-n', '--nb',help='number of addresses by page, default=50' , type=int, default=50, dest='nb_elem')
    parser.add_argument('-fo', '--filter_objects',help='filter correlation by object : domain, paste/item' , type=str, default=[], dest='objects', nargs="*")
    parser.add_argument('--node' ,help='correlation graph: max number of nodes, default=50' , type=int, default=50, dest='max_nodes')
    args = parser.parse_args()

    subtype = args.type
    if subtype is None:
        parser.print_help()
        sys.exit(0)
    else:
        res = Cryptocurrency.cryptocurrency.api_check_objs_type([args.type])
        if res:
            print(json.dumps(res[0]))
            sys.exit(0)

    page = sanitise_int(args.page, 1)
    nb_elem = sanitise_int(args.nb_elem, 50)
    max_nodes = sanitise_int(args.max_nodes, 300)
    if args.objects:
        res = Correlate_object.api_check_correlation_objects(args.objects)
        if res:
            print(json.dumps(res[0]))
            sys.exit(0)

    dict_json = {}
    if args.address:
        l_addresse = Cryptocurrency.cryptocurrency.paginate_list(args.address, nb_elem=nb_elem, page=page)
    else:
        l_addresse = Cryptocurrency.cryptocurrency.get_all_correlations_by_subtype_pagination(subtype, nb_elem=nb_elem, page=page)
    for address in l_addresse:
        dict_json[address] = get_object_correlation_json(address, subtype, max_nodes)

    print(json.dumps(dict_json))