Changeset 4057

Timestamp:

11/09/06 10:45:11 (2 years ago)

Author:

jkocherhans

Message:

[generic-auth] Added an extensible function for filtering querysets based on permission into the tests. This should be moved out into the actual codebase at some point. Note that it's far from complete.

Files:

• django/branches/generic-auth/tests/regressiontests/generic_auth/models.py (modified) (1 diff)
• django/branches/generic-auth/tests/regressiontests/generic_auth/tests.py (modified) (2 diffs)

django/branches/generic-auth/tests/regressiontests/generic_auth/models.py

@@ -6 +6 @@
-
-class Article(models.Model):
-nam
+titl
-    body = models.TextField()
-    creator = models.ForeignKey(User)

django/branches/generic-auth/tests/regressiontests/generic_auth/tests.py

@@ -106 +106 @@
-created above.
-
-nam
+titl
->>> article.save()
-
@@ -143 +143 @@
-
-
+
+QuerySet Filtering by Permissions
+---------------------------------
+
+Checking permissions on a single object only solves half of the problem. We
+also need a way to get a list of objects for which a user has a given permission. 
+
+This is implemented as a function for now, but it might be more desirable to
+be a method of QuerySet. We'll just to use plain old functions to start with, 
+then we'll test the extensible function version.
+
+>>> def owner_filter(user, permission, queryset):
+...     return queryset.filter(creator=user)
+... 
+
+>>> user_2 = User.objects.create_user('user 2', 'test@example.com', 'password')
+>>> user_2.save()
+
+>>> article_1 = Article(title='article 1', body='article 1', creator=user_2)
+>>> article_1.save()
+>>> article_2 = Article(title='article 2', body='article 2', creator=user)
+>>> article_2.save()
+>>> article_3 = Article(title='article 3', body='article 3', creator=user)
+>>> article_3.save()
+
+>>> qs = owner_filter(user, change_permission, Article.objects.all())
+>>> qs.count()
+3
+>>> [a.title for a in qs]
+['test', 'article 2', 'article 3']
+
+
+QuerySetFilter is a craptacular name... let's change that.
+
+>>> class QuerySetFilter(object):
+...     def __init__(self):
+...         self.registry = {}
+...
+...     def __call__(self, user, permission, queryset):
+...         types = (type(user), type(permission), queryset.model)
+...         #print "Looking up types: " + types.__repr__()
+...         func = self.registry.get(types)
+...         if func:
+...             return func(user, permission, queryset)
+...         else:
+...             return queryset
+...
+...     def register(self, func, user_type, permission_type, qs_model_type):
+...         types = (user_type, permission_type, qs_model_type)
+...         #print "Registering types: " + types.__repr__()
+...         self.registry[types] = func
+...
+
+Now try the same stuff using the extensible function.
+
+>>> qs_filter = QuerySetFilter()
+>>> qs_filter.register(owner_filter, User, Permission, Article)
+
+>>> qs = qs_filter(user, change_permission, Article.objects.all())
+>>> qs.count()
+3
+>>> [a.title for a in qs]
+['test', 'article 2', 'article 3']
+
-"""