From 39a658eac22a70726d4b7bfa8139ee423325731e Mon Sep 17 00:00:00 2001
From: Matthias <xmatthias@outlook.com>
Date: Thu, 16 Feb 2023 07:04:45 +0000
Subject: [PATCH] Update DeclarativeBase

---
 freqtrade/persistence/base.py        | 7 +++----
 freqtrade/persistence/models.py      | 6 +++---
 freqtrade/persistence/pairlock.py    | 4 ++--
 freqtrade/persistence/trade_model.py | 6 +++---
 4 files changed, 11 insertions(+), 12 deletions(-)

diff --git a/freqtrade/persistence/base.py b/freqtrade/persistence/base.py
index fb2d561e1..2fed715d7 100644
--- a/freqtrade/persistence/base.py
+++ b/freqtrade/persistence/base.py
@@ -1,7 +1,6 @@
 
-from typing import Any
-
-from sqlalchemy.orm import declarative_base
+from sqlalchemy.orm import DeclarativeBase
 
 
-_DECL_BASE: Any = declarative_base()
+class ModelBase(DeclarativeBase):
+    pass
diff --git a/freqtrade/persistence/models.py b/freqtrade/persistence/models.py
index 38dbf212f..9f90f8a74 100644
--- a/freqtrade/persistence/models.py
+++ b/freqtrade/persistence/models.py
@@ -10,7 +10,7 @@ from sqlalchemy.orm import scoped_session, sessionmaker
 from sqlalchemy.pool import StaticPool
 
 from freqtrade.exceptions import OperationalException
-from freqtrade.persistence.base import _DECL_BASE
+from freqtrade.persistence.base import ModelBase
 from freqtrade.persistence.migrations import check_migrate
 from freqtrade.persistence.pairlock import PairLock
 from freqtrade.persistence.trade_model import Order, Trade
@@ -60,5 +60,5 @@ def init_db(db_url: str) -> None:
     PairLock.query = Trade._session.query_property()
 
     previous_tables = inspect(engine).get_table_names()
-    _DECL_BASE.metadata.create_all(engine)
-    check_migrate(engine, decl_base=_DECL_BASE, previous_tables=previous_tables)
+    ModelBase.metadata.create_all(engine)
+    check_migrate(engine, decl_base=ModelBase, previous_tables=previous_tables)
diff --git a/freqtrade/persistence/pairlock.py b/freqtrade/persistence/pairlock.py
index 2c4cfa3b4..e1f659b0a 100644
--- a/freqtrade/persistence/pairlock.py
+++ b/freqtrade/persistence/pairlock.py
@@ -5,10 +5,10 @@ from sqlalchemy import Boolean, Column, DateTime, Integer, String, or_
 from sqlalchemy.orm import Query
 
 from freqtrade.constants import DATETIME_PRINT_FORMAT
-from freqtrade.persistence.base import _DECL_BASE
+from freqtrade.persistence.base import ModelBase
 
 
-class PairLock(_DECL_BASE):
+class PairLock(ModelBase):
     """
     Pair Locks database model.
     """
diff --git a/freqtrade/persistence/trade_model.py b/freqtrade/persistence/trade_model.py
index 9355d2a84..ebec36e81 100644
--- a/freqtrade/persistence/trade_model.py
+++ b/freqtrade/persistence/trade_model.py
@@ -17,14 +17,14 @@ from freqtrade.enums import ExitType, TradingMode
 from freqtrade.exceptions import DependencyException, OperationalException
 from freqtrade.exchange import amount_to_contract_precision, price_to_precision
 from freqtrade.leverage import interest
-from freqtrade.persistence.base import _DECL_BASE
+from freqtrade.persistence.base import ModelBase
 from freqtrade.util import FtPrecise
 
 
 logger = logging.getLogger(__name__)
 
 
-class Order(_DECL_BASE):
+class Order(ModelBase):
     """
     Order database model
     Keeps a record of all orders placed on the exchange
@@ -1161,7 +1161,7 @@ class LocalTrade():
                 logger.info(f"New stoploss: {trade.stop_loss}.")
 
 
-class Trade(_DECL_BASE, LocalTrade):
+class Trade(ModelBase, LocalTrade):
     """
     Trade database model.
     Also handles updating and querying trades