from urllib import parse

from django import template

register = template.Library()


@register.simple_tag
def url_parameter(url, parametro, valor):
    queries = url

    queries = parse.urlsplit(url).query
    dict_params = dict(parse.parse_qsl(queries))
    dict_params.update({parametro: valor})
    queries_list = []
    print(dict_params)
    for (k, v) in dict_params.items():
        queries_list.append('%s=%s' % (k, v))
    base_url_list = url.split('?')
    base_url = base_url_list[0]

    if len(base_url_list) <= 1:
        base_url = ''

    queries_str = '&'.join(queries_list)

    return '%s?%s' % (base_url, queries_str)