mirror of
https://github.com/discountry/ritmex-bot.git
synced 2026-09-09 16:28:06 +00:00
251 lines
8.7 KiB
Markdown
251 lines
8.7 KiB
Markdown
- [Cli](./examples/py/)
|
|
|
|
|
|
```python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import argparse
|
|
import os
|
|
import re
|
|
import sys
|
|
import json
|
|
import platform
|
|
from pprint import pprint
|
|
import asyncio
|
|
import ccxt.pro as ccxtpro
|
|
import ccxt.async_support as ccxt # noqa: E402
|
|
|
|
class Argv(object):
|
|
|
|
table = False
|
|
verbose = False
|
|
sandbox = False
|
|
demo = False
|
|
testnet = False
|
|
test = False
|
|
nonce = None
|
|
exchange_id = ''
|
|
debug = False
|
|
cors = False
|
|
method = None
|
|
symbol = None
|
|
spot = False
|
|
swap = False
|
|
future = False
|
|
signIn = False
|
|
args = []
|
|
no_keys = False
|
|
raw = False
|
|
no_load_markets = False
|
|
|
|
|
|
argv = Argv()
|
|
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument('--table', action='store_true', help='output as table')
|
|
parser.add_argument('--cors', action='store_true', help='enable CORS proxy')
|
|
parser.add_argument('--verbose', action='store_true', help='enable verbose output')
|
|
parser.add_argument('--debug', action='store_true', help='enable debug output')
|
|
parser.add_argument('--sandbox', action='store_true', help='enable sandbox/testnet')
|
|
parser.add_argument('--demo', action='store_true', help='enable demo mode')
|
|
parser.add_argument('--testnet', action='store_true', help='enable sandbox/testnet')
|
|
parser.add_argument('--test', action='store_true', help='enable sandbox/testnet')
|
|
parser.add_argument('--spot', action='store_true', help='enable spot markets')
|
|
parser.add_argument('--swap', action='store_true', help='enable swap markets')
|
|
parser.add_argument('--future', action='store_true', help='enable future markets')
|
|
parser.add_argument('--option', action='store_true', help='enable option markets')
|
|
parser.add_argument('--signIn', action='store_true', help='sign in')
|
|
parser.add_argument('--no-keys', action='store_true', help='don t load keys')
|
|
parser.add_argument('--raw', action='store_true', help='raw output')
|
|
parser.add_argument('--no-load-markets', action='store_true', help='no load markets')
|
|
parser.add_argument('exchange_id', type=str, help='exchange id in lowercase', nargs='?')
|
|
parser.add_argument('method', type=str, help='method or property', nargs='?')
|
|
parser.add_argument('args', type=str, help='arguments', nargs='*')
|
|
|
|
parser.parse_args(namespace=argv)
|
|
|
|
def table(values):
|
|
first = values[0]
|
|
keys = list(first.keys()) if isinstance(first, dict) else range(0, len(first))
|
|
widths = [max([len(str(v[k])) for v in values]) for k in keys]
|
|
string = ' | '.join(['{:<' + str(w) + '}' for w in widths])
|
|
return "\n".join([string.format(*[str(v[k]) for k in keys]) for v in values])
|
|
|
|
|
|
def print_supported_exchanges():
|
|
print('Supported exchanges: ' + ', '.join(ccxt.exchanges) + '\n')
|
|
|
|
def print_usage():
|
|
print('\nThis is an example of a basic command-line interface to all exchanges\n')
|
|
print('Usage:\n')
|
|
print('python ' + sys.argv[0] + ' exchange_id method "param1" param2 "param3" param4 ...\n')
|
|
print('Examples:\n')
|
|
print('python ' + sys.argv[0] + ' okcoin fetch_ohlcv BTC/USD 15m')
|
|
print('python ' + sys.argv[0] + ' bitfinex fetch_balance')
|
|
print('python ' + sys.argv[0] + ' kraken fetch_order_book ETH/BTC\n')
|
|
print_supported_exchanges()
|
|
|
|
async def main():
|
|
if not argv.raw:
|
|
print('Python v' + platform.python_version())
|
|
print('CCXT v' + ccxt.__version__)
|
|
|
|
# prefer local testing keys to global keys
|
|
keys_global = root + '/keys.json'
|
|
keys_local = root + '/keys.local.json'
|
|
keys_file = keys_local if os.path.exists(keys_local) else keys_global
|
|
|
|
# load the api keys and other settings from a JSON config
|
|
with open(keys_file, encoding="utf-8") as file:
|
|
keys = json.load(file)
|
|
|
|
config = {
|
|
# 'verbose': argv.verbose, # set later, after load_markets
|
|
'timeout': 30000,
|
|
}
|
|
|
|
if not argv.exchange_id:
|
|
print_usage()
|
|
sys.exit()
|
|
|
|
# # check here if we have a arg like this: binance.fetchOrders()
|
|
# call_reg = "\s*(\w+)\s*\.\s*(\w+)\s*\(([^()]*)\)"
|
|
# match = re.match(call_reg, argv.exchange_id)
|
|
# if match is not None:
|
|
# groups = match.groups()
|
|
# argv.exchange_id = groups[0]
|
|
# argv.method = groups[1]
|
|
# argv.args = list(map(lambda x: x.strip().replace("'", "\""), groups[2].split(',')))
|
|
|
|
# ------------------------------------------------------------------------------
|
|
|
|
if argv.exchange_id not in ccxt.exchanges:
|
|
print_usage()
|
|
raise Exception('Exchange "' + argv.exchange_id + '" not found.')
|
|
|
|
if argv.exchange_id in keys:
|
|
config.update(keys[argv.exchange_id])
|
|
|
|
exchange = None
|
|
if (argv.exchange_id in ccxtpro.exchanges):
|
|
exchange = getattr(ccxtpro, argv.exchange_id)(config)
|
|
else:
|
|
exchange = getattr(ccxt, argv.exchange_id)(config)
|
|
|
|
if argv.spot:
|
|
exchange.options['defaultType'] = 'spot'
|
|
elif argv.swap:
|
|
exchange.options['defaultType'] = 'swap'
|
|
elif argv.future:
|
|
exchange.options['defaultType'] = 'future'
|
|
elif argv.option:
|
|
exchange.options['defaultType'] = 'option'
|
|
|
|
if not argv.no_keys:
|
|
# check auth keys in env var
|
|
requiredCredentials = exchange.requiredCredentials
|
|
for credential, isRequired in requiredCredentials.items():
|
|
if isRequired and credential and not getattr(exchange, credential, None):
|
|
credentialEnvName = (argv.exchange_id + '_' + credential).upper() # example: KRAKEN_APIKEY
|
|
if credentialEnvName in os.environ:
|
|
credentialValue = os.environ[credentialEnvName]
|
|
if credentialValue.startswith('-----BEGIN'):
|
|
credentialValue = credentialValue.replace('\\n', '\n')
|
|
|
|
setattr(exchange, credential, credentialValue)
|
|
|
|
if argv.cors:
|
|
exchange.proxy = 'https://cors-anywhere.herokuapp.com/'
|
|
exchange.origin = exchange.uuid()
|
|
|
|
# pprint(dir(exchange))
|
|
|
|
# ------------------------------------------------------------------------------
|
|
|
|
args = []
|
|
|
|
for arg in argv.args:
|
|
|
|
# unpack json objects (mostly for extra params)
|
|
if arg[0] == '{' or arg[0] == '[':
|
|
args.append(json.loads(arg))
|
|
elif arg == 'None':
|
|
args.append(None)
|
|
elif re.match(r'^\'(.)+\'$', arg):
|
|
args.append(str(arg.replace('\'', '')))
|
|
elif re.match(r'^"(.)+"$', arg):
|
|
args.append(str(arg.replace('"', '')))
|
|
elif re.match(r'^[0-9+-]+$', arg):
|
|
args.append(int(arg))
|
|
elif re.match(r'^[.eE0-9+-]+$', arg):
|
|
args.append(float(arg))
|
|
elif re.match(r'^[0-9]{4}[-]?[0-9]{2}[-]?[0-9]{2}[T\s]?[0-9]{2}[:]?[0-9]{2}[:]?[0-9]{2}', arg):
|
|
args.append(exchange.parse8601(arg))
|
|
else:
|
|
args.append(arg)
|
|
|
|
if argv.testnet or argv.sandbox or argv.test:
|
|
exchange.set_sandbox_mode(True)
|
|
elif argv.demo:
|
|
exchange.enable_demo_trading(True)
|
|
|
|
if argv.verbose and argv.debug:
|
|
exchange.verbose = argv.verbose
|
|
|
|
if not argv.no_load_markets:
|
|
markets_path = '.cache/' + exchange.id + '-markets.json'
|
|
if os.path.exists(markets_path):
|
|
with open(markets_path, 'r') as f:
|
|
exchange.markets = json.load(f)
|
|
else:
|
|
await exchange.load_markets()
|
|
|
|
exchange.verbose = argv.verbose # now set verbose mode
|
|
|
|
if argv.signIn:
|
|
await exchange.sign_in()
|
|
|
|
is_ws_method = False
|
|
|
|
if argv.method:
|
|
method = getattr(exchange, argv.method)
|
|
# if it is a method, call it
|
|
if callable(method):
|
|
if argv.method.startswith('watch'):
|
|
is_ws_method = True # handle ws methods
|
|
if not argv.raw:
|
|
print(f"{argv.exchange_id}.{argv.method}({','.join(map(str, args))})")
|
|
|
|
while True:
|
|
result = method(*args)
|
|
if asyncio.iscoroutine(result):
|
|
result = await result
|
|
if argv.table:
|
|
result = list(result.values()) if isinstance(result, dict) else result
|
|
print(table([exchange.omit(v, 'info') for v in result]))
|
|
elif argv.raw:
|
|
print(exchange.json(result))
|
|
else:
|
|
pprint(result)
|
|
if not is_ws_method:
|
|
await exchange.close()
|
|
return
|
|
else: # otherwise it's a property, print it
|
|
result = method
|
|
if argv.table:
|
|
result = list(result.values()) if isinstance(result, dict) else result
|
|
print(table([exchange.omit(v, 'info') for v in result]))
|
|
elif argv.raw:
|
|
print(exchange.json(result))
|
|
else:
|
|
pprint(result)
|
|
await exchange.close()
|
|
else:
|
|
pprint(dir(exchange))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
asyncio.run(main())
|
|
|
|
``` |