# HG changeset patch
# User Sverre Rabbelier <srabbelier@gmail.com>
# Date 1228676522 0
# Node ID e71b20847eb0bfcb831c9e74531cf9bf32653e88
# Parent  89f1a8c2ea59884de4b28e2d94f26f221b6b5785
Add checkIsHost in access

Patch by: Sverre Rabbelier

diff -r 89f1a8c2ea59 -r e71b20847eb0 app/soc/views/helper/access.py
--- a/app/soc/views/helper/access.py	Sun Dec 07 19:01:35 2008 +0000
+++ b/app/soc/views/helper/access.py	Sun Dec 07 19:02:02 2008 +0000
@@ -35,6 +35,7 @@
 from django.utils.translation import ugettext_lazy
 
 from soc.logic import accounts
+from soc.logic.models import host as host_logic
 from soc.logic.models import user as user_logic
 from soc.logic.models import request as request_logic
 from soc.views import helper
@@ -115,7 +116,7 @@
     alternate response that should be returned by the calling view.
   """
 
-  context = helper.responses.getUniversalContext(request)
+  context = {}
   context['title'] = 'Access denied'
 
   raise out_of_band.AccessViolation(DEF_PAGE_DENIED_MSG, context=context)
@@ -215,6 +216,46 @@
   raise out_of_band.LoginRequest(message_fmt=login_message_fmt)
 
 
+def checkIsHost(request):
+  """Returns an alternate HTTP response if Google Account has no Host entity
+     for the specified program.
+
+  Args:
+    request: a Django HTTP request
+
+   Raises:
+     AccessViolationResponse: if the required authorization is not met
+
+  Returns:
+    None if Host exists for the specified program, or a subclass of
+    django.http.HttpResponse which contains the alternate response
+    should be returned by the calling view.
+  """
+
+  try:
+    # if the current user is a developer we allow access
+    checkIsInvited(request)
+    return
+  except out_of_band.Error:
+    pass
+
+  checkIsUser(request)
+
+  user = user_logic.logic.getForFields(
+      {'account': users.get_current_user()}, unique=True)
+
+  host = host_logic.logic.getForFields(
+      {'user': user}, unique=True)
+
+  if host:
+    return
+
+  login_message_fmt = DEF_DEV_LOGOUT_LOGIN_MSG_FMT % {
+      'role': 'a host '}
+
+  raise out_of_band.LoginRequest(message_fmt=login_message_fmt)
+
+
 def checkIsInvited(request):
   """Returns an alternate HTTP response if Google Account has no Host entity
      for the specified program.
@@ -231,15 +272,15 @@
     should be returned by the calling view.
   """
 
-  checkIsUser(request)
-  
   try:
     # if the current user is a developer we allow access
-    checkIsDeveloper(request)  
+    checkIsDeveloper(request)
     return
   except out_of_band.Error:
     pass
-  
+
+  checkIsUser(request)
+
   login_message_fmt = DEF_DEV_LOGOUT_LOGIN_MSG_FMT % {
       'role': 'a host for this program'}