Skip to content

Commit cb17c95

Browse files
committed
Add base url to embed block if needed
1 parent c2a82a5 commit cb17c95

1 file changed

Lines changed: 9 additions & 2 deletions

File tree

grapple/types/streamfield.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
import wagtail.embeds.blocks
77
import wagtail.images.blocks
88
import wagtail.snippets.blocks
9+
from django.conf import settings
910
from graphene.types import Scalar
1011
from graphene_django.converter import convert_django_field
1112
from wagtail.core.fields import StreamField
@@ -292,6 +293,12 @@ def resolve_choices(self, info, **kwargs):
292293
return choices
293294

294295

296+
def get_media_url(url):
297+
if url[0] == "/":
298+
return settings.BASE_URL + url
299+
return url
300+
301+
295302
class EmbedBlock(graphene.ObjectType):
296303
value = graphene.String()
297304
url = graphene.String()
@@ -301,8 +308,8 @@ class Meta:
301308

302309
def resolve_url(self, info, **kwargs):
303310
if hasattr(self, "value"):
304-
return self.value.url
305-
return self.url
311+
return get_media_url(self.value.url)
312+
return get_media_url(self.url)
306313

307314

308315
class StaticBlock(graphene.ObjectType):

0 commit comments

Comments
 (0)