plain.models 0.50.0__py3-none-any.whl → 0.51.1__py3-none-any.whl

plain/models/backends/sqlite3/schema.py

@@ -4,7 +4,7 @@ import copy
 from decimal import Decimal
 from typing import TYPE_CHECKING, Any
 
-from plain.models import register_model
+from plain.models import Options
 from plain.models.backends.base.schema import BaseDatabaseSchemaEditor
 from plain.models.backends.ddl_references import Statement
 from plain.models.backends.utils import strip_quotes
@@ -17,6 +17,8 @@ if TYPE_CHECKING:
     from plain.models.base import Model
     from plain.models.constraints import BaseConstraint
     from plain.models.fields import Field
+    from plain.models.fields.related import ManyToManyField
+    from plain.models.fields.reverse_related import ManyToManyRel
 
 
 class DatabaseSchemaEditor(BaseDatabaseSchemaEditor):
@@ -139,7 +141,7 @@ class DatabaseSchemaEditor(BaseDatabaseSchemaEditor):
         if not self._field_should_be_altered(old_field, new_field):
             return
         old_field_name = old_field.name
-        table_name = model._meta.db_table
+        table_name = model.model_options.db_table
         _, old_column_name = old_field.get_attname_column()
         if (
             new_field.name != old_field_name
@@ -150,7 +152,7 @@ class DatabaseSchemaEditor(BaseDatabaseSchemaEditor):
         ):
             if self.connection.in_atomic_block:
                 raise NotSupportedError(
-                    f"Renaming the {model._meta.db_table!r}.{old_field_name!r} column while in a transaction is not "
+                    f"Renaming the {model.model_options.db_table!r}.{old_field_name!r} column while in a transaction is not "
                     "supported on SQLite < 3.26 because it would break referential "
                     "integrity. Try adding `atomic = False` to the Migration class."
                 )
@@ -217,13 +219,13 @@ class DatabaseSchemaEditor(BaseDatabaseSchemaEditor):
         # Work out the new fields dict / mapping
         body = {
             f.name: f.clone() if is_self_referential(f) else f
-            for f in model._meta.local_concrete_fields
+            for f in model._model_meta.local_concrete_fields
         }
         # Since mapping might mix column names and default values,
         # its values must be already quoted.
         mapping = {
             f.column: self.quote_name(f.column)
-            for f in model._meta.local_concrete_fields
+            for f in model._model_meta.local_concrete_fields
         }
         # If any of the new or altered fields is introducing a new PK,
         # remove the old one
@@ -270,13 +272,13 @@ class DatabaseSchemaEditor(BaseDatabaseSchemaEditor):
         # Work inside a new app registry
         models_registry = ModelsRegistry()
 
-        indexes = model._meta.indexes
+        indexes = model.model_options.indexes
         if delete_field:
             indexes = [
                 index for index in indexes if delete_field.name not in index.fields
             ]
 
-        constraints = list(model._meta.constraints)
+        constraints = list(model.model_options.constraints)
 
         # Provide isolated instances of the fields to the new model body so
         # that the existing model's internals aren't interfered with when
@@ -288,32 +290,33 @@ class DatabaseSchemaEditor(BaseDatabaseSchemaEditor):
         # table and solely exists for foreign key reference resolution purposes.
         # This wouldn't be required if the schema editor was operating on model
         # states instead of rendered models.
-        meta_contents = {
-            "package_label": model._meta.package_label,
-            "db_table": model._meta.db_table,
-            "indexes": indexes,
-            "constraints": constraints,
-            "models_registry": models_registry,
-        }
-        meta = type("Meta", (), meta_contents)
-        body_copy["Meta"] = meta
+        meta_options = Options(
+            package_label=model.model_options.package_label,
+            db_table=model.model_options.db_table,
+            indexes=indexes,
+            constraints=constraints,
+        )
+        body_copy["model_options"] = meta_options
         body_copy["__module__"] = model.__module__
-        register_model(type(model._meta.object_name, model.__bases__, body_copy))  # type: ignore[arg-type]
+        temp_model: type[Model] = type(  # type: ignore[assignment]
+            model.model_options.object_name, model.__bases__, body_copy
+        )
+        models_registry.register_model(model.model_options.package_label, temp_model)
 
         # Construct a model with a renamed table name.
         body_copy = copy.deepcopy(body)
-        meta_contents = {
-            "package_label": model._meta.package_label,
-            "db_table": f"new__{strip_quotes(model._meta.db_table)}",
-            "indexes": indexes,
-            "constraints": constraints,
-            "models_registry": models_registry,
-        }
-        meta = type("Meta", (), meta_contents)
-        body_copy["Meta"] = meta
+        meta_options = Options(
+            package_label=model.model_options.package_label,
+            db_table=f"new__{strip_quotes(model.model_options.db_table)}",
+            indexes=indexes,
+            constraints=constraints,
+        )
+        body_copy["model_options"] = meta_options
         body_copy["__module__"] = model.__module__
-        new_model = type(f"New{model._meta.object_name}", model.__bases__, body_copy)
-        register_model(new_model)  # type: ignore[arg-type]
+        new_model: type[Model] = type(  # type: ignore[assignment]
+            f"New{model.model_options.object_name}", model.__bases__, body_copy
+        )
+        models_registry.register_model(model.model_options.package_label, new_model)
 
         # Create a new table with the updated schema.
         self.create_model(new_model)
@@ -321,10 +324,10 @@ class DatabaseSchemaEditor(BaseDatabaseSchemaEditor):
         # Copy data from the old table into the new table
         self.execute(
             "INSERT INTO {} ({}) SELECT {} FROM {}".format(
-                self.quote_name(new_model._meta.db_table),  # type: ignore[attr-defined]
+                self.quote_name(new_model.model_options.db_table),
                 ", ".join(self.quote_name(x) for x in mapping),
                 ", ".join(mapping.values()),
-                self.quote_name(model._meta.db_table),
+                self.quote_name(model.model_options.db_table),
             )
         )
 
@@ -334,8 +337,8 @@ class DatabaseSchemaEditor(BaseDatabaseSchemaEditor):
         # Rename the new table to take way for the old
         self.alter_db_table(
             new_model,  # type: ignore[arg-type]
-            new_model._meta.db_table,  # type: ignore[attr-defined]
-            model._meta.db_table,
+            new_model.model_options.db_table,
+            model.model_options.db_table,
             disable_constraints=False,
         )
 
@@ -355,13 +358,13 @@ class DatabaseSchemaEditor(BaseDatabaseSchemaEditor):
             self.execute(
                 self.sql_delete_table
                 % {
-                    "table": self.quote_name(model._meta.db_table),
+                    "table": self.quote_name(model.model_options.db_table),
                 }
             )
             # Remove all deferred statements referencing the deleted table.
             for sql in list(self.deferred_sql):
                 if isinstance(sql, Statement) and sql.references_table(
-                    model._meta.db_table
+                    model.model_options.db_table
                 ):
                     self.deferred_sql.remove(sql)
 
@@ -434,7 +437,7 @@ class DatabaseSchemaEditor(BaseDatabaseSchemaEditor):
         ):
             return self.execute(
                 self._rename_field_sql(
-                    model._meta.db_table, old_field, new_field, new_type
+                    model.model_options.db_table, old_field, new_field, new_type
                 )
             )
         # Alter by remaking table
@@ -446,8 +449,8 @@ class DatabaseSchemaEditor(BaseDatabaseSchemaEditor):
             old_type != new_type or old_collation != new_collation
         ):
             related_models = set()
-            opts = new_field.model._meta
-            for remote_field in opts.related_objects:
+            meta = new_field.model._model_meta
+            for remote_field in meta.related_objects:
                 # Ignore self-relationship since the table was already rebuilt.
                 if remote_field.related_model == model:
                     continue
@@ -455,7 +458,7 @@ class DatabaseSchemaEditor(BaseDatabaseSchemaEditor):
                     if remote_field.field_name == new_field.name:
                         related_models.add(remote_field.related_model)
             if new_field.primary_key:
