53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
from __future__ import unicode_literals
|
||
|
|
|
||
|
|
from django.db import models, migrations
|
||
|
|
from wagtail.wagtailcore.rich_text import RichText
|
||
|
|
|
||
|
|
|
||
|
|
models = ["ContentPage", "NewsIndexPage", "NewsPage", ]
|
||
|
|
|
||
|
|
|
||
|
|
def convert_to_streamfield(apps, schema_editor):
|
||
|
|
for model_name in models:
|
||
|
|
_convert_to_streamfield(apps, model_name)
|
||
|
|
|
||
|
|
|
||
|
|
def _convert_to_streamfield(apps, model_name):
|
||
|
|
Model = apps.get_model("cms_pages", model_name)
|
||
|
|
for page in Model.objects.all():
|
||
|
|
if page.body.raw_text and not page.body:
|
||
|
|
page.body = [('rich_text', RichText(page.body.raw_text))]
|
||
|
|
page.save()
|
||
|
|
|
||
|
|
|
||
|
|
def convert_to_richtext(apps, schema_editor):
|
||
|
|
for model_name in models:
|
||
|
|
_convert_to_richtext(apps, model_name)
|
||
|
|
|
||
|
|
|
||
|
|
def _convert_to_richtext(apps, model_name):
|
||
|
|
Model = apps.get_model("cms_pages", model_name)
|
||
|
|
for page in Model.objects.all():
|
||
|
|
if page.body.raw_text is None:
|
||
|
|
raw_text = ''.join([
|
||
|
|
child.value.source for child in page.body
|
||
|
|
if child.block_type == 'rich_text'
|
||
|
|
])
|
||
|
|
page.body = raw_text
|
||
|
|
page.save()
|
||
|
|
|
||
|
|
|
||
|
|
class Migration(migrations.Migration):
|
||
|
|
|
||
|
|
dependencies = [
|
||
|
|
('cms_pages', '0007_auto_20160916_0417'),
|
||
|
|
]
|
||
|
|
|
||
|
|
operations = [
|
||
|
|
migrations.RunPython(
|
||
|
|
convert_to_streamfield,
|
||
|
|
convert_to_richtext,
|
||
|
|
),
|
||
|
|
]
|