cic-cli/cic/cmd/export.py

# standard imports
import logging
import importlib
import os

# local imports
from cic import (
        Proof,
        Processor,
        )
from cic.meta import Meta
from cic.attachment import Attachment
from cic.network import Network
from cic.token import Token

logg = logging.getLogger(__name__)


def process_args(argparser):
    argparser.add_argument('-d', '--directory', type=str, dest='directory', default='.', help='directory')
    argparser.add_argument('-o', '--output-directory', type=str, dest='output_directory', help='output directory')
    argparser.add_argument('-y', '--signer', type=str, dest='y', help='target-specific signer to use for export')
    argparser.add_argument('-p', type=str, help='RPC endpoint')
    argparser.add_argument('target', type=str, help='target network type')


def validate_args(args):
    pass


def init_writers_from_config(config):
    w = {
            'meta': None,
            'attachment': None,
            'proof': None,
            'ext': None,
        }
    for v in w.keys():
        k = 'CIC_CORE_{}_WRITER'.format(v.upper())
        (d, c) = config.get(k).rsplit('.', maxsplit=1)
        m = importlib.import_module(d)
        o = getattr(m, c)
        w[v] = o
        
    return w


def execute(config, eargs):
    modname = 'cic.ext.{}'.format(eargs.target)
    cmd_mod = importlib.import_module(modname)

    writers = init_writers_from_config(config)

    ct = Token(path=eargs.directory)
    cm = Meta(path=eargs.directory, writer=writers['meta'](path=eargs.output_directory))
    ca = Attachment(path=eargs.directory, writer=writers['attachment'](path=eargs.output_directory))
    cp = Proof(path=eargs.directory, attachments=ca, writer=writers['proof'](path=eargs.output_directory))
    cn = Network(path=eargs.directory)

    ct.load()
    cp.load()
    cm.load()
    ca.load()
    cn.load()

    chain_spec = None
    try:
        chain_spec = config.get('CHAIN_SPEC')
    except KeyError:
        chain_spec = cn.chain_spec
        config.add(chain_spec, 'CHAIN_SPEC', exists_ok=True)
        logg.debug('CHAIN_SPEC config set to {}'.format(str(chain_spec)))

    #signer = cmd_mod.parse_signer(eargs.y)
    (rpc, signer) = cmd_mod.parse_adapter(config, eargs.y)

    ref = cn.resource(eargs.target)
    chain_spec = cn.chain_spec(eargs.target)
    logg.debug('found reference {} chain spec {} for target {}'.format(ref['contents'], chain_spec, eargs.target))
    c = getattr(cmd_mod, 'new')(chain_spec, ref['contents'], cp, signer_hint=signer, rpc=rpc, outputs_writer=writers['ext'](path=eargs.output_directory))
    c.apply_token(ct)

    p = Processor(proof=cp, attachment=ca, metadata=cm, extensions=[c])
    p.process()