Browse Source

Oddělení tagu {%mailurl%}

pull/22/head
Pavel "LEdoian" Turinsky 2 years ago
parent
commit
0956b0780a
  1. 7
      various/templatetags/mail.py
  2. 18
      various/tests.py

7
various/templatetags/mail.py

@ -4,7 +4,7 @@ from urllib.request import quote as urlencode
register = template.Library() register = template.Library()
@register.simple_tag @register.simple_tag
def maillink(text: str, subject=None, body=None, to=[], attrs=None): def mailurl(*, subject=None, body=None, to=[]):
"""TODO: Dokumentace""" """TODO: Dokumentace"""
if isinstance(to, str): if isinstance(to, str):
to = [to] to = [to]
@ -24,6 +24,11 @@ def maillink(text: str, subject=None, body=None, to=[], attrs=None):
url = parts[0] + '?' + str.join('&', parts[1:]) url = parts[0] + '?' + str.join('&', parts[1:])
else: else:
url = parts[0] url = parts[0]
return url
@register.simple_tag
def maillink(text, subject=None, body=None, to=[], attrs=None):
url = mailurl(subject=subject, body=body, to=to)
if not attrs: attrs = '' if not attrs: attrs = ''
mezera = ' '*bool(attrs) mezera = ' '*bool(attrs)
full_link = f'<a href="{url}"{mezera}{attrs}>{text}</a>' full_link = f'<a href="{url}"{mezera}{attrs}>{text}</a>'

18
various/tests.py

@ -1,6 +1,6 @@
from django.test import TestCase from django.test import TestCase
# TODO: Možná vyrobit separátní soubory v tests/… než mít všechny testy v jednom souboru? # TODO: Možná vyrobit separátní soubory v tests/… než mít všechny testy v jednom souboru?
from various.templatetags.mail import maillink from various.templatetags.mail import maillink, mailurl
class MailTagsTest(TestCase): class MailTagsTest(TestCase):
"""Testuje template tagy ohledně mailů.""" """Testuje template tagy ohledně mailů."""
@ -24,6 +24,16 @@ class MailTagsTest(TestCase):
self.assertRaises(ValueError, lambda: maillink('Nemám příjemce')) self.assertRaises(ValueError, lambda: maillink('Nemám příjemce'))
self.assertRaises(TypeError, lambda: maillink()) # Nemá text, takže to shodí python self.assertRaises(TypeError, lambda: maillink()) # Nemá text, takže to shodí python
def test_mailurl(self):
self.assertEquals(mailurl(to='some@body.test'), r'mailto:some@body.test')
self.assertEquals(mailurl(to=['some@body.test']), r'mailto:some@body.test')
self.assertEquals(mailurl(to=['alice@test.test', 'bob@jinde.test']), r'mailto:alice@test.test,bob@jinde.test')
self.assertEquals(
mailurl(to='some@body.test', body='Tělo', subject='Předmět'),
r'mailto:some@body.test?subject=P%C5%99edm%C4%9Bt&body=T%C4%9Blo',
)
self.assertRaises(ValueError, lambda: mailurl())
def test_render_in_template(self): def test_render_in_template(self):
# Pomocná funkce: vykreslí template do stringu # Pomocná funkce: vykreslí template do stringu
# Ref: https://stackoverflow.com/a/1690879 # Ref: https://stackoverflow.com/a/1690879
@ -42,3 +52,9 @@ class MailTagsTest(TestCase):
render_template(template), render_template(template),
r'<a href="mailto:alice@test.test?subject=Oprava%20%C5%99e%C5%A1en%C3%AD">Text</a>', r'<a href="mailto:alice@test.test?subject=Oprava%20%C5%99e%C5%A1en%C3%AD">Text</a>',
) )
mailurltemplate = (
r'{% load mail %}'
r'{% mailurl to="alice@test.test" subject="Čau Alice" %}'
)
self.assertEquals(render_template(mailurltemplate), r'mailto:alice@test.test?subject=%C4%8Cau%20Alice')

Loading…
Cancel
Save