mirror of https://github.com/CIRCL/AIL-framework
63 lines
1.8 KiB
Python
Executable File
63 lines
1.8 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import json
|
|
import logging
|
|
import os
|
|
import sys
|
|
import requests
|
|
|
|
sys.path.append(os.environ['AIL_BIN'])
|
|
|
|
logger = logging.getLogger()
|
|
|
|
blockchain_all = 'https://blockchain.info/rawaddr'
|
|
|
|
# pre-alpha script
|
|
|
|
# default nb_transactions = 50
|
|
def get_bitcoin_info(bitcoin_address, nb_transaction=50):
|
|
dict_btc = {}
|
|
set_btc_in = set()
|
|
set_btc_out = set()
|
|
try:
|
|
req = requests.get(f'{blockchain_all}/{bitcoin_address}?limit={nb_transaction}')
|
|
jreq = req.json()
|
|
except Exception as e:
|
|
logger.warning(e)
|
|
return dict_btc
|
|
|
|
if not jreq.get('n_tx'):
|
|
logger.critical(json.dumps(jreq))
|
|
return dict_btc
|
|
|
|
dict_btc['n_tx'] = jreq['n_tx']
|
|
dict_btc['total_received'] = float(jreq['total_received'] / 100000000)
|
|
dict_btc['total_sent'] = float(jreq['total_sent'] / 100000000)
|
|
dict_btc['final_balance'] = float(jreq['final_balance'] / 100000000)
|
|
|
|
for transaction in jreq['txs']:
|
|
for t_input in transaction['inputs']:
|
|
if 'addr' in t_input['prev_out']:
|
|
if t_input['prev_out']['addr'] != bitcoin_address:
|
|
set_btc_in.add(t_input['prev_out']['addr'])
|
|
for output in transaction['out']:
|
|
if 'addr' in output:
|
|
if output['addr'] != bitcoin_address:
|
|
set_btc_out.add(output['addr'])
|
|
|
|
dict_btc['btc_in'] = filter_btc_seen(set_btc_in)
|
|
dict_btc['btc_out'] = filter_btc_seen(set_btc_out)
|
|
return dict_btc
|
|
|
|
# filter btc seen in ail
|
|
def filter_btc_seen(btc_addr_set):
|
|
from lib.objects import CryptoCurrencies
|
|
|
|
list_seen_btc = []
|
|
for btc_addr in btc_addr_set:
|
|
cryptocurrency = CryptoCurrencies.CryptoCurrency(btc_addr, 'bitcoin')
|
|
if cryptocurrency.exists():
|
|
list_seen_btc.append(btc_addr)
|
|
return list_seen_btc
|