crypto_seller/__init__.py

@@ -217,7 +217,8 @@ def run_auto_sell(config: AutoSellConfig) -> AutoSellRunResults:
             logger.info('Attempting to sell %s %s', amount_to_sell, config.input_asset)
 
             order_details = config.seller.place_market_order(
-                fin_defs.AssetAmount(config.input_asset, amount_to_sell),
+                config.input_asset,
+                amount_to_sell,
                 config.output_asset,
             )
 

crypto_seller/order_csv.py

@@ -32,12 +32,4 @@ class CsvFileLogger:
         # Write row in CSV file
         with open(self.path, 'a') as f:
             writer = csv.DictWriter(f, fieldnames=fieldnames)
-            d = trade_order.__dict__
-            d['input_asset'] = d['input'].asset
-            d['input_amount'] = d['input'].amount
-            d['output_asset'] = d['output'].asset
-            d['output_amount'] = d['output'].amount
-            d['fee_asset'] = d['fee'].asset
-            d['fee_amount'] = d['fee'].amount
-            del d['fee'], d['input'], d['output']
-            writer.writerow(d)
+            writer.writerow(trade_order.__dict__)

crypto_seller/secrets_config.py

@@ -2,11 +2,7 @@
 
 from secret_loader import SecretLoader
 
-__all__ = [
-    'KUCOIN_KEY',
-    'KUCOIN_PASS',
-    'KUCOIN_SECRET',
-]
+__all__ = ['KUCOIN_KEY', 'KUCOIN_SECRET', 'KUCOIN_PASS']
 
 secret_loader = SecretLoader()
 

test/test_auto_sell.py

@@ -23,19 +23,23 @@ class SellerMock(fin_depo.data.DepoFetcher):
 
     def place_market_order(
         self,
-        input_: fin_defs.AssetAmount,
+        input_asset: fin_defs.Asset,
+        input_amount: Decimal,
         output_asset: fin_defs.Asset,
-    ) -> fin_depo.data.TradeOrderDetails:
-        assert input_.amount <= self.amount_left, 'Attempt to sell too much'
-        self.amount_left -= input_.amount
+    ) -> fin_depo.data.Depo:
+        assert input_amount <= self.amount_left, 'Attempt to sell too much'
+        self.amount_left -= input_amount
 
         executed_time = datetime.datetime.now(tz=datetime.UTC)
 
         return fin_depo.data.TradeOrderDetails(
             executed_time=executed_time,
-            input=input_,
-            output=fin_defs.AssetAmount(output_asset, input_.amount),
-            fee=fin_defs.AssetAmount(input_.asset, Decimal(0)),
+            input_asset=input_asset,
+            input_amount=input_amount,
+            output_asset=output_asset,
+            output_amount=input_amount,
+            fee_asset=input_asset,
+            fee_amount=Decimal(0),
             order_id=10000000 - self.amount_left,
             raw_order_details={'TEST': 1, 'DATA': 2},
         )

test/test_init.py

@@ -1,2 +0,0 @@
-def test_init_main():
-    import crypto_seller.__main__  # noqa: F401

test/test_secrets.py

@@ -1,6 +1,7 @@
-import pytest
-
-
 def test_secrets():
-    with pytest.raises(ValueError, match='Failed to load secret with key:.*'):
-        import crypto_seller.secrets_config  # noqa: F401
+    try:
+        import crypto_seller.secrets_config
+    except ModuleNotFoundError:
+        raise
+    except:
+        pass