golemcli.py

#!/usr/bin/env python

import os
import argparse
import sys

from multiprocessing import freeze_support
import click

from golem.config.environments import set_environment  # noqa
from golem.core.simpleenv import get_local_datadir
from golem.rpc.cert import CertificateManager

from golem.rpc.common import CROSSBAR_HOST, CROSSBAR_PORT, CROSSBAR_DIR

# Export pbr version for peewee_migrate user
os.environ["PBR_VERSION"] = '3.1.1'

# pylint: disable=wrong-import-position, unused-import
from golem.core.common import config_logging, install_reactor  # noqa
from golem.interface.cli import CLI  # noqa
from golem.interface.client import debug  # noqa
from golem.interface.client.account import Account  # noqa
from golem.interface.client.environments import Environments  # noqa
from golem.interface.client.network import Network  # noqa
from golem.interface.client.payments import payments, incomes  # noqa
from golem.interface.client.resources import Resources  # noqa
from golem.interface.client.settings import Settings  # noqa
from golem.interface.client.tasks import Tasks, Subtasks  # noqa
from golem.interface.client.terms import Terms  # noqa
from golem.interface.client.test_task import TestTask  # noqa
from golem.interface.websockets import WebSocketCLI  # noqa


def start():
    freeze_support()
    delete_reactor()

    flags = dict(
        interactive=('-i', '--interactive'),
        mainnet=('-m', '--mainnet'),
        address=('-a', '--address'),
        port=('-p', '--port'),
        trust=('-t', '--verify-trust'),
        datadir=("-d", "--datadir")
    )

    flag_options = dict(
        interactive=dict(dest="interactive", action="store_true",
                         default=False, help="Enter interactive mode"),
        mainnet=dict(dest="mainnet", action="store_true", default=False,
                     help="Use mainnet chain"),
        address=dict(dest="address", type=str, default=CROSSBAR_HOST,
                     help="Golem node's RPC address"),
        port=dict(dest="port", type=int, default=CROSSBAR_PORT,
                  help="Golem node's RPC port"),
        datadir=dict(dest="datadir", default=None,
                     type=click.Path(
                         file_okay=False,
                         writable=True
                     ),
                     help="Golem node's data dir"),
        trust=dict(dest="verify_trust", action="store_true", default=False,
                   help="Verify Golem node's certificate"),
    )

    # process initial arguments
    parser = argparse.ArgumentParser(add_help=False)
    for flag_name, flag in flags.items():
        parser.add_argument(*flag, **flag_options[flag_name])

    args = sys.argv[1:]
    parsed, forwarded = parser.parse_known_args(args)

    install_reactor()

    # platform trust settings
    if not parsed.verify_trust:
        disable_platform_trust()

    # setup logging if in interactive mode
    interactive = parsed.interactive

    if interactive:
        config_logging("_cli")
        cli = CLI()
    else:
        import logging
        logging.raiseExceptions = 0
        cli = CLI(main_parser=parser, main_parser_options=flag_options)

    if parsed.mainnet:
        set_environment('mainnet', None)

    datadir = get_local_datadir('default', root_dir=parsed.datadir)
    working_dir = os.path.join(datadir, CROSSBAR_DIR)

    # run the cli
    ws_cli = WebSocketCLI(
        cli,
        CertificateManager(working_dir),
        host=parsed.address,
        port=parsed.port
    )
    ws_cli.execute(forwarded, interactive=interactive)


def disable_platform_trust():
    from twisted.internet import _sslverify  # pylint: disable=protected-access
    _sslverify.platformTrust = lambda: None


def delete_reactor():
    if 'twisted.internet.reactor' in sys.modules:
        del sys.modules['twisted.internet.reactor']


if __name__ == '__main__':
    start()