0

Django 1.3.3 を使用する ReviewBoard 1.6.11 での作業。次のように定義された「アクセス可能」と呼ばれるメソッドを持つ RepositoryManager クラスがあります。

class RepositoryManager(Manager):
    def accessible(self, user, visible_only=True, local_site=None):
        """Returns repositories that are accessible by the given user."""
        if user.is_superuser:
            qs = self.all()
        else:
            q = Q(public=True)

            if visible_only:
                q = q & Q(visible=True)

            if user.is_authenticated():
                q = q | (Q(users__pk=user.pk) |
                         Q(review_groups__users=user.pk))

            qs = self.filter(q).distinct()

        return qs.filter(local_site=local_site)

問題は、このクエリの結果を、データベースと対話しない別のもの (ユーザーのファイルシステムのアクセス許可) でフィルター処理したいことです。ただし、accessible() は QuerySet を返す必要があります。それ以外の場合は、リストを作成して、結果から適切な項目を入力するだけです。

私はDjangoにかなり慣れていません。これは合理的なことですか、それとも私はこれについてすべて間違っていますか?

4

1 に答える 1

0

次のように QuerySet サブクラスを作成することで、これを解決できました。

class RepositoryQuerySet(QuerySet):
    def __init__(self, *args, **kwargs):
        super(RepositoryQuerySet, self).__init__(*args, **kwargs)
        self._user_filter = None

    def _clone(self, *args, **kwargs):
        retval = super(RepositoryQuerySet, self)._clone(*args, **kwargs)
        retval._user_filter = self._user_filter
        return retval

    def filter_by_user_access(self, user):
        self._user_filter = user.username

    def iterator(self):
        for repo in super(RepositoryQuerySet, self).iterator():
            if self._user_filter is None:
                yield repo
                continue
            # ...logic for restricting access by user...
            # If user has access, just yield repo.  If not, just 'continue'

次に、RepositoryManager.accessible で、distinct() の呼び出しの直後に、次のように呼び出します。

            qs.filter_by_user_access(user)

最後に、マネージャーが新しいクエリ セット クラスを使用できるように、以下を作成します。

    def get_query_set(self):
        return RepositoryQuerySet(self.model, using=self.db)

RepositoryManager クラスで。

于 2012-09-05T20:24:29.223 に答える