-                for many_to_many in opts.many_to_many:
+                for many_to_many in meta.many_to_many:
                     # Ignore self-relationship since the table was already rebuilt.
                     if many_to_many.related_model == model:
                         continue
@@ -463,26 +466,34 @@ class DatabaseSchemaEditor(BaseDatabaseSchemaEditor):
                 self._remake_table(related_model)
 
     def _alter_many_to_many(
-        self, model: type[Model], old_field: Field, new_field: Field, strict: bool
+        self,
+        model: type[Model],
+        old_field: ManyToManyField,
+        new_field: ManyToManyField,
+        strict: bool,
     ) -> None:
         """Alter M2Ms to repoint their to= endpoints."""
+        # Type narrow for ManyToManyField.remote_field
+        old_rel: ManyToManyRel = old_field.remote_field  # type: ignore[assignment]
+        new_rel: ManyToManyRel = new_field.remote_field  # type: ignore[assignment]
+
         if (
-            old_field.remote_field.through._meta.db_table  # type: ignore[attr-defined]
-            == new_field.remote_field.through._meta.db_table  # type: ignore[attr-defined]
+            old_rel.through.model_options.db_table
+            == new_rel.through.model_options.db_table
         ):
             # The field name didn't change, but some options did, so we have to
             # propagate this altering.
             self._remake_table(
-                old_field.remote_field.through,  # type: ignore[attr-defined]
+                old_rel.through,
                 alter_fields=[
                     (
                         # The field that points to the target model is needed,
                         # so that table can be remade with the new m2m field -
                         # this is m2m_reverse_field_name().
-                        old_field.remote_field.through._meta.get_field(  # type: ignore[attr-defined]
+                        old_rel.through._model_meta.get_field(
                             old_field.m2m_reverse_field_name()  # type: ignore[attr-defined]
                         ),
-                        new_field.remote_field.through._meta.get_field(  # type: ignore[attr-defined]
+                        new_rel.through._model_meta.get_field(
                             new_field.m2m_reverse_field_name()  # type: ignore[attr-defined]
                         ),
                     ),
@@ -490,10 +501,10 @@ class DatabaseSchemaEditor(BaseDatabaseSchemaEditor):
                         # The field that points to the model itself is needed,
                         # so that table can be remade with the new self field -
                         # this is m2m_field_name().
-                        old_field.remote_field.through._meta.get_field(  # type: ignore[attr-defined]
+                        old_rel.through._model_meta.get_field(
                             old_field.m2m_field_name()  # type: ignore[attr-defined]
                         ),
-                        new_field.remote_field.through._meta.get_field(  # type: ignore[attr-defined]
+                        new_rel.through._model_meta.get_field(
                             new_field.m2m_field_name()  # type: ignore[attr-defined]
                         ),
                     ),
@@ -502,11 +513,11 @@ class DatabaseSchemaEditor(BaseDatabaseSchemaEditor):
             return
 
         # Make a new through table
-        self.create_model(new_field.remote_field.through)  # type: ignore[attr-defined]
+        self.create_model(new_rel.through)
         # Copy the data across
         self.execute(
             "INSERT INTO {} ({}) SELECT {} FROM {}".format(
-                self.quote_name(new_field.remote_field.through._meta.db_table),  # type: ignore[attr-defined]
+                self.quote_name(new_rel.through.model_options.db_table),
                 ", ".join(
                     [
                         "id",
@@ -521,11 +532,11 @@ class DatabaseSchemaEditor(BaseDatabaseSchemaEditor):
                         old_field.m2m_reverse_name(),  # type: ignore[attr-defined]
                     ]
                 ),
-                self.quote_name(old_field.remote_field.through._meta.db_table),  # type: ignore[attr-defined]
+                self.quote_name(old_rel.through.model_options.db_table),
             )
         )
         # Delete the old through table
-        self.delete_model(old_field.remote_field.through)  # type: ignore[attr-defined]
+        self.delete_model(old_rel.through)
 
     def add_constraint(self, model: type[Model], constraint: BaseConstraint) -> None:
         if isinstance(constraint, UniqueConstraint) and (