mirrors/wagtail
mirrors/wagtail
0
0
Fork 0
mirror of https://github.com/wagtail/wagtail.git synced 2024-12-01 11:41:20 +01:00
Code Issues Releases Wiki Activity

Make migrations in project_template home app reversible

Browse Source
This commit is contained in:
Benjamin Bach 2016-04-28 10:36:59 +02:00 committed by Matt Westcott
parent 7686290007
commit 1718926c99
1 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
wagtail/project_template/home/migrations/0002_create_homepage.py
View File

@ -12,10 +12,11 @@ def create_homepage(apps, schema_editor):
HomePage = apps.get_model('home.HomePage')
# Delete the default homepage
Page.objects.get(id=2).delete()
# If migration is run multiple times, it may have already been deleted
Page.objects.filter(id=2).delete()
# Create content type for homepage model
homepage_content_type, created = ContentType.objects.get_or_create(
homepage_content_type, __ = ContentType.objects.get_or_create(
model='homepage', app_label='home')
# Create a new homepage
@ -34,6 +35,19 @@ def create_homepage(apps, schema_editor):
hostname='localhost', root_page=homepage, is_default_site=True)
def remove_homepage(apps, schema_editor):
# Get models
ContentType = apps.get_model('contenttypes.ContentType')
HomePage = apps.get_model('home.HomePage')
# Delete the default homepage
# Page and Site objects CASCADE
HomePage.objects.filter(slug='home', depth=2).delete()
# Delete content type for homepage model
ContentType.objects.filter(model='homepage', app_label='home').delete()
class Migration(migrations.Migration):
dependencies = [
@ -41,5 +55,5 @@ class Migration(migrations.Migration):
]
operations = [
migrations.RunPython(create_homepage),
migrations.RunPython(create_homepage, remove_homepage),
]