Remove redunant code from RunPython operation

This commit is contained in:
Andrew Godwin 2014-02-12 17:21:25 +00:00
parent b333de0f5d
commit 80bbe2265d
1 changed files with 2 additions and 16 deletions

View File

@ -128,26 +128,12 @@ class RunPython(Operation):
# object, representing the versioned models as an app registry. # object, representing the versioned models as an app registry.
# We could try to override the global cache, but then people will still # We could try to override the global cache, but then people will still
# use direct imports, so we go with a documentation approach instead. # use direct imports, so we go with a documentation approach instead.
if callable(self.code): self.code(models=from_state.render(), schema_editor=schema_editor)
self.code(models=from_state.render(), schema_editor=schema_editor)
else:
context = {
"models": from_state.render(),
"schema_editor": schema_editor,
}
eval(self.code, context)
def database_backwards(self, app_label, schema_editor, from_state, to_state): def database_backwards(self, app_label, schema_editor, from_state, to_state):
if self.reverse_code is None: if self.reverse_code is None:
raise NotImplementedError("You cannot reverse this operation") raise NotImplementedError("You cannot reverse this operation")
elif callable(self.reverse_code): self.reverse_code(models=from_state.render(), schema_editor=schema_editor)
self.reverse_code(models=from_state.render(), schema_editor=schema_editor)
else:
context = {
"models": from_state.render(),
"schema_editor": schema_editor,
}
eval(self.reverse_code, context)
def describe(self): def describe(self):
return "Raw Python operation" return "Raw Python operation"