djangoldp/tests/permissions.py

+from djangoldp.filters import BaseFilterBackend
+from djangoldp.permissions import LDPBasePermission
+
+class StartsWithAFilter(BaseFilterBackend):
+    """Only objects whose title starts in A get through"""
+    def filter_queryset(self, request, queryset, view):
+        return queryset.filter(title__startswith='A')
+
+class ReadOnlyStartsWithA(LDPBasePermission):
+    """Only gives read-only access and only to objects which title starts with A"""
+    filter_backend = StartsWithAFilter
+    permissions = {'view', 'list'}
+    def check_perms(self, obj):
+        return getattr(obj, 'title', '').startswith('A')
+    def has_object_permission(self, request, view, obj=None):
+        return self.check_perms(obj)
+    def get_permissions(self, user, model, obj=None):
+        return self.permissions if self.check_perms(obj) else set()
+
+
+class ContainsSpace(BaseFilterBackend):
+    """Only objects whose title contains a space get through"""
+    def filter_queryset(self, request, queryset, view):
+        if request.user.username != 'toto':
+            return queryset.none()
+        return queryset.filter(title__contains=' ')
+
+class Only2WordsForToto(LDPBasePermission):
+    """Only gives access if the user's username is toto and only to objects whose title has two words (contains space)"""
+    filter_backend = ContainsSpace
+    def has_permission(self, request, view):
+        return request.user.username == 'toto'
+    def check_perms(self, obj):
+        return ' ' in getattr(obj, 'title', '')
+    def has_object_permission(self, request, view, obj=None):
+        return self.check_perms(obj)
+    def get_permissions(self, user, model, obj=None):
+        return self.permissions if self.check_perms(obj) else set()
\ No newline at end of file

djangoldp/tests/scripts/prod_data_generator.py

+import json
+import argparse
+from pathlib import Path
+from datetime import datetime
+from utils import generate_users, generate_projects, generate_skills
+
+'''
+A script which generates and outputs random production data, into a parameterised file (json), which can be used as
+a Django fixture or imported into a live database
+e.g. python manage.py loaddata fixture.json
+for help run python prod_data_generator.py -h
+'''
+
+# starting from offset ensures that existing users etc are not disturbed
+parser = argparse.ArgumentParser(description='generates and outputs random test data, into a file used by the performance unit tests')
+parser.add_argument(dest='count', metavar='N', type=int, help='the number of users (and projects) to generate')
+parser.add_argument('--offset', dest='offset', type=int, default=100, help='an offset to start primary keys at (should be larger than the largest pre-existing project/user primary key)')
+parser.add_argument('-f', dest='file_dest', type=str, default="../fixtures/live.json", help='the file destination to write to')
+parser.add_argument('-s', dest='generate_skills', type=bool, default=False, help='Do you want to generate skills too ?')
+
+args = parser.parse_args()
+count = args.count
+OFFSET = args.offset
+
+user_template = {
+    'model': 'djangoldp_account.ldpuser',
+    'pk': 0,
+    'fields': {
+        'username': 'john',
+        'email': 'jlennon@c.coop',
+        'password':'glassonion',
+        'first_name': 'John',
+        'last_name': 'Lennon'
+    }
+}
+
+project_template = {
+    'model': 'djangoldp_project.project',
+    'pk': 0,
+    'fields': {
+        'description': 'Test',
+        'status': 'Public',
+        'creationDate': str(datetime.date(datetime.now()))
+    }
+}
+
+skill_template = {
+    'model': 'djangoldp_skill.skill',
+    'pk': 0,
+    'fields': {
+        'name': 'PHP',
+    }
+}
+
+fixture = generate_users(count, user_template, offset=OFFSET)
+fixture = generate_projects(count, project_template, fixture=fixture, offset=OFFSET)
+
+if args.generate_skills:
+    fixture = generate_skills(count, skill_template, fixture=fixture, offset=OFFSET)
+
+with open(Path(__file__).parent / args.file_dest, 'w') as output:
+    json.dump(fixture, output)
+
+print(str(count))