Split main class.

basicswap/base.py

@@ -0,0 +1,115 @@
+# -*- coding: utf-8 -*-
+
+# Copyright (c) 2019 tecnovert
+# Distributed under the MIT software license, see the accompanying
+# file LICENSE.txt or http://www.opensource.org/licenses/mit-license.php.
+
+import os
+import threading
+import logging
+import subprocess
+
+import basicswap.config as cfg
+
+from .chainparams import (
+    chainparams,
+    Coins,
+)
+from .util import (
+    callrpc,
+)
+
+
+class BaseApp:
+    def __init__(self, fp, data_dir, settings, chain, log_name='BasicSwap'):
+        self.log_name = log_name
+        self.fp = fp
+        self.is_running = True
+        self.fail_code = 0
+
+        self.data_dir = data_dir
+        self.chain = chain
+        self.settings = settings
+        self.coin_clients = {}
+        self.mxDB = threading.RLock()
+        self.debug = self.settings.get('debug', cfg.DEBUG)
+
+        self.prepareLogging()
+        self.log.info('Network: {}'.format(self.chain))
+
+    def stopRunning(self, with_code=0):
+        self.fail_code = with_code
+        self.is_running = False
+
+    def prepareLogging(self):
+        self.log = logging.getLogger(self.log_name)
+        self.log.propagate = False
+
+        formatter = logging.Formatter('%(asctime)s %(levelname)s : %(message)s')
+        stream_stdout = logging.StreamHandler()
+        if self.log_name != 'BasicSwap':
+            stream_stdout.setFormatter(logging.Formatter('%(asctime)s %(name)s %(levelname)s : %(message)s'))
+        else:
+            stream_stdout.setFormatter(formatter)
+        stream_fp = logging.StreamHandler(self.fp)
+        stream_fp.setFormatter(formatter)
+
+        self.log.setLevel(logging.DEBUG if self.debug else logging.INFO)
+        self.log.addHandler(stream_fp)
+        self.log.addHandler(stream_stdout)
+
+    def getChainClientSettings(self, coin):
+        try:
+            return self.settings['chainclients'][chainparams[coin]['name']]
+        except Exception:
+            return {}
+
+    def setDaemonPID(self, name, pid):
+        if isinstance(name, Coins):
+            self.coin_clients[name]['pid'] = pid
+            return
+        for c, v in self.coin_clients.items():
+            if v['name'] == name:
+                v['pid'] = pid
+
+    def getChainDatadirPath(self, coin):
+        datadir = self.coin_clients[coin]['datadir']
+        testnet_name = '' if self.chain == 'mainnet' else chainparams[coin][self.chain].get('name', self.chain)
+        return os.path.join(datadir, testnet_name)
+
+    def getTicker(self, coin_type):
+        ticker = chainparams[coin_type]['ticker']
+        if self.chain == 'testnet':
+            ticker = 't' + ticker
+        if self.chain == 'regtest':
+            ticker = 'rt' + ticker
+        return ticker
+
+    def callrpc(self, method, params=[], wallet=None):
+        return callrpc(self.coin_clients[Coins.PART]['rpcport'], self.coin_clients[Coins.PART]['rpcauth'], method, params, wallet)
+
+    def callcoinrpc(self, coin, method, params=[], wallet=None):
+        return callrpc(self.coin_clients[coin]['rpcport'], self.coin_clients[coin]['rpcauth'], method, params, wallet)
+
+    def calltx(self, cmd):
+        bindir = self.coin_clients[Coins.PART]['bindir']
+        command_tx = os.path.join(bindir, cfg.PARTICL_TX)
+        chainname = '' if self.chain == 'mainnet' else (' -' + self.chain)
+        args = command_tx + chainname + ' ' + cmd
+        p = subprocess.Popen(args, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
+        out = p.communicate()
+        if len(out[1]) > 0:
+            raise ValueError('TX error ' + str(out[1]))
+        return out[0].decode('utf-8').strip()
+
+    def callcoincli(self, coin_type, params, wallet=None, timeout=None):
+        bindir = self.coin_clients[coin_type]['bindir']
+        datadir = self.coin_clients[coin_type]['datadir']
+        command_cli = os.path.join(bindir, chainparams[coin_type]['name'] + '-cli' + ('.exe' if os.name == 'nt' else ''))
+        chainname = '' if self.chain == 'mainnet' else (' -' + self.chain)
+        args = command_cli + chainname + ' ' + '-datadir=' + datadir + ' ' + params
+        p = subprocess.Popen(args, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
+        out = p.communicate(timeout=timeout)
+        if len(out[1]) > 0:
+            raise ValueError('CLI error ' + str(out[1]))
+        return out[0].decode('utf-8').strip()