Django custom simple tags
from django import template
from django.urls import reverse_lazy
register = template.Library()
@register.simple_tag(takes_context=True)
def get_url(context, *args):
'''
example 1 "{% get_url 'list' %}"
example 2 "{% get_url 'create' %}"
example 3 "{% get_url 'create' 'cms' %}"
example 4 "{% get_url 'detail' obj %}"
example 5 "{% get_url 'detail' obj 'cms' %}"
the first argument is action create or list or detail or update or delete
the second argument is a model object
the third argument is custom app
the name of url pattern so as to work
app:model-create
app:model-update
app:model-delete
app:model-detail
'''
model = context['model']
obj = None
print(len(args))
lower_name = model.__name__.lower()
app = model._meta.app_label
if len(args) == 0:
return ''
if len(args) == 2:
action = args[0]
app = args[1]
if len(args) ==3:
action = args[0]
obj = args[1]
app = args[2]
lower_name = obj.__class__.__name__.lower()
url_string = '{}:{}-{}'.format(app, lower_name, action)
if hasattr(obj, 'uuid'):
url = reverse_lazy(url_string, kwargs={'uuid': obj.uuid})
if(hasattr(obj, 'slug')):
url = reverse_lazy(url_string, kwargs={'slug': obj.slug})
if hasattr(obj, 'pk'):
url = reverse_lazy(url_string, kwargs={'pk': obj.pk})
if not obj:
url_string = '{}:{}-{}'.format(app, lower_name, action)
url = reverse_lazy(url_string)
return url
