stable/freqtrade/enums/interestmode.py
2021-07-28 12:25:55 -06:00

31 lines
787 B
Python

from enum import Enum, auto
from decimal import Decimal
one = Decimal(1.0)
four = Decimal(4.0)
twenty_four = Decimal(24.0)
class FunctionProxy:
"""Allow to mask a function as an Object."""
def __init__(self, function):
self.function = function
def __call__(self, *args, **kwargs):
return self.function(*args, **kwargs)
class InterestMode(Enum):
"""Equations to calculate interest"""
# Interest_rate is per day, minimum time of 1 hour
HOURSPERDAY = FunctionProxy(
lambda borrowed, rate, hours: borrowed * rate * max(hours, one)/twenty_four
)
# Interest_rate is per 4 hours, minimum time of 4 hours
HOURSPER4 = FunctionProxy(
lambda borrowed, rate, hours: borrowed * rate * (1 + max(0, (hours-four)/four))
)