ruff
This commit is contained in:
parent
18635cd52e
commit
e9380a127a
|
@ -2,8 +2,8 @@
|
|||
|
||||
import datetime
|
||||
import logging
|
||||
from decimal import Decimal
|
||||
from collections.abc import Iterator
|
||||
from decimal import Decimal
|
||||
|
||||
import fin_defs
|
||||
import krakenex
|
||||
|
@ -64,9 +64,8 @@ class KrakenDepoFetcher(DepoFetcher):
|
|||
updated_time=now,
|
||||
)
|
||||
|
||||
|
||||
def _get_withdrawals(self) -> list[WithdrawalDetails]:
|
||||
raise NotImplementedError("work in progress")
|
||||
raise NotImplementedError('work in progress')
|
||||
"""
|
||||
json = self.client.query_private('WithdrawStatus',data={'start':1})
|
||||
results = []
|
||||
|
@ -82,7 +81,7 @@ class KrakenDepoFetcher(DepoFetcher):
|
|||
"""
|
||||
|
||||
def _get_deposits(self) -> list[DepositDetails]:
|
||||
raise NotImplementedError("work in progress")
|
||||
raise NotImplementedError('work in progress')
|
||||
"""
|
||||
json = self.client.query_private('DepositStatus',data={'start':1})
|
||||
results = []
|
||||
|
@ -98,7 +97,7 @@ class KrakenDepoFetcher(DepoFetcher):
|
|||
"""
|
||||
|
||||
def _get_historic_spot_orders(self) -> Iterator[TradeOrderDetails]:
|
||||
raise NotImplementedError("work in progress")
|
||||
raise NotImplementedError('work in progress')
|
||||
"""
|
||||
json = self.client.query_private('ClosedOrders',data={'start':1})
|
||||
for order_id, v in json['result']['closed'].items():
|
||||
|
@ -121,12 +120,14 @@ class KrakenDepoFetcher(DepoFetcher):
|
|||
json = self.client.query_private('Ledgers', data={'start': 1})
|
||||
return list(parse_from_ledger(json['result']['ledger'].values()))
|
||||
|
||||
|
||||
def parse_asset_from_ticker(ticker: str) -> fin_defs.Asset:
|
||||
account = ticker.removesuffix('.HOLD')
|
||||
if account == 'ZEUR':
|
||||
return fin_defs.EUR
|
||||
return fin_defs.WELL_KNOWN_SYMBOLS[account]
|
||||
|
||||
|
||||
def parse_from_ledger(ledger_items: list[dict[str, str]]) -> Iterator:
|
||||
collected_items = {}
|
||||
for item in ledger_items:
|
||||
|
@ -141,7 +142,9 @@ def parse_from_ledger(ledger_items: list[dict[str,str]]) -> Iterator:
|
|||
yield DepositDetails(
|
||||
deposit=fin_defs.AssetAmount(asset, Decimal(items[0]['amount'])),
|
||||
fee=fin_defs.AssetAmount(asset, Decimal(items[0]['fee'])),
|
||||
executed_time = datetime.datetime.fromtimestamp(items[0]['time'],tz=datetime.UTC),
|
||||
executed_time=datetime.datetime.fromtimestamp(
|
||||
items[0]['time'], tz=datetime.UTC,
|
||||
),
|
||||
raw_details=items[0],
|
||||
)
|
||||
elif items[0]['type'] == 'withdrawal':
|
||||
|
@ -150,7 +153,9 @@ def parse_from_ledger(ledger_items: list[dict[str,str]]) -> Iterator:
|
|||
yield WithdrawalDetails(
|
||||
withdrawn=-fin_defs.AssetAmount(asset, Decimal(items[0]['amount'])),
|
||||
fee=fin_defs.AssetAmount(asset, Decimal(items[0]['fee'])),
|
||||
executed_time = datetime.datetime.fromtimestamp(items[0]['time'],tz=datetime.UTC),
|
||||
executed_time=datetime.datetime.fromtimestamp(
|
||||
items[0]['time'], tz=datetime.UTC,
|
||||
),
|
||||
raw_details=items[0],
|
||||
)
|
||||
else:
|
||||
|
@ -170,7 +175,9 @@ def parse_from_ledger(ledger_items: list[dict[str,str]]) -> Iterator:
|
|||
input=input,
|
||||
output=output,
|
||||
fee=fee_of_input if fee_of_input.amount != 0 else fee_of_output,
|
||||
executed_time = datetime.datetime.fromtimestamp(items[1]['time'],tz=datetime.UTC),
|
||||
executed_time=datetime.datetime.fromtimestamp(
|
||||
items[1]['time'], tz=datetime.UTC,
|
||||
),
|
||||
order_id=refid,
|
||||
raw_order_details=items[0],
|
||||
)
|
||||
|
|
|
@ -244,7 +244,9 @@ class KucoinDepoFetcher(DepoFetcher):
|
|||
timestamp = int(end_time.timestamp() * 1000)
|
||||
page_index = 1
|
||||
while True:
|
||||
raw_details = self.kucoin_client.get_orders(end=timestamp, page=page_index)
|
||||
raw_details = self.kucoin_client.get_orders(
|
||||
end=timestamp, page=page_index,
|
||||
)
|
||||
yield from (order_from_json(item) for item in raw_details['items'])
|
||||
|
||||
page_index = raw_details['currentPage'] + 1
|
||||
|
|
|
@ -9,6 +9,7 @@ needs_secrets = pytest.mark.skipif(
|
|||
reason='Secret kraken_USERNAME required',
|
||||
)
|
||||
|
||||
|
||||
def get_fetcher():
|
||||
return fin_depo.defi_kraken.KrakenDepoFetcher(
|
||||
secrets.KRAKEN_KEY,
|
||||
|
|
|
@ -32,7 +32,7 @@ def test_place_market_order_requires_allow_trades():
|
|||
|
||||
with pytest.raises(PermissionError) as m:
|
||||
fetcher.place_market_order(
|
||||
fin_defs.AssetAmount(fin_defs.MPC, Decimal(1)), fin_defs.USDT
|
||||
fin_defs.AssetAmount(fin_defs.MPC, Decimal(1)), fin_defs.USDT,
|
||||
)
|
||||
|
||||
assert 'KucoinDepoFetcher.allow_trades is not enabled: Cannot make trades' in str(m)
|
||||
|
|
Loading…
Reference in New Issue
Block a user