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

Allow retrieval of project state at start of migrations

This commit is contained in:
Andrew Godwin 2013-05-30 17:56:26 +01:00
parent 05929ee89b
commit dfa7c5cade
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -120,11 +120,14 @@ class MigrationGraph(object):
def __str__(self): def __str__(self):
return "Graph: %s nodes, %s edges" % (len(self.nodes), sum(len(x) for x in self.dependencies.values())) return "Graph: %s nodes, %s edges" % (len(self.nodes), sum(len(x) for x in self.dependencies.values()))
def project_state(self, node): def project_state(self, node, at_end=True):
""" """
Given a migration node, returns a complete ProjectState for it. Given a migration node, returns a complete ProjectState for it.
If at_end is False, returns the state before the migration has run.
""" """
plan = self.forwards_plan(node) plan = self.forwards_plan(node)
if not at_end:
plan = plan[:-1]
project_state = ProjectState() project_state = ProjectState()
for node in plan: for node in plan:
project_state = self.nodes[node].mutate_state(project_state) project_state = self.nodes[node].mutate_state(project_state)