Update migration to use the form "document.versions" instead of "document.version_set"

This commit is contained in:
Roberto Rosario
2014-05-19 00:28:03 -04:00
parent 66bb049881
commit 518137415a

View File

@@ -8,7 +8,7 @@ class Migration(DataMigration):
def forwards(self, orm):
for document in orm.Document.objects.all():
document_version = document.documentversion_set.create(
document_version = document.versions.create(
document = document,
timestamp = document.date_added,
file = document.file,
@@ -24,7 +24,7 @@ class Migration(DataMigration):
def backwards(self, orm):
for document in orm.Document.objects.all():
document_version = document.documentversion_set.all()[0]
document_version = document.versions.all()[0]
document.date_added = document_version.timestamp
document.file = document_version.file
document.file_mimetype = document_version.mimetype