Skip to content

Commit

Permalink
Update snippet chooser URL routing for Wagtail 4
Browse files Browse the repository at this point in the history
  • Loading branch information
gasman authored and zerolab committed Jul 19, 2022
1 parent 6120133 commit 78dfb10
Showing 1 changed file with 35 additions and 14 deletions.
49 changes: 35 additions & 14 deletions wagtail_localize/views/edit_translation.py
Original file line number Diff line number Diff line change
Expand Up @@ -381,6 +381,23 @@ def widget_from_field(field):
return {"type": "image_chooser"}

elif issubclass(field.related_model, tuple(get_snippet_models())):
if WAGTAIL_VERSION >= (4, 0, 0):
chooser_url = reverse(
"wagtailsnippetchoosers_%s_%s:choose"
% (
field.related_model._meta.app_label,
field.related_model._meta.model_name,
)
)
else:
chooser_url = reverse(
"wagtailsnippets:choose",
args=[
field.related_model._meta.app_label,
field.related_model._meta.model_name,
],
)

return {
"type": "snippet_chooser",
"snippet_model": {
Expand All @@ -389,13 +406,7 @@ def widget_from_field(field):
"verbose_name": field.related_model._meta.verbose_name,
"verbose_name_plural": field.related_model._meta.verbose_name_plural,
},
"chooser_url": reverse(
"wagtailsnippets:choose",
args=[
field.related_model._meta.app_label,
field.related_model._meta.model_name,
],
),
"chooser_url": chooser_url,
}

elif isinstance(
Expand Down Expand Up @@ -428,6 +439,22 @@ def widget_from_block(block, content_components=None):
return {"type": "image_chooser"}

elif isinstance(block, SnippetChooserBlock):
if WAGTAIL_VERSION >= (4, 0, 0):
chooser_url = reverse(
"wagtailsnippetchoosers_%s_%s:choose"
% (
block.target_model._meta.app_label,
block.target_model._meta.model_name,
)
)
else:
chooser_url = reverse(
"wagtailsnippets:choose",
args=[
block.target_model._meta.app_label,
block.target_model._meta.model_name,
],
)
return {
"type": "snippet_chooser",
"snippet_model": {
Expand All @@ -436,13 +463,7 @@ def widget_from_block(block, content_components=None):
"verbose_name": block.target_model._meta.verbose_name,
"verbose_name_plural": block.target_model._meta.verbose_name_plural,
},
"chooser_url": reverse(
"wagtailsnippets:choose",
args=[
block.target_model._meta.app_label,
block.target_model._meta.model_name,
],
),
"chooser_url": chooser_url,
}

elif isinstance(
Expand Down

0 comments on commit 78dfb10

Please sign in to comment.