Add custom class for SQLAlchemy + enable custom Select

src/utils_flask_sqla/models.py

@@ -0,0 +1,15 @@
+from .sqlalchemy import CustomSelect
+from flask_sqlalchemy.model import Model
+
+
+class SelectModelMixin(Model):
+    __abstract__ = True
+
+    @classmethod
+    @property
+    def select(cls):
+        if hasattr(cls, "__select_class__"):
+            select_cls = cls.__select_class__
+        else:
+            select_cls = CustomSelect
+        return select_cls._create_future_select(cls)  # SQLA 2.0: _create_future_select → _create

src/utils_flask_sqla/sqlalchemy.py

@@ -0,0 +1,17 @@
+from flask_sqlalchemy import SQLAlchemy
+from sqlalchemy.util.langhelpers import public_factory
+from sqlalchemy.sql.expression import Select
+
+
+class CustomSelect(Select):
+    def where_if(self, condition_to_execute_where, whereclause):
+        if condition_to_execute_where:
+            return self.where(whereclause)
+        else:
+            return self
+
+
+class CustomSQLAlchemy(SQLAlchemy):
+    @staticmethod
+    def select(*entities):
+        return CustomSelect._create_future_select(*entities)