From 241bfc409f20f43daed69b83222a163453cc383a Mon Sep 17 00:00:00 2001
From: Sam Germain <sam.germain@usask.ca>
Date: Thu, 5 Aug 2021 23:23:02 -0600
Subject: [PATCH] Added leverage enums

---
 freqtrade/enums/__init__.py     |  3 +++
 freqtrade/enums/collateral.py   | 11 +++++++++++
 freqtrade/enums/exchangename.py | 10 ++++++++++
 freqtrade/enums/signaltype.py   |  2 ++
 freqtrade/enums/tradingmode.py  | 11 +++++++++++
 5 files changed, 37 insertions(+)
 create mode 100644 freqtrade/enums/collateral.py
 create mode 100644 freqtrade/enums/exchangename.py
 create mode 100644 freqtrade/enums/tradingmode.py

diff --git a/freqtrade/enums/__init__.py b/freqtrade/enums/__init__.py
index 6099f7003..c60baad2a 100644
--- a/freqtrade/enums/__init__.py
+++ b/freqtrade/enums/__init__.py
@@ -1,8 +1,11 @@
 # flake8: noqa: F401
 from freqtrade.enums.backteststate import BacktestState
+from freqtrade.enums.collateral import Collateral
+from freqtrade.enums.exchangename import ExchangeName
 from freqtrade.enums.interestmode import InterestMode
 from freqtrade.enums.rpcmessagetype import RPCMessageType
 from freqtrade.enums.runmode import NON_UTIL_MODES, OPTIMIZE_MODES, TRADING_MODES, RunMode
 from freqtrade.enums.selltype import SellType
 from freqtrade.enums.signaltype import SignalTagType, SignalType
 from freqtrade.enums.state import State
+from freqtrade.enums.tradingmode import TradingMode
diff --git a/freqtrade/enums/collateral.py b/freqtrade/enums/collateral.py
new file mode 100644
index 000000000..0a5988698
--- /dev/null
+++ b/freqtrade/enums/collateral.py
@@ -0,0 +1,11 @@
+from enum import Enum
+
+
+class Collateral(Enum):
+    """
+        Enum to distinguish between
+        cross margin/futures collateral and
+        isolated margin/futures collateral
+    """
+    CROSS = "cross"
+    ISOLATED = "isolated"
diff --git a/freqtrade/enums/exchangename.py b/freqtrade/enums/exchangename.py
new file mode 100644
index 000000000..288754305
--- /dev/null
+++ b/freqtrade/enums/exchangename.py
@@ -0,0 +1,10 @@
+from enum import Enum
+
+
+class ExchangeName(Enum):
+    """All the exchanges supported by freqtrade that support leverage"""
+
+    BINANCE = "Binance"
+    KRAKEN = "Kraken"
+    FTX = "FTX"
+    OTHER = None
diff --git a/freqtrade/enums/signaltype.py b/freqtrade/enums/signaltype.py
index d2995d57a..09426b0e8 100644
--- a/freqtrade/enums/signaltype.py
+++ b/freqtrade/enums/signaltype.py
@@ -7,6 +7,8 @@ class SignalType(Enum):
     """
     BUY = "buy"
     SELL = "sell"
+    SHORT = "short"
+    EXIT_SHORT = "exit_short"
 
 
 class SignalTagType(Enum):
diff --git a/freqtrade/enums/tradingmode.py b/freqtrade/enums/tradingmode.py
new file mode 100644
index 000000000..a8de60c19
--- /dev/null
+++ b/freqtrade/enums/tradingmode.py
@@ -0,0 +1,11 @@
+from enum import Enum
+
+
+class TradingMode(Enum):
+    """
+        Enum to distinguish between
+        spot, margin, futures or any other trading method
+    """
+    SPOT = "spot"
+    MARGIN = "margin"
+    FUTURES = "futures"