153 lines
6.1 KiB
Python
153 lines
6.1 KiB
Python
from django.utils.datastructures import SortedSet
|
|
from django.db.migrations.state import ProjectState
|
|
|
|
|
|
class MigrationGraph(object):
|
|
"""
|
|
Represents the digraph of all migrations in a project.
|
|
|
|
Each migration is a node, and each dependency is an edge. There are
|
|
no implicit dependencies between numbered migrations - the numbering is
|
|
merely a convention to aid file listing. Every new numbered migration
|
|
has a declared dependency to the previous number, meaning that VCS
|
|
branch merges can be detected and resolved.
|
|
|
|
Migrations files can be marked as replacing another set of migrations -
|
|
this is to support the "squash" feature. The graph handler isn't resposible
|
|
for these; instead, the code to load them in here should examine the
|
|
migration files and if the replaced migrations are all either unapplied
|
|
or not present, it should ignore the replaced ones, load in just the
|
|
replacing migration, and repoint any dependencies that pointed to the
|
|
replaced migrations to point to the replacing one.
|
|
|
|
A node should be a tuple: (app_path, migration_name). The tree special-cases
|
|
things within an app - namely, root nodes and leaf nodes ignore dependencies
|
|
to other apps.
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.nodes = {}
|
|
self.dependencies = {}
|
|
self.dependents = {}
|
|
|
|
def add_node(self, node, implementation):
|
|
self.nodes[node] = implementation
|
|
|
|
def add_dependency(self, child, parent):
|
|
if child not in self.nodes:
|
|
raise KeyError("Dependency references nonexistent child node %r" % (child,))
|
|
if parent not in self.nodes:
|
|
raise KeyError("Dependency references nonexistent parent node %r" % (parent,))
|
|
self.dependencies.setdefault(child, set()).add(parent)
|
|
self.dependents.setdefault(parent, set()).add(child)
|
|
|
|
def forwards_plan(self, node):
|
|
"""
|
|
Given a node, returns a list of which previous nodes (dependencies)
|
|
must be applied, ending with the node itself.
|
|
This is the list you would follow if applying the migrations to
|
|
a database.
|
|
"""
|
|
if node not in self.nodes:
|
|
raise ValueError("Node %r not a valid node" % (node, ))
|
|
return self.dfs(node, lambda x: self.dependencies.get(x, set()))
|
|
|
|
def backwards_plan(self, node):
|
|
"""
|
|
Given a node, returns a list of which dependent nodes (dependencies)
|
|
must be unapplied, ending with the node itself.
|
|
This is the list you would follow if removing the migrations from
|
|
a database.
|
|
"""
|
|
if node not in self.nodes:
|
|
raise ValueError("Node %r not a valid node" % (node, ))
|
|
return self.dfs(node, lambda x: self.dependents.get(x, set()))
|
|
|
|
def root_nodes(self):
|
|
"""
|
|
Returns all root nodes - that is, nodes with no dependencies inside
|
|
their app. These are the starting point for an app.
|
|
"""
|
|
roots = set()
|
|
for node in self.nodes:
|
|
if not any(key[0] == node[0] for key in self.dependencies.get(node, set())):
|
|
roots.add(node)
|
|
return roots
|
|
|
|
def leaf_nodes(self):
|
|
"""
|
|
Returns all leaf nodes - that is, nodes with no dependents in their app.
|
|
These are the "most current" version of an app's schema.
|
|
Having more than one per app is technically an error, but one that
|
|
gets handled further up, in the interactive command - it's usually the
|
|
result of a VCS merge and needs some user input.
|
|
"""
|
|
leaves = set()
|
|
for node in self.nodes:
|
|
if not any(key[0] == node[0] for key in self.dependents.get(node, set())):
|
|
leaves.add(node)
|
|
return leaves
|
|
|
|
def dfs(self, start, get_children):
|
|
"""
|
|
Dynamic programming based depth first search, for finding dependencies.
|
|
"""
|
|
cache = {}
|
|
def _dfs(start, get_children, path):
|
|
# If we already computed this, use that (dynamic programming)
|
|
if (start, get_children) in cache:
|
|
return cache[(start, get_children)]
|
|
# If we've traversed here before, that's a circular dep
|
|
if start in path:
|
|
raise CircularDependencyError(path[path.index(start):] + [start])
|
|
# Build our own results list, starting with us
|
|
results = []
|
|
results.append(start)
|
|
# We need to add to results all the migrations this one depends on
|
|
children = sorted(get_children(start))
|
|
path.append(start)
|
|
for n in children:
|
|
results = _dfs(n, get_children, path) + results
|
|
path.pop()
|
|
# Use SortedSet to ensure only one instance of each result
|
|
results = list(SortedSet(results))
|
|
# Populate DP cache
|
|
cache[(start, get_children)] = results
|
|
# Done!
|
|
return results
|
|
return _dfs(start, get_children, [])
|
|
|
|
def __str__(self):
|
|
return "Graph: %s nodes, %s edges" % (len(self.nodes), sum(len(x) for x in self.dependencies.values()))
|
|
|
|
def project_state(self, nodes=None, at_end=True):
|
|
"""
|
|
Given a migration node or nodes, returns a complete ProjectState for it.
|
|
If at_end is False, returns the state before the migration has run.
|
|
If nodes is not provided, returns the overall most current project state.
|
|
"""
|
|
if nodes is None:
|
|
nodes = list(self.leaf_nodes())
|
|
if len(nodes) == 0:
|
|
return ProjectState()
|
|
if not isinstance(nodes[0], tuple):
|
|
nodes = [nodes]
|
|
plan = []
|
|
for node in nodes:
|
|
for migration in self.forwards_plan(node):
|
|
if migration not in plan:
|
|
if not at_end and migration in nodes:
|
|
continue
|
|
plan.append(migration)
|
|
project_state = ProjectState()
|
|
for node in plan:
|
|
project_state = self.nodes[node].mutate_state(project_state)
|
|
return project_state
|
|
|
|
|
|
class CircularDependencyError(Exception):
|
|
"""
|
|
Raised when there's an impossible-to-resolve circular dependency.
|
|
"""
|
|
pass
|