Hopiu/wagtail-modeltranslation
1
0
Fork 0
mirror of https://github.com/Hopiu/wagtail-modeltranslation.git synced 2026-03-16 22:10:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove .specific calls from set_translation_url_paths command

Browse source
This commit is contained in:
Dario Marcelino 2018-01-03 12:04:16 +00:00
parent 066d00b20b
commit 9e2c2358ae
2 changed files with 19 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
wagtail_modeltranslation/management/commands/set_translation_url_paths.py
View file

@ -1,36 +1,28 @@
# coding: utf-8
from django.conf import settings
from django.core.management.base import BaseCommand
from modeltranslation import settings as mt_settings
from modeltranslation.utils import build_localized_fieldname
from wagtail.wagtailcore.models import Page
from wagtail_modeltranslation.contextlib import use_language
class Command(BaseCommand):
def set_subtree(self, root, root_path, lang=None):
update_fields = ['url_path_' + lang] if hasattr(root.specific, 'url_path_' + lang) else ['url_path']
def __init__(self):
super(Command, self).__init__()
update_fields = ['url_path']
for language in mt_settings.AVAILABLE_LANGUAGES:
localized_url_path = build_localized_fieldname('url_path', language)
update_fields.append(localized_url_path)
self.update_fields = update_fields
if hasattr(root.specific, 'url_path_' + lang):
setattr(root.specific, 'url_path_' + lang, root_path)
else:
setattr(root, 'url_path', root_path)
if lang == settings.LANGUAGE_CODE:
setattr(root, 'url_path', root_path)
update_fields.append('url_path')
root.specific.save(update_fields=update_fields)
def set_subtree(self, root, parent):
root.set_url_path(parent)
root.save(update_fields=self.update_fields)
for child in root.get_children():
slug = getattr(
child.specific, 'slug_' + lang) if hasattr(
child.specific, 'slug_' + lang) else getattr(child, 'slug')
if not slug or slug == '':
slug = getattr(
child.specific, 'slug_' + settings.LANGUAGE_CODE) if hasattr(child.specific,
'slug_' + settings.LANGUAGE_CODE) and getattr(
child.specific, 'slug_' + settings.LANGUAGE_CODE) else getattr(child, 'slug')
self.set_subtree(child, root_path + slug + '/', lang)
self.set_subtree(child, root)
def handle(self, **options):
for node in Page.get_root_nodes():
for lang in settings.LANGUAGES:
self.set_subtree(node, '/', lang=lang[0])
with use_language(mt_settings.DEFAULT_LANGUAGE):
for node in Page.get_root_nodes():
self.set_subtree(node, None)

3
wagtail_modeltranslation/tests/tests.py
View file

@ -881,7 +881,7 @@ class WagtailModeltranslationTest(WagtailModeltranslationTestBase):
.rewrite(False).update(url_path='corrupted', url_path_de='corrupted')
grandchild2 = models.TestSlugPage2.objects.get(slug_de='grandchild2-untranslated')
self.assertEqual(grandchild2.url_path, 'corrupted')
self.assertEqual(grandchild2.__dict__['url_path'], 'corrupted')
call_command('set_translation_url_paths', verbosity=0)
@ -894,6 +894,7 @@ class WagtailModeltranslationTest(WagtailModeltranslationTestBase):
self.assertEqual(grandgrandchild.url_path_en,
'/child-untranslated/grandchild1-untranslated/grandgrandchild-untranslated/')
grandchild2 = models.TestSlugPage2.objects.get(slug_de='grandchild2-untranslated')
self.assertEqual(grandchild2.__dict__['url_path'], '/child-untranslated/grandchild2-untranslated/')
self.assertEqual(grandchild2.url_path_de, '/child-untranslated/grandchild2-untranslated/')
self.assertEqual(grandchild2.url_path_en, '/child-untranslated/grandchild2-untranslated/')