Вопросы Теги

Как мне представить поле 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.

15
задан Dagg Nabbit 12 August 2010 в 13:16
поделиться

3 ответа

Отрабатывая фрагмент в ответе Эндрю, вот изменения, которые вам необходимо внести: 

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)
5
ответ дан 1 December 2019 в 04:33
поделиться

Я думаю, что лучшим решением здесь будет создание нового типа поля путем подклассификации 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 есть отличная подробная статья о том, как перейти к созданию подкласса models.Field:

Writing Custom Model Fields
Кажется, она охватывает все, что вам нужно сделать, включая:

  • Specifying Form Field (привязка формы к полю, чтобы django-admin знал, как его отображать)
  • Preparing Values for Lookups (что позволит вам использовать поле для поиска и фильтрации)

Если вы ищете пример подклассифицированного поля, этот фрагмент может быть полезен. Его цель аналогична (множественный выбор в качестве поля модели), но способ хранения в базе данных отличается (используется текстовое поле CSV вместо битовых флагов).

5
ответ дан 1 December 2019 в 04:33
поделиться

Вот как я бы использовал флаги с моим классом 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]

Конечно, вместо того, чтобы печатать флаги, вы можете создать строку, разделенную запятыми, для отображения в панели администратора, или как хотите.

Для администратора просто используйте логическое значение для каждого из значений группы.

1
ответ дан 1 December 2019 в 04:33
поделиться
Другие вопросы по тегам:

Похожие вопросы: