DjangoBB LoFi version

Полная версия: необязательный параметр...

Начало » Python для новичков » необязательный параметр...

whitehat

Март 27, 2012 21:46:30

Ниже кусок из довольно объёмного кода, который приходится дублировать из-за того, что я не понимаю, как можно сделать включение одного из параметров необязательным

if profile_type == 'D':
            q_table = UserDownloads.objects.filter(date__lt = end_day, username__in = userlist)
           ...
elif profile_type == 'A':
           ...
else:
            q_table = UserDownloads.objects.filter(date__lt = end_day)

В примере по одной строке, а их несколько десятков, очень напрягает править сначала для одного profile_type, потом для второго, потом для третьего… Есть ли способ как-то записать этот код одной строкой, что бы username__in появлялся только по profile_type == ‘D’?
Спасибо

kmike

Март 27, 2012 22:14:17

если это django, то можно так:

q_table = UserDownloads.objects.filter(date__lt = end_day)
if profile_type == 'D':
    q_table = q_table.filter(username__in = userlist)

Еще можно так (хоть django, хоть не django):

filters = dict(date__lt=end_day)
if profile_type == 'D':
    filters['username__in'] = userlist
q_table = UserDownloads.objects.filter(**filters)

agalen

Март 27, 2012 22:34:20

q_table = UserDownloads.objects.filter(date__lt = end_day, **( { "username__in": userlist } if profile_type == 'D' else {} ) )

whitehat

Март 28, 2012 00:27:12

Спасибо большое! :D