Load configuration from json
This commit is contained in:
parent
5e6b88b78a
commit
f236586183
|
@ -86,7 +86,6 @@ import logging
|
||||||
import random
|
import random
|
||||||
from collections.abc import Callable
|
from collections.abc import Callable
|
||||||
from decimal import Decimal
|
from decimal import Decimal
|
||||||
from pathlib import Path
|
|
||||||
from typing import Any
|
from typing import Any
|
||||||
|
|
||||||
import fin_defs
|
import fin_defs
|
||||||
|
@ -99,6 +98,7 @@ logger = logging.getLogger(__name__)
|
||||||
################################################################################
|
################################################################################
|
||||||
# Main code #
|
# Main code #
|
||||||
|
|
||||||
|
|
||||||
@dataclasses.dataclass(frozen=True)
|
@dataclasses.dataclass(frozen=True)
|
||||||
class AutoSellConfig:
|
class AutoSellConfig:
|
||||||
"""Configuration for `run_auto_sell`.
|
"""Configuration for `run_auto_sell`.
|
||||||
|
@ -143,19 +143,19 @@ class AutoSellRunResults:
|
||||||
|
|
||||||
def total_input_amount(self) -> Decimal:
|
def total_input_amount(self) -> Decimal:
|
||||||
"""Total amount of assets sold."""
|
"""Total amount of assets sold."""
|
||||||
return sum(o.input_amount for o in self.order_details)
|
return sum((o.input_amount for o in self.order_details), start=Decimal(0))
|
||||||
|
|
||||||
def total_output_amount(self) -> Decimal:
|
def total_output_amount(self) -> Decimal:
|
||||||
"""Total amount of assets "bought"."""
|
"""Total amount of assets "bought"."""
|
||||||
return sum(o.output_amount for o in self.order_details)
|
return sum((o.output_amount for o in self.order_details), start=Decimal(0))
|
||||||
|
|
||||||
|
|
||||||
def sample_from_range(rng: random.Random, range: tuple[Any, Any]) -> Any:
|
def sample_from_range(rng: random.Random, rang: tuple[Any, Any]) -> Any:
|
||||||
"""Samples uniformly from the given range."""
|
"""Samples uniformly from the given range."""
|
||||||
multiplier: float | Decimal = rng.random()
|
multiplier: float | Decimal = rng.random()
|
||||||
if isinstance(range[0], Decimal):
|
if isinstance(rang[0], Decimal):
|
||||||
multiplier = Decimal(multiplier)
|
multiplier = Decimal(multiplier)
|
||||||
return range[0] + (range[1] - range[0]) * multiplier
|
return rang[0] + (rang[1] - rang[0]) * multiplier
|
||||||
|
|
||||||
|
|
||||||
def run_auto_sell(config: AutoSellConfig) -> AutoSellRunResults:
|
def run_auto_sell(config: AutoSellConfig) -> AutoSellRunResults:
|
||||||
|
|
|
@ -1,7 +1,7 @@
|
||||||
import argparse
|
import argparse
|
||||||
import datetime
|
import datetime
|
||||||
import logging
|
|
||||||
import json
|
import json
|
||||||
|
import logging
|
||||||
import logging.handlers
|
import logging.handlers
|
||||||
import time
|
import time
|
||||||
from decimal import Decimal
|
from decimal import Decimal
|
||||||
|
@ -31,6 +31,7 @@ PATH_TRADES_FILE = PATH_OUTPUT / 'trades.csv'
|
||||||
################################################################################
|
################################################################################
|
||||||
# Application Setup #
|
# Application Setup #
|
||||||
|
|
||||||
|
|
||||||
def setup_logging():
|
def setup_logging():
|
||||||
"""Enables logging for the terminal and to a log file."""
|
"""Enables logging for the terminal and to a log file."""
|
||||||
PATH_LOG_FILE.parent.mkdir(parents=True, exist_ok=True)
|
PATH_LOG_FILE.parent.mkdir(parents=True, exist_ok=True)
|
||||||
|
@ -45,6 +46,7 @@ def setup_logging():
|
||||||
stream_handler = logging.StreamHandler()
|
stream_handler = logging.StreamHandler()
|
||||||
try:
|
try:
|
||||||
import logging_color
|
import logging_color
|
||||||
|
|
||||||
stream_handler = logging_color.ColorStreamHandler()
|
stream_handler = logging_color.ColorStreamHandler()
|
||||||
except ImportError:
|
except ImportError:
|
||||||
pass
|
pass
|
||||||
|
@ -67,17 +69,21 @@ def load_config(config_path: Path) -> AutoSellConfig:
|
||||||
kucoin_secret=config.KUCOIN_SECRET,
|
kucoin_secret=config.KUCOIN_SECRET,
|
||||||
kucoin_key=config.KUCOIN_KEY,
|
kucoin_key=config.KUCOIN_KEY,
|
||||||
kucoin_pass=config.KUCOIN_PASS,
|
kucoin_pass=config.KUCOIN_PASS,
|
||||||
allow_trades = True,
|
allow_trades=True,
|
||||||
)
|
)
|
||||||
|
|
||||||
with open(config_path) as f:
|
with open(config_path) as f:
|
||||||
json_config = json.load(f)
|
json_config = json.load(f)
|
||||||
|
|
||||||
return AutoSellConfig(
|
return AutoSellConfig(
|
||||||
input_amount_range=(Decimal(json_config['input_amount_low']),
|
input_amount_range=(
|
||||||
Decimal(json_config['input_amount_high'])),
|
Decimal(json_config['input_amount_low']),
|
||||||
interval_range=(datetime.timedelta(minutes=json_config['interval_minutes_low']),
|
Decimal(json_config['input_amount_high']),
|
||||||
datetime.timedelta(minutes=json_config['interval_minutes_high'])),
|
),
|
||||||
|
interval_range=(
|
||||||
|
datetime.timedelta(minutes=json_config['interval_minutes_low']),
|
||||||
|
datetime.timedelta(minutes=json_config['interval_minutes_high']),
|
||||||
|
),
|
||||||
input_asset=fin_defs.WELL_KNOWN_SYMBOLS[json_config['input_asset']],
|
input_asset=fin_defs.WELL_KNOWN_SYMBOLS[json_config['input_asset']],
|
||||||
output_asset=fin_defs.WELL_KNOWN_SYMBOLS[json_config['output_asset']],
|
output_asset=fin_defs.WELL_KNOWN_SYMBOLS[json_config['output_asset']],
|
||||||
exit_when_empty=True,
|
exit_when_empty=True,
|
||||||
|
@ -89,7 +95,7 @@ def load_config(config_path: Path) -> AutoSellConfig:
|
||||||
|
|
||||||
def parse_args():
|
def parse_args():
|
||||||
parser = argparse.ArgumentParser('crypto_seller', description=CLI_DESCRIPTION)
|
parser = argparse.ArgumentParser('crypto_seller', description=CLI_DESCRIPTION)
|
||||||
parser.add_argument('--config', type=Path, dest="config_file", required=True)
|
parser.add_argument('--config', type=Path, dest='config_file', required=True)
|
||||||
return parser.parse_args()
|
return parser.parse_args()
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -33,7 +33,7 @@ class SellerMock(fin_depo.data.DepoFetcher):
|
||||||
executed_time = datetime.datetime.now(tz=datetime.UTC)
|
executed_time = datetime.datetime.now(tz=datetime.UTC)
|
||||||
|
|
||||||
return fin_depo.data.TradeOrderDetails(
|
return fin_depo.data.TradeOrderDetails(
|
||||||
executed_time = executed_time,
|
executed_time=executed_time,
|
||||||
input_asset=input_asset,
|
input_asset=input_asset,
|
||||||
input_amount=input_amount,
|
input_amount=input_amount,
|
||||||
output_asset=output_asset,
|
output_asset=output_asset,
|
||||||
|
|
Loading…
Reference in New Issue
Block a user