Правильный способ автоматической фильтрации запросов SQLAlchemy? - PullRequest
8 голосов
/ 28 мая 2009

Я только что рассмотрел довольно неприятную схему из приложения CRM с sqlalchemy. У всех таблиц есть удаленный столбец, и я хотел автоматически отфильтровать все эти объекты и отношения, помеченные как удаленные. Вот что я придумал:


class CustomizableQuery(Query):
    """An overridden sqlalchemy.orm.query.Query to filter entities

    Filters itself by BinaryExpressions
    found in :attr:`CONDITIONS`
    """

    CONDITIONS = []

    def __init__(self, mapper, session=None):
        super(CustomizableQuery, self).__init__(mapper, session)
        for cond in self.CONDITIONS:
            self._add_criterion(cond)

    def _add_criterion(self, criterion):
        criterion = self._adapt_clause(criterion, False, True)
        if self._criterion is not None:
            self._criterion = self._criterion & criterion
        else:
            self._criterion = criterion

И он используется так:

class UndeletedContactQuery(CustomizableQuery):
    CONDITIONS = [contacts.c.deleted != True]

    def by_email(self, email_address):
        return EmailInfo.query.by_module_and_address('Contacts', email_address).contact

    def by_username(self, uname):
        return self.filter_by(twod_username_c=uname).one()

class Contact(object):
    query = session.query_property(UndeletedContactQuery)

Contact.query.by_email('someone@some.com')

EmailInfo - это класс, который сопоставляется с таблицей соединений между электронными письмами и другими модулями, с которыми они связаны.

Вот пример маппера:

contacts_map = mapper(Contact, join(contacts, contacts_cstm), {
    '_emails': dynamic_loader(EmailInfo,
                              foreign_keys=[email_join.c.bean_id],
                              primaryjoin=contacts.c.id==email_join.c.bean_id,
                              query_class=EmailInfoQuery),
    })

class EmailInfoQuery(CustomizableQuery):

    CONDITIONS = [email_join.c.deleted != True]
    # More methods here

Это дает мне то, что я хочу, потому что я отфильтровал все удаленные контакты. Я также могу использовать это как аргумент query_class для dynamic_loader в моих мапперах - Однако ...

  1. Есть ли лучший способ сделать это, я на самом деле не очень доволен, чтобы выискивать внутренности такого сложного класса, как Query, как и я.
  2. Кто-нибудь решил это другим способом, которым они могут поделиться?

Ответы [ 2 ]

7 голосов
/ 28 мая 2009

Вы можете отобразить на выбор. Как это:

mapper(EmailInfo, select([email_join], email_join.c.deleted == False))
0 голосов
/ 29 мая 2009

Я хотел бы рассмотреть возможность создания представлений для этих таблиц, которые отфильтровывают удаленные элементы, и тогда вы сможете отобразить непосредственно на это представление вместо базовой таблицы, по крайней мере, для запроса операций. Однако я никогда не пробовал это сам!

...