Properly type "side" parameter

This commit is contained in:
Matthias
2022-04-24 10:58:21 +02:00
parent b7cada1edd
commit 7c79d937e0
9 changed files with 30 additions and 17 deletions

View File

@@ -3,6 +3,7 @@ import logging
from datetime import datetime, timedelta
from typing import Optional
from freqtrade.constants import LongShort
from freqtrade.persistence import Trade
from freqtrade.plugins.protections import IProtection, ProtectionReturn
@@ -54,7 +55,7 @@ class CooldownPeriod(IProtection):
return None
def global_stop(self, date_now: datetime, side: str) -> Optional[ProtectionReturn]:
def global_stop(self, date_now: datetime, side: LongShort) -> Optional[ProtectionReturn]:
"""
Stops trading (position entering) for all pairs
This must evaluate to true for the whole period of the "cooldown period".
@@ -64,7 +65,8 @@ class CooldownPeriod(IProtection):
# Not implemented for cooldown period.
return None
def stop_per_pair(self, pair: str, date_now: datetime, side: str) -> Optional[ProtectionReturn]:
def stop_per_pair(
self, pair: str, date_now: datetime, side: LongShort) -> Optional[ProtectionReturn]:
"""
Stops trading (position entering) for this pair
This must evaluate to true for the whole period of the "cooldown period".