# HG changeset patch
# User Lennard de Rijk <ljvderijk@gmail.com>
# Date 1249955875 25200
# Node ID 11d510941d2fe765e6883bdc140c2d99e0b613a5
# Parent  fb3bed3eecd8be3b880bf24e3f485d525134f7b8
Set the proper filters for the ProjectSurvey results view.

Also opened it up to users.

diff -r fb3bed3eecd8 -r 11d510941d2f app/soc/views/models/project_survey.py
--- a/app/soc/views/models/project_survey.py	Mon Aug 10 18:57:00 2009 -0700
+++ b/app/soc/views/models/project_survey.py	Mon Aug 10 18:57:55 2009 -0700
@@ -58,7 +58,7 @@
           [project_survey_logic, 'student', 'id']),
         ('checkIsSurveyReadable', [project_survey_logic]),
         ]])]
-    rights['results'] = ['checkIsDeveloper'] # TODO: proper access check
+    rights['results'] = ['checkIsUser']
     rights['take'] = [('checkIsSurveyTakeable', project_survey_logic),
                       ('checkIsAllowedToTakeProjectSurveyAs',
                        [project_survey_logic, 'student', 'project'])]
@@ -256,6 +256,55 @@
 
     return self._list(request, student_project_params, contents, page_name)
 
+  def _getResultsViewRecordFields(self, survey, allowed_to_read):
+    """Get the Results View filter for ProjectSurveyRecords.
+
+    For args see survey.View()._getResultsViewRecordFields()
+
+    Returns:
+      Returns the dictionary containing the fields to filter on
+    """
+
+    from soc.logic.models.org_admin import logic as org_admin_logic
+    from soc.logic.models.student import logic as student_logic
+
+    if allowed_to_read:
+      return super(View, self)._getResultsViewRecordFields(survey,
+                                                           allowed_to_read)
+
+    fields = {'survey': survey}
+
+    program_entity = survey.scope
+    user_entity = user_logic.getForCurrentAccount()
+
+    student_fields = {'scope': program_entity,
+                      'user': user_entity,
+                      'status': ['active', 'inactive']}
+    student_entity = student_logic.getForFields(student_fields, unique=True)
+
+    if student_entity:
+      # just get all records for the current user
+      fields['user'] = user_entity
+      return fields
+
+    org_admin_fields = {'user': user_entity,
+                        'program': program_entity,
+                        'status': ['active', 'inactive']}
+
+    org_admins = org_admin_logic.getForFields(org_admin_fields)
+
+    if org_admins:
+      # filter on all the organizations this user is org admin for
+      organizations = []
+
+      for org_admin in org_admins:
+        organizations.append(org_admin.scope)
+
+      # TODO: this might blow up if the user is org admin for too many orgs
+      fields['org'] = organizations
+
+    return fields
+
   @decorators.merge_params
   @decorators.check_access
   def sendReminder(self, request, access_type, page_name=None,