BotEntity params hints, extended BotEnum db support

model/bot_enum.py

@@ -1,6 +1,7 @@
 from aiogram.utils.i18n import I18n
 from pydantic_core.core_schema import str_schema
-from sqlalchemy.types import TypeDecorator, String
+from sqlalchemy.types import TypeDecorator
+from sqlmodel import AutoString
 from typing import Any, Self, overload
 
 
@@ -119,12 +120,14 @@ class EnumMember(object):
     def __str__(self):
         return self.value
 
-    def __eq__(self, other: Self | str) -> bool:
+    def __eq__(self, other: Self | str | Any | None) -> bool:
         if other is None:
             return False
         if isinstance(other, str):
             return self.value == other
-        return self.value == other.value
+        if isinstance(other, EnumMember):
+            return self.value == other.value and self._parent is other._parent
+        return other.__eq__(self.value)
 
     def __hash__(self):
         return hash(self.value)
@@ -151,24 +154,25 @@ class BotEnum(EnumMember, metaclass=BotEnumMetaclass):
 
 
 class EnumType(TypeDecorator):
-    impl = String(64)
+    impl = AutoString
     cache_ok = True
 
-    # class comparator_factory(TypeDecorator.Comparator):
-    #     def __eq__(self, other):
-    #         expr = type_coerce(self.expr, String)
-    #         return expr != other.value
-
     def __init__(self, enum_type: BotEnum):
         self._enum_type = enum_type
         super().__init__()
 
-    def process_bind_param(self, value, dialect):
+    def _process_param(self, value):
         if value and isinstance(value, EnumMember):
             return value.value
         return str(value)
 
+    def process_bind_param(self, value, dialect):
+        return self._process_param(value)
+
     def process_result_value(self, value, dialect):
         if value:
             return self._enum_type(value)
         return None
+
+    def process_literal_param(self, value, dialect):
+        return self._process_param(value)