Как я должен представлять бит int int в django admin?
У меня есть модель данных с битовым полем, что-то вроде этого:
alter table MemberFlags add column title varchar(50) not null default '';
alter table MemberFlags add column value integer( 3) not null default 0;
insert into MemberFlags (title, value) values
("Blacklisted", 1),
("Special Guest", 2),
("Attend Ad-hoc Sessions", 4),
("Attend VIP Sessions", 8),
("Access Facility A", 16),
("Access Facility B", 32)
И используется так:
alter table Membership add column title varchar(50) not null default '';
alter table Membership add column flags integer( 3) not null default 0;
insert into Membership (title, flags) values
("Guest Pass", 4+2 ),
("Silver Plan", 16+ 4 ),
("Gold Plan", 32+16+ 4+2 ),
("VIP Pass", 32+16+8+4+2 )
Мои вопросы:
A) Какой самый простой способ представить различные битфлаги как отдельные элементы на сайте администратора? Должен ли я переопределить шаблон или сделать что-то с помощью форм?
B) Как насчет списка поиска? Я мог бы создавать функции в модели для представления каждого бита, но как будет выполняться поиск и сортировка?
Я новичок в Django.
Ответы
Ответ 1
Отказываясь от фрагмента в ответе Эндрю, выполните следующие изменения:
from django.db import models
from django import forms
class BitFlagFormField(forms.MultipleChoiceField):
widget = forms.CheckboxSelectMultiple
def __init__(self, *args, **kwargs):
super(BitFlagFormField, self).__init__(*args, **kwargs)
class BitFlagField(models.Field):
__metaclass__ = models.SubfieldBase
def get_internal_type(self):
return "Integer"
def get_choices_default(self):
return self.get_choices(include_blank=False)
def _get_FIELD_display(self, field):
value = getattr(self, field.attname)
choicedict = dict(field.choices)
def formfield(self, **kwargs):
# do not call super, as that overrides default widget if it has choices
defaults = {'required': not self.blank, 'label': capfirst(self.verbose_name),
'help_text': self.help_text, 'choices':self.choices}
if self.has_default():
defaults['initial'] = self.get_default()
defaults.update(kwargs)
return BitFlagFormField(**defaults)
def get_db_prep_value(self, value):
if isinstance(value, int):
return value
elif isinstance(value, list):
return sum(value)
def to_python(self, value):
result = []
n = 1
while value > 0:
if (value % 2) > 0:
result.append(n)
n *= 2
value /= 2
return sorted(result)
def contribute_to_class(self, cls, name):
super(BitFlagField, self).contribute_to_class(cls, name)
if self.choices:
func = lambda self, fieldname = name, choicedict = dict(self.choices):" and ".join([choicedict.get(value,value) for value in getattr(self,fieldname)])
setattr(cls, 'get_%s_display' % self.name, func)
Ответ 2
Я думаю, что лучшим решением здесь будет создание нового типа поля путем подкласса models.Field
. Вы можете использовать параметр choices, чтобы назначить действительные флаги бит и их значения. Это помогло бы сохранить декларацию модели чистой и удобочитаемой, с конечным результатом в следующем порядке:
class BitFlagField(models.Field):
...
class MyModel(models.Model):
...
FLAG_CHOICES = (
(1, 'Blacklisted'),
(2, 'Special Guest'),
(4, 'Attend Ad-hoc Sessions'),
(8, 'Attend VIP Sessions'),
(16, 'Access Facility A'),
(32, 'Access Facility B'),
)
flags = BitFlagField(choices=FLAG_CHOICES)
...
Документация Django содержит обширную статью о том, как продвигать модели подкласса. Поле:
Запись пользовательских полей моделей
Кажется, он охватывает все, что вам нужно сделать, в том числе:
Если вы ищете пример поля подкласса, этот фрагмент может быть полезен. Его цель аналогична (множественный выбор как поле модели), но способ хранения их в базе данных отличается (он использует текстовое поле CSV вместо битовых флагов).
Ответ 3
Отличное протестированное решение, даже если оно не соответствует вашей модели сразу, будет использовать django-bitfield
Ответ 4
Вот как я буду использовать флаги с моим классом User:
FLAGS = {
1:"Blacklisted",
2:"SpecialGuest",
4:"AttendAd-hocSessions",
8:"AttendVIPSessions",
16:"AccessFacilityA",
32:"AccessFacilityB",
}
class User(object):
def __init__(self, name="John Doe", groups=0):
self.name = name
self.groups = groups
def memberof(self):
''' Display string representation of the groups. '''
for flag in sorted(FLAGS):
if (flag & self.groups) == flag:
print FLAGS[flag]
Конечно, вместо печати флагов вы можете создать строку, разделенную запятой, для отображения в представлении администратора или независимо от того, что вы хотите.
Для администратора просто используйте boolean для каждого из значений группы.