CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2025-12-17 13:21:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: type rpc._freqtradebot for improved DX

Browse Source
This commit is contained in:
Matthias
2024-12-17 20:39:04 +01:00
parent d805ee6f3d
commit 44d1f169c3
1 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
freqtrade/rpc/rpc.py
View File

@@ -7,7 +7,7 @@ from abc import abstractmethod
from collections.abc import Generator, Sequence from collections.abc import Generator, Sequence
from datetime import date, datetime, timedelta, timezone from datetime import date, datetime, timedelta, timezone
from math import isnan from math import isnan
from typing import Any from typing import TYPE_CHECKING, Any
import psutil import psutil
from dateutil.relativedelta import relativedelta from dateutil.relativedelta import relativedelta
@@ -98,6 +98,10 @@ class RPC:
# Bind _fiat_converter if needed # Bind _fiat_converter if needed
_fiat_converter: CryptoToFiatConverter | None = None _fiat_converter: CryptoToFiatConverter | None = None
if TYPE_CHECKING:
from freqtrade.freqtradebot import FreqtradeBot
_freqtrade: FreqtradeBot
def __init__(self, freqtrade) -> None: def __init__(self, freqtrade) -> None:
""" """
@@ -201,7 +205,7 @@ class RPC:
# calculate profit and send message to user # calculate profit and send message to user
if trade.is_open: if trade.is_open:
try: try:
current_rate = self._freqtrade.exchange.get_rate( current_rate: float = self._freqtrade.exchange.get_rate(
trade.pair, side="exit", is_short=trade.is_short, refresh=False trade.pair, side="exit", is_short=trade.is_short, refresh=False
) )
except (ExchangeError, PricingError): except (ExchangeError, PricingError):
@@ -219,7 +223,7 @@ class RPC:
else: else:
# Closed trade ... # Closed trade ...
current_rate = trade.close_rate current_rate = trade.close_rate or 0.0
current_profit = trade.close_profit or 0.0 current_profit = trade.close_profit or 0.0
current_profit_abs = trade.close_profit_abs or 0.0 current_profit_abs = trade.close_profit_abs or 0.0
@@ -576,8 +580,8 @@ class RPC:
# Doing the sum is not right - overall profit needs to be based on initial capital # Doing the sum is not right - overall profit needs to be based on initial capital
profit_all_ratio_sum = sum(profit_all_ratio) if profit_all_ratio else 0.0 profit_all_ratio_sum = sum(profit_all_ratio) if profit_all_ratio else 0.0
starting_balance = self._freqtrade.wallets.get_starting_balance() starting_balance = self._freqtrade.wallets.get_starting_balance()
profit_closed_ratio_fromstart = 0 profit_closed_ratio_fromstart = 0.0
profit_all_ratio_fromstart = 0 profit_all_ratio_fromstart = 0.0
if starting_balance: if starting_balance:
profit_closed_ratio_fromstart = profit_closed_coin_sum / starting_balance profit_closed_ratio_fromstart = profit_closed_coin_sum / starting_balance
profit_all_ratio_fromstart = profit_all_coin_sum / starting_balance profit_all_ratio_fromstart = profit_all_coin_sum / starting_balance
@@ -890,10 +894,10 @@ class RPC:
if amount and amount < trade.amount: if amount and amount < trade.amount:
# Partial exit ... # Partial exit ...
min_exit_stake = self._freqtrade.exchange.get_min_pair_stake_amount( min_exit_stake = self._freqtrade.exchange.get_min_pair_stake_amount(
trade.pair, current_rate, trade.stop_loss_pct trade.pair, current_rate, trade.stop_loss_pct or 0.0
) )
remaining = (trade.amount - amount) * current_rate remaining = (trade.amount - amount) * current_rate
if remaining < min_exit_stake: if min_exit_stake and remaining < min_exit_stake:
raise RPCException(f"Remaining amount of {remaining} would be too small.") raise RPCException(f"Remaining amount of {remaining} would be too small.")
sub_amount = amount sub_amount = amount
@@ -1233,7 +1237,7 @@ class RPC:
for pair in add: for pair in add:
if pair not in self._freqtrade.pairlists.blacklist: if pair not in self._freqtrade.pairlists.blacklist:
try: try:
expand_pairlist([pair], self._freqtrade.exchange.get_markets().keys()) expand_pairlist([pair], list(self._freqtrade.exchange.get_markets().keys()))
self._freqtrade.pairlists.blacklist.append(pair) self._freqtrade.pairlists.blacklist.append(pair)
except ValueError: except ValueError: