bigchaindb-examples/init_assets.py at master · bigchaindb/bigchaindb-examples · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
import random
import logging

import bigchaindb
import bigchaindb.config_utils

import apps_config
from server.lib.models.accounts import retrieve_accounts
from server.lib.models.assets import create_asset
from server.config_bigchaindb import get_bigchain

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

APPS = apps_config.APPS


def get_accounts_by_name(accounts):
    # returns a dict with key = 'name-<ledger_id>' value = account
    return {'{}-{}'.format(account['name'], account['ledger']['id']): account for account in accounts}


def main():
    for app in APPS:
        app_name = '{}'.format(app['name'])
        if 'num_accounts' in app:
            ledger_name = 'bigchaindb_examples_{}'.format(app['ledger'])
            bigchain = get_bigchain(ledger_id=app['ledger'])
            accounts = retrieve_accounts(bigchain, app_name)
            assets = []
            for i in range(app['num_assets']):
                asset = create_asset(bigchain=bigchain,
                                     to=accounts[random.randint(0, app['num_accounts'] - 1)]['vk'],
                                     payload=app['payload_func'](i))
                assets.append(asset)
            logging.info('{} assets initialized for app {} on ledger {}'.format(len(assets),
                                                                                app_name,
                                                                                ledger_name))
        elif 'accounts' in app:
            bigchain = bigchaindb.Bigchain()
            accounts_by_name = get_accounts_by_name(retrieve_accounts(bigchain, app['name']))
            for account in app['accounts']:
                for ledger in account['ledgers']:
                    ledger_name = 'bigchaindb_examples_{}'.format(ledger['id'])
                    account_name = '{}-{}'.format(account['name'], ledger['id'])
                    bigchain = bigchaindb.Bigchain(dbname=ledger_name)
                    assets = []
                    for i in range(ledger['num_assets']):
                        asset = create_asset(bigchain=bigchain,
                                             to=accounts_by_name[account_name]['vk'],
                                             payload=app['payload_func'](i))
                        assets.append(asset)
                    logging.info('{} assets initialized for account {} in app {} on ledger {}'
                                 .format(len(assets), account['name'], app_name, ledger_name))


if __name__ == '__main__':
    main()