keningle · July 2, 2013 19:43 · Jul 2, 2013
diff --git a/decorator.py b/decorator.py
@@ -0,0 +1,17 @@
+from django.utils.functional import wraps
+...
+
+def check_company_admin(view):
+    @wraps(view)
+    def inner(request, slug, *args, **kwargs):
+        # Get the company object
+        company = get_object_or_404(Company, slug=slug)
+
+        # Check and see if the logged in user is admin
+        if company.admin_user != request.user:
+            return HttpResponseForbidden()
+
+        # Return the actual company object to the view
+        return view(request, company, *args, **kwargs)
+
+    return inner
diff --git a/new_view.py b/new_view.py
@@ -0,0 +1,9 @@
+@login_required
+@check_company_admin
+def some_view(request, company):
+    # Even though the slug is in the URL. We are getting a company 
+    # object back from the decorator
+
+    .... view logic ....
+
+    return render(request, 'company/index.html', {'company': company})
diff --git a/orginial_view.py b/orginial_view.py
@@ -0,0 +1,12 @@
+@login_required
+def some_view(request, slug):
+    # Get the company object based on slug
+    company = get_object_or_404(Company, slug=slug)
+
+    # Check and see if the logged in user is admin
+    if company.admin_user != request.user:
+        return HttpResponseForbidden()
+
+    .... view logic ....
+
+    return render(request, 'company/index.html', {'company': company})
diff --git a/url.py b/url.py
@@ -0,0 +1 @@
+url(r'^example/(?P<slug>[a-z0-9-]+)/$', 'some.view'),
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		url(r'^example/(?P<slug>[a-z0-9-]+)/$', 'some.view'),
No results found