remove Overriding default primary key

Author: adhami3310

packages/reflex-base/src/reflex_base/utils/compat.py

@@ -2,10 +2,7 @@
 
 import sys
 from collections.abc import Mapping
-from typing import TYPE_CHECKING, Any
-
-if TYPE_CHECKING:
-    from pydantic.fields import FieldInfo
+from typing import Any
 
 
 async def windows_hot_reload_lifespan_hack():
@@ -51,19 +48,3 @@ def annotations_from_namespace(namespace: Mapping[str, Any]) -> dict[str, Any]:
         if annotate := get_annotate_from_class_namespace(namespace):
             return call_annotate_function(annotate, format=Format.FORWARDREF)
     return namespace.get("__annotations__", {})
-
-
-def sqlmodel_field_has_primary_key(field_info: "FieldInfo") -> bool:
-    """Determines if a field is a primary.
-
-    Args:
-        field_info: a rx.model field
-
-    Returns:
-        If field_info is a primary key (Bool)
-    """
-    if getattr(field_info, "primary_key", None) is True:
-        return True
-    if getattr(field_info, "sa_column", None) is None:
-        return False
-    return bool(getattr(field_info.sa_column, "primary_key", None))  # pyright: ignore[reportAttributeAccessIssue]

reflex/model.py

@@ -13,8 +13,6 @@
 from reflex_base.utils import console
 from reflex_base.utils.serializers import serializer
 
-from reflex.utils.compat import sqlmodel_field_has_primary_key
-
 if TYPE_CHECKING:
     from typing import TypeVar
 
@@ -369,26 +367,6 @@ class Model(sqlmodel.SQLModel):
             "extra": "allow",
         }
 
-        @classmethod
-        def __pydantic_init_subclass__(cls):
-            """Drop the default primary key field if any primary key field is defined."""
-            non_default_primary_key_fields = [
-                field_name
-                for field_name, field_info in cls.model_fields.items()
-                if field_name != "id" and sqlmodel_field_has_primary_key(field_info)
-            ]
-            if non_default_primary_key_fields:
-                cls.model_fields.pop("id", None)
-                console.deprecate(
-                    feature_name="Overriding default primary key",
-                    reason=(
-                        "Register sqlmodel.SQLModel classes with `@rx.ModelRegistry.register`"
-                    ),
-                    deprecation_version="0.8.15",
-                    removal_version="0.9.0",
-                )
-            super().__pydantic_init_subclass__()
-
         @staticmethod
         def create_all():
             """Create all the tables."""