Added types to setters

This commit is contained in:
Sam Germain 2021-06-26 21:34:58 -06:00
parent 691a042e29
commit c68a0f05d8

View File

@ -285,17 +285,17 @@ class LocalTrade():
return self.__is_short or False return self.__is_short or False
@is_short.setter @is_short.setter
def is_short(self, val): def is_short(self, val: bool):
self.__is_short = val self.__is_short = val
@leverage.setter @leverage.setter
def leverage(self, lev): def leverage(self, lev: float):
self.__leverage = lev self.__leverage = lev
self.__borrowed = self.amount * (lev-1) self.__borrowed = self.amount * (lev-1)
self.amount = self.amount * lev self.amount = self.amount * lev
@borrowed.setter @borrowed.setter
def borrowed(self, bor): def borrowed(self, bor: float):
self.__leverage = self.amount / (self.amount - self.borrowed) self.__leverage = self.amount / (self.amount - self.borrowed)
self.__borrowed = bor self.__borrowed = bor
# End of margin trading properties # End of margin trading properties