Skip to content

Commit

Permalink
Made autodetector add genereted field after foreign keys and added re…
Browse files Browse the repository at this point in the history
…lated tests.
  • Loading branch information
DevilsAutumn committed Apr 16, 2024
1 parent 3cb2479 commit 5f25b42
Show file tree
Hide file tree
Showing 3 changed files with 113 additions and 4 deletions.
22 changes: 19 additions & 3 deletions django/db/migrations/autodetector.py
Expand Up @@ -1170,12 +1170,28 @@ def _generate_added_field(self, app_label, model_name, field_name):
for child in expr.children
if isinstance(child, tuple)
}
newly_added_fields = sorted(self.new_field_keys - self.old_field_keys)
for ref_field_name in f_references.union(q_references):
if not (app_label, model_name, ref_field_name) in sorted(
self.new_field_keys - self.old_field_keys
):
if (app_label, model_name, ref_field_name) not in newly_added_fields:
continue
self._generate_added_field(app_label, model_name, ref_field_name)

# Add dependencies on the fks so that the Generated field is added even
# after the fks.
for app_label, model_name, added_field_name in newly_added_fields:
added_field = self.to_state.models[app_label, model_name].get_field(
added_field_name
)
if added_field.remote_field and added_field.remote_field.model:
dependencies.append(
OperationDependency(
app_label,
model_name,
added_field.name,
OperationDependency.Type.CREATE,
)
)

self.add_operation(
app_label,
operations.AddField(
Expand Down
45 changes: 45 additions & 0 deletions tests/migrations/test_autodetector.py
Expand Up @@ -1401,6 +1401,51 @@ def test_add_field_before_generated_field(self):
changes, "testapp", 0, 1, expression=Concat(Lower("name"), Lower("surname"))
)

def test_add_fk_before_generated_field(self):
initial_state = ModelState(
"testapp",
"Author",
[
("name", models.CharField(max_length=20)),
],
)
updated_state = [
ModelState(
"testapp",
"Publisher",
[
("name", models.CharField(max_length=20)),
],
),
ModelState(
"testapp",
"Author",
[
("name", models.CharField(max_length=20)),
(
"publisher",
models.ForeignKey("testapp.Publisher", models.CASCADE),
),
(
"lower_full_name",
models.GeneratedField(
expression=Concat("name", "publisher_id"),
output_field=models.CharField(max_length=20),
db_persist=True,
),
),
],
),
]
changes = self.get_changes([initial_state], updated_state)
self.assertNumberMigrations(changes, "testapp", 1)
self.assertOperationTypes(
changes, "testapp", 0, ["CreateModel", "AddField", "AddField"]
)
self.assertOperationFieldAttributes(
changes, "testapp", 0, 2, expression=Concat("name", "publisher_id")
)

def test_remove_field(self):
"""Tests autodetection of removed fields."""
changes = self.get_changes([self.author_name], [self.author_empty])
Expand Down
50 changes: 49 additions & 1 deletion tests/migrations/test_operations.py
Expand Up @@ -9,7 +9,7 @@
from django.db.migrations.state import ModelState, ProjectState
from django.db.models import F
from django.db.models.expressions import Value
from django.db.models.functions import Abs, Pi
from django.db.models.functions import Abs, Concat, Pi
from django.db.transaction import atomic
from django.test import (
SimpleTestCase,
Expand Down Expand Up @@ -1379,6 +1379,54 @@ def test_add_field(self):
self.assertEqual(definition[1], [])
self.assertEqual(sorted(definition[2]), ["field", "model_name", "name"])

@skipUnlessDBFeature("supports_stored_generated_columns")
def _test_add_generate_field(self):
app_label = "test_adgf"
project_state = self.apply_operations(
app_label,
ProjectState(),
operations=[
migrations.CreateModel(
"Rider",
fields=[
("id", models.AutoField(primary_key=True)),
],
),
migrations.CreateModel(
"Pony",
fields=[
("id", models.AutoField(primary_key=True)),
("name", models.CharField(max_length=20)),
(
"rider",
models.ForeignKey(
f"{app_label}.Rider", on_delete=models.CASCADE
),
),
(
"name_and_id",
models.GeneratedField(
expression=Concat(("name"), ("rider_id")),
output_field=models.TextField(),
db_persist=True,
),
),
],
),
],
)
Pony = project_state.apps.get_model(app_label, "Pony")
Rider = project_state.apps.get_model(app_label, "Rider")
rider = Rider.objects.create()
pony = Pony.objects.create(name="pony", rider=rider)
self.assertEqual(pony.name_and_id, str(pony.name) + str(rider.id))

new_rider = Rider.objects.create()
pony.rider = new_rider
pony.save()
pony.refresh_from_db()
self.assertEqual(pony.name_and_id, str(pony.name) + str(new_rider.id))

def test_add_charfield(self):
"""
Tests the AddField operation on TextField.
Expand Down

0 comments on commit 5f25b42

Please sign in to comment.