Output and secrets are now configurable

crypto_seller/__init__.py

@@ -216,50 +216,65 @@ def run_auto_sell(
     all_executed_orders: list[fin_depo.data.TradeOrderDetails] = []
     total_sleep_duration = datetime.timedelta(seconds=0)
 
-    while True:
-        # Check that account has tokens.
-        input_amount_available = config.seller.get_depo().get_amount_of_asset(
-            config.input_asset,
-        )
-        logger.info(
-            'Currently own %s %s',
-            input_amount_available,
-            config.input_asset.raw_short_name(),
-        )
-
-        if initial_rounds_to_skip > 0:
-            initial_rounds_to_skip -= 1
-            logger.info('skipping this round')
-        elif input_amount_available > 0:
-            amount_to_sell = sample_from_range(rng, config.input_amount_range)
-            amount_to_sell = min(input_amount_available, amount_to_sell)
-            amount_to_sell = amount_to_sell.quantize(
-                ROUND_TO_WHOLE,
-                rounding=ROUND_DOWN,
-            )
-            logger.info('Attempting to sell %s %s', amount_to_sell, config.input_asset)
-
-            order_details = config.seller.place_market_order(
+    try:
+        while True:
+            # Check that account has tokens.
+            input_amount_available = config.seller.get_depo().get_amount_of_asset(
                 config.input_asset,
-                amount_to_sell,
-                config.output_asset,
+            )
+            logger.info(
+                'Currently own %s %s',
+                input_amount_available,
+                config.input_asset.raw_short_name(),
             )
 
-            config.log_order_to_csv(order_details)
-            all_executed_orders.append(order_details)
+            if initial_rounds_to_skip > 0:
+                initial_rounds_to_skip -= 1
+                logger.info('skipping this round')
+            elif input_amount_available > 0:
+                amount_to_sell = sample_from_range(rng, config.input_amount_range)
+                amount_to_sell = min(input_amount_available, amount_to_sell)
+                amount_to_sell = amount_to_sell.quantize(
+                    ROUND_TO_WHOLE,
+                    rounding=ROUND_DOWN,
+                )
+                logger.info(
+                    'Attempting to sell %s %s',
+                    amount_to_sell,
+                    config.input_asset,
+                )
 
-            del amount_to_sell
-        elif config.exit_when_empty:
-            break
+                order_details = config.seller.place_market_order(
+                    config.input_asset,
+                    amount_to_sell,
+                    config.output_asset,
+                )
 
-        # Time out
-        time_to_sleep = sample_from_range(rng, config.interval_range)
-        time_to_sleep_secs = time_to_sleep.total_seconds()
-        logger.info('Sleeping %s (%d seconds)', time_to_sleep, time_to_sleep_secs)
-        config.sleep(time_to_sleep_secs)
-        total_sleep_duration += time_to_sleep
+                config.log_order_to_csv(order_details)
+                all_executed_orders.append(order_details)
 
-        del input_amount_available
+                del amount_to_sell
+            elif config.exit_when_empty:
+                break
+
+            # Time out
+            time_to_sleep = sample_from_range(rng, config.interval_range)
+            time_to_sleep_secs = time_to_sleep.total_seconds()
+            logger.info('Sleeping %s (%d seconds)', time_to_sleep, time_to_sleep_secs)
+            config.sleep(time_to_sleep_secs)
+            total_sleep_duration += time_to_sleep
+
+            del input_amount_available
+    except KeyboardInterrupt:
+        logger.warning('Manual interrupt')
+    except RuntimeError:
+        logger.exception(
+            'A unexpected and serious error occured. The program will be winding down',
+        )
+        logger.fatal('Please send the above error message to: mailto:jonjmaa@gmail.com')
+        logger.fatal(
+            'He will attempt to diagnosticate the problem, and help with recovery',
+        )
 
     time_end = datetime.datetime.now(tz=datetime.UTC)
 
@@ -288,41 +303,45 @@ def log_estimates(config: AutoSellConfig):
         current_balance / config.input_amount_range[0],
     )
 
+    minisleep = 0.1
+
+    logger.info('')
     logger.info('Welcome to crypto seller!')
-    config.sleep(1)
+    config.sleep(3)
     logger.info('')
-    config.sleep(1)
+    config.sleep(minisleep)
     logger.info('I, the great crypto seller, your humble servant, will')
+    config.sleep(minisleep)
     logger.info('now analyse your configuration and divine some estimates!')
-    config.sleep(1)
+    config.sleep(3)
     logger.info('')
-    config.sleep(1)
+    config.sleep(minisleep)
     logger.info(
         '- Current balance: %s %s',
         current_balance,
         config.input_asset.raw_short_name(),
     )
-    config.sleep(1)
+    config.sleep(minisleep)
     logger.info('- Average sleep:   %s seconds', average_sleep)
-    config.sleep(1)
+    config.sleep(minisleep)
     logger.info(
         '- Average amount:  %s %s',
         average_amount,
         config.input_asset.raw_short_name(),
     )
-    config.sleep(1)
+    config.sleep(minisleep)
     logger.info('- Expected counts: %s', expected_num_sell_offs)
-    config.sleep(1)
+    config.sleep(minisleep)
     logger.info('- Expected time:   %s', expected_duration)
-    config.sleep(1)
+    config.sleep(minisleep)
     logger.info('- Fastest time:    %s', fastest_duration)
-    config.sleep(1)
+    config.sleep(minisleep)
     logger.info('- Slowest time:    %s', slowest_duration)
-    config.sleep(1)
+    config.sleep(minisleep)
     logger.info('')
-    config.sleep(1)
+    config.sleep(minisleep)
     logger.info('Do you still want to proceed?')
-    config.sleep(1)
+    config.sleep(minisleep)
     logger.info('If not, press CTRL+C within the next 10 seconds...')
     config.sleep(10)
     logger.info('')

crypto_seller/__main__.py

@@ -3,6 +3,7 @@ import datetime
 import json
 import logging
 import logging.handlers
+import sys
 import time
 from decimal import Decimal
 from pathlib import Path
@@ -24,18 +25,18 @@ logger = logging.getLogger(__name__)
 ################################################################################
 # Constants #
 
-PATH_OUTPUT = Path('./output').absolute()
-PATH_LOG_FILE = PATH_OUTPUT / 'log.txt'
-PATH_TRADES_FILE = PATH_OUTPUT / 'trades.csv'
+PATH_FILE_LOG = 'log.txt'
+PATH_FILE_TRADES = 'trades.csv'
 
 ################################################################################
 # Application Setup #
 
 
-def setup_logging():
+def setup_logging(output_directory: Path):
     """Enables logging for the terminal and to a log file."""
-    PATH_LOG_FILE.parent.mkdir(parents=True, exist_ok=True)
-    file_handler = logging.handlers.WatchedFileHandler(filename=PATH_LOG_FILE)
+    path_log_file = output_directory / PATH_FILE_LOG
+    path_log_file.parent.mkdir(parents=True, exist_ok=True)
+    file_handler = logging.handlers.WatchedFileHandler(filename=path_log_file)
     file_handler.setFormatter(
         logging.Formatter(
             '%(levelname)s:%(asctime)s: %(message)s',
@@ -69,7 +70,7 @@ License  :  MIT License (see website for full text)
 """.strip()
 
 
-def load_config(config_path: Path) -> AutoSellConfig:
+def load_config(config_path: Path, output_directory: Path) -> AutoSellConfig:
     logger.info('Loading configuration')
 
     from . import secrets_config
@@ -98,7 +99,7 @@ def load_config(config_path: Path) -> AutoSellConfig:
         exit_when_empty=True,
         seller=seller_backend,
         sleep=time.sleep,
-        log_order_to_csv=order_csv.CsvFileLogger(PATH_TRADES_FILE),
+        log_order_to_csv=order_csv.CsvFileLogger(output_directory / PATH_FILE_TRADES),
     )
 
 
@@ -116,6 +117,13 @@ def parse_args():
         required=True,
         help='Trading configuration file',
     )
+    parser.add_argument(
+        '--output',
+        type=Path,
+        dest='output_directory',
+        required=True,
+        help='Directory for outputing logs and trades list',
+    )
     parser.add_argument(
         '--wait-before-first',
         action='store_true',
@@ -127,16 +135,23 @@ def parse_args():
 
 def main():
     """Initializes the program."""
-    setup_logging()
-
-    logger.info('Initializing crypto_seller')
     args = parse_args()
 
+    setup_logging(output_directory=args.output_directory)
+
+    logger.info('Initializing crypto_seller')
+
     # Load config
-    auto_sell_config = load_config(args.config_file)
+    auto_sell_config = load_config(
+        args.config_file,
+        output_directory=args.output_directory,
+    )
 
     # Display estimates
-    log_estimates(auto_sell_config)
+    try:
+        log_estimates(auto_sell_config)
+    except KeyboardInterrupt:
+        sys.exit(1)
 
     # Run auto sell
     results = run_auto_sell(

crypto_seller/secrets_config.py

@@ -4,7 +4,7 @@ from secret_loader import SecretLoader
 
 __all__ = ['KUCOIN_KEY', 'KUCOIN_SECRET', 'KUCOIN_PASS']
 
-secret_loader = SecretLoader()
+secret_loader = SecretLoader(ENV_KEY_PREFIX='CS')
 
 KUCOIN_KEY = secret_loader.load_or_fail('KUCOIN_KEY')
 KUCOIN_SECRET = secret_loader.load_or_fail('KUCOIN_SECRET')

requirements_test.txt

@@ -0,0 +1 @@
+pytest

test/test_auto_sell.py

@@ -65,7 +65,7 @@ def test_auto_run():
         exit_when_empty=True,
         seller=seller_mock,
         log_order_to_csv=crypto_seller.order_csv.CsvFileLogger(
-            Path('output/test-trades.csv'),
+            Path('test/output/test-trades.csv'),
         ),
         sleep=sleep_mock,
     )
@@ -82,4 +82,4 @@ def test_auto_run():
 
     # Check mocks agree
     assert seller_mock.amount_left == 0
-    assert sleep_mock.time_slept == 1000 + 23
+    assert sleep_mock.time_slept == 1000 + 17.2