migrate remaining __fields__ to model_fields

This commit is contained in:
Benedikt Bartscher 2024-02-28 23:58:36 +01:00
parent ce06bf0114
commit 67c81489cf
No known key found for this signature in database
7 changed files with 11 additions and 11 deletions

View File

@ -188,7 +188,7 @@ def _compile_client_storage_recursive(
cookies = {} cookies = {}
local_storage = {} local_storage = {}
state_name = state.get_full_name() state_name = state.get_full_name()
for name, field in state.__fields__.items(): for name, field in state.model_fields.items():
if name in state.inherited_vars: if name in state.inherited_vars:
# only include vars defined in this state # only include vars defined in this state
continue continue

View File

@ -70,7 +70,7 @@ class ChakraProvider(ChakraComponent):
def _get_imports(self) -> imports.ImportDict: def _get_imports(self) -> imports.ImportDict:
_imports = super()._get_imports() _imports = super()._get_imports()
_imports.setdefault(self.__fields__["library"].default, []).append( _imports.setdefault(self.model_fields["library"].default, []).append(
imports.ImportVar(tag="extendTheme", is_default=False), imports.ImportVar(tag="extendTheme", is_default=False),
) )
_imports.setdefault("/utils/theme.js", []).append( _imports.setdefault("/utils/theme.js", []).append(

View File

@ -101,7 +101,7 @@ class RadixThemesComponent(Component):
""" """
component = super().create(*children, **props) component = super().create(*children, **props)
if component.library is None: if component.library is None:
component.library = RadixThemesComponent.__fields__["library"].default component.library = RadixThemesComponent.model_fields["library"].default
component.alias = "RadixThemes" + ( component.alias = "RadixThemes" + (
component.tag or component.__class__.__name__ component.tag or component.__class__.__name__
) )

View File

@ -61,12 +61,12 @@ class Model(Base, sqlmodel.SQLModel):
"""Drop the default primary key field if any primary key field is defined.""" """Drop the default primary key field if any primary key field is defined."""
non_default_primary_key_fields = [ non_default_primary_key_fields = [
field_name field_name
for field_name, field in cls.__fields__.items() for field_name, field in cls.model_fields.items()
if field_name != "id" if field_name != "id"
and getattr(field.field_info, "primary_key", None) is True and getattr(field.field_info, "primary_key", None) is True
] ]
if non_default_primary_key_fields: if non_default_primary_key_fields:
cls.__fields__.pop("id", None) cls.model_fields.pop("id", None)
super().__init_subclass__() super().__init_subclass__()
@ -95,9 +95,9 @@ class Model(Base, sqlmodel.SQLModel):
Returns: Returns:
The object as a dictionary. The object as a dictionary.
""" """
base_fields = {name: getattr(self, name) for name in self.__fields__} base_fields = {name: getattr(self, name) for name in self.model_fields}
relationships = {} relationships = {}
# SQLModel relationships do not appear in __fields__, but should be included if present. # SQLModel relationships do not appear in model_fields, but should be included if present.
for name in self.__sqlmodel_relationships__: for name in self.__sqlmodel_relationships__:
try: try:
relationships[name] = self._dict_recursive(getattr(self, name)) relationships[name] = self._dict_recursive(getattr(self, name))

View File

@ -760,7 +760,7 @@ class BaseState(Base, ABC, extra=pydantic.Extra.allow):
Raises: Raises:
NameError: if a variable of this name already exists NameError: if a variable of this name already exists
""" """
if name in cls.__fields__: if name in cls.model_fields:
raise NameError( raise NameError(
f"The variable '{name}' already exist. Use a different name" f"The variable '{name}' already exist. Use a different name"
) )

View File

@ -373,7 +373,7 @@ def _extract_class_props_as_ast_nodes(
# with the annotation in some cases. # with the annotation in some cases.
with contextlib.suppress(AttributeError, KeyError): with contextlib.suppress(AttributeError, KeyError):
# Try to get default from pydantic field definition. # Try to get default from pydantic field definition.
default = target_class.__fields__[name].default default = target_class.model_fields[name].default
if isinstance(default, Var): if isinstance(default, Var):
default = default._decode() # type: ignore default = default._decode() # type: ignore

View File

@ -45,7 +45,7 @@ def test_default_primary_key(model_default_primary):
Args: Args:
model_default_primary: Fixture. model_default_primary: Fixture.
""" """
assert "id" in model_default_primary.__class__.__fields__ assert "id" in model_default_primary.__class__.model_fields
def test_custom_primary_key(model_custom_primary): def test_custom_primary_key(model_custom_primary):
@ -54,7 +54,7 @@ def test_custom_primary_key(model_custom_primary):
Args: Args:
model_custom_primary: Fixture. model_custom_primary: Fixture.
""" """
assert "id" not in model_custom_primary.__class__.__fields__ assert "id" not in model_custom_primary.__class__.model_fields
@pytest.mark.filterwarnings( @pytest.mark.filterwarnings(