p31829507
/
django
mirror of https://github.com/django/django.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Refs #29180 -- Added MigrationGraph._generate_plan() for testing.

This commit is contained in:
Carlton Gibson 2018-03-09 08:50:42 +01:00
parent 2d9ec4d735
commit 4d420a53cf
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
django/db/migrations/graph.py
View File

@ -352,6 +352,14 @@ class MigrationGraph:
def _nodes_and_edges(self): def _nodes_and_edges(self):
return len(self.nodes), sum(len(node.parents) for node in self.node_map.values()) return len(self.nodes), sum(len(node.parents) for node in self.node_map.values())
def _generate_plan(self, nodes, at_end):
plan = []
for node in nodes:
for migration in self.forwards_plan(node):
if migration in plan or at_end or migration not in nodes:
plan.append(migration)
return plan
def make_state(self, nodes=None, at_end=True, real_apps=None): def make_state(self, nodes=None, at_end=True, real_apps=None):
""" """
Given a migration node or nodes, return a complete ProjectState for it. Given a migration node or nodes, return a complete ProjectState for it.
@ -364,11 +372,7 @@ class MigrationGraph:
return ProjectState() return ProjectState()
if not isinstance(nodes[0], tuple): if not isinstance(nodes[0], tuple):
nodes = [nodes] nodes = [nodes]
plan = [] plan = self._generate_plan(nodes, at_end)
for node in nodes:
for migration in self.forwards_plan(node):
if migration in plan or at_end or migration not in nodes:
plan.append(migration)
project_state = ProjectState(real_apps=real_apps) project_state = ProjectState(real_apps=real_apps)
for node in plan: for node in plan:
project_state = self.nodes[node].mutate_state(project_state, preserve=False) project_state = self.nodes[node].mutate_state(project_state, preserve=False)