From 736edf4e874eb1ca9643d8be382eeee042bb521e Mon Sep 17 00:00:00 2001 From: Talitha Date: Wed, 20 Jun 2018 10:14:56 -0300 Subject: [PATCH] Update more pythonic --- sapl/base/templatetags/common_tags.py | 25 +++++++++++-------------- 1 file changed, 11 insertions(+), 14 deletions(-) diff --git a/sapl/base/templatetags/common_tags.py b/sapl/base/templatetags/common_tags.py index bf0d408dd..73d71faa8 100644 --- a/sapl/base/templatetags/common_tags.py +++ b/sapl/base/templatetags/common_tags.py @@ -47,22 +47,19 @@ def split(value, arg): @register.filter def sort_by_keys(value, key): transformed = [] - key_descricao = '' id_props = [x.id for x in value] qs = Proposicao.objects.filter(pk__in=id_props) - - if key[0] == '-': - key_descricao = '-' - if key == '1' or key == '-1': - key_descricao += 'data_envio' - elif key == '2' or key == '-2': - key_descricao += 'tipo' - elif key == '3' or key == '-3': - key_descricao += 'descricao' - else: - key_descricao += 'autor' - - transformed = qs.order_by(key_descricao) + key_descricao = {'1': 'data_envio', + '-1': '-data_envio', + '2': 'tipo', + '-2': '-tipo', + '3': 'descricao', + '-3': '-descricao', + '4': 'autor', + '-4': '-autor' + } + + transformed = qs.order_by(key_descricao[key]) return transformed