# HG changeset patch
# User Daniel Diniz <ajaksu@gmail.com>
# Date 1247220389 -7200
# Node ID 37584af2420e9fc0a4d04b4ee3165bbc7f717dbe
# Parent  10858e5d5712834a758b388977568a76a96707a4
Completed the Create and Edit view for GradingSurveyGroup.

Note that I've added a TODO to redo _editPost to give us kwargs. Would also get rid of hidden fields elsewhere.

Reviewed by: Lennard de Rijk

diff -r 10858e5d5712 -r 37584af2420e app/soc/views/models/grading_survey_group.py
--- a/app/soc/views/models/grading_survey_group.py	Fri Jul 10 11:59:00 2009 +0200
+++ b/app/soc/views/models/grading_survey_group.py	Fri Jul 10 12:06:29 2009 +0200
@@ -18,15 +18,75 @@
 """
 
 __authors__ = [
+    '"Daniel Diniz" <ajaksu@gmail.com>',
     '"Lennard de Rijk" <ljvderijk@gmail.com>',
   ]
 
 
+import time
+
+from google.appengine.ext.db import djangoforms
+
 from soc.logic import dicts
+from soc.logic.models.program import logic as program_logic
+from soc.logic.models.survey import grading_logic
+from soc.logic.models.survey import project_logic
+from soc.logic.models.user import logic as user_logic
+from soc.logic.models.grading_survey_group import logic as survey_group_logic
+from soc.models.grading_survey_group import GradingSurveyGroup
+from soc.models.grading_project_survey import GradingProjectSurvey
+from soc.models.project_survey import ProjectSurvey
 from soc.views.helper import access
+from soc.views.helper import decorators
+from soc.views.helper import redirects
 from soc.views.models import base
 
-import soc.logic.models.grading_survey_group
+
+class GroupForm(djangoforms.ModelForm):
+  """Form for creating a GradingSurveyGroup.
+  """
+
+  grading_survey = djangoforms.ModelChoiceField(GradingProjectSurvey)
+
+  student_survey = djangoforms.ModelChoiceField(ProjectSurvey, required=False)
+
+  def __init__(self, *args, **kwargs):
+    """Process field names for readable display and initialize form.
+    """
+
+    # use survey titles in drop-downs
+    self.choiceTitles('grading_survey', grading_logic)
+    self.choiceTitles('student_survey', project_logic)
+
+    super(GroupForm, self).__init__(*args, **kwargs)
+
+  def choiceTitles(self, field, logic):
+    """Fetch entity titles for choice field entries.
+    """
+
+    # TODO(ajaksu): subclass ModelChoiceField so we don't need this method
+    choice_list = []
+
+    model = logic.getModel()
+
+    for value, text in tuple(self.base_fields[field].choices):
+      if value:
+        entity = model.get(value)
+        text = entity.title
+      choice_list.append((value,text))
+
+    choices = tuple(choice_list)
+
+    self.base_fields[field].choices = choices
+
+  class Meta:
+    """Inner Meta class for fetching fields from model.
+    """
+    model = GradingSurveyGroup
+
+    # exclude the necessary fields from the form
+    exclude = ['link_id', 'scope', 'scope_path', 'last_update_started',
+               'last_update_complete']
 
 
 class View(base.View):
@@ -49,7 +109,7 @@
     rights['list'] = ['checkIsDeveloper']
 
     new_params = {}
-    new_params['logic'] = soc.logic.models.grading_survey_group.logic
+    new_params['logic'] = survey_group_logic
     new_params['rights'] = rights
     new_params['name'] = "Grading Survey Group"
 
@@ -57,17 +117,83 @@
     new_params['no_create_raw'] = True
     new_params['no_create_with_key_fields'] = True
 
-    patterns = []
+    new_params['create_form'] = GroupForm
+    new_params['edit_form'] = GroupForm
 
     params = dicts.merge(params, new_params)
 
     super(View, self).__init__(params=params)
 
+  @decorators.merge_params
+  @decorators.check_access
+  def create(self, request, access_type,
+             page_name=None, params=None, **kwargs):
+    """Pass the correct survey queries to GroupForm.
+
+    For params see base.View.create().
+    """
+
+    self.setQueries(kwargs['scope_path'], params)
+
+    return super(View, self).create(request, access_type, page_name=page_name,
+                                    params=params, **kwargs)
+
+  @decorators.merge_params
+  @decorators.check_access
+  def edit(self, request, access_type,
+           page_name=None, params=None, seed=None, **kwargs):
+    """Pass the correct survey queries to GroupForm.
+
+    For params see base.View.edit().
+    """
+
+    self.setQueries(kwargs['scope_path'], params)
+
+    return super(View, self).edit(request, access_type, page_name=page_name,
+                                  params=params, seed=seed, **kwargs)
+
+  def _editPost(self, request, entity, fields):
+    """See base.View._editPost().
+    """
+
+    if not entity:
+      # generate a unique link_id
+      fields['link_id'] = 't%i' % (int(time.time()*100))
+
+      # TODO: seriously redesign _editPost to pass along kwargs
+      fields['scope_path'] = fields['grading_survey'].scope_path
+    else:
+      fields['link_id'] = entity.link_id
+
+    # fill in the scope via call to super
+    super(View, self)._editPost(request, entity, fields)
+
+  def setQueries(self, program, params):
+    """Add program filtering queries to the GroupForm.
+    """
+
+    # fetch the program
+    program = program_logic.getFromKeyNameOr404(program)
+
+    # filter grading surveys by program and use title for display
+    grading_query = grading_logic.getQueryForFields(filter={'scope':program})
+
+    # filter project surveys by program and use title for display
+    student_query = project_logic.getQueryForFields(filter={'scope':program})
+
+    if params.get('edit_form'):
+      params['edit_form'].base_fields['student_survey'].query = student_query
+      params['edit_form'].base_fields['grading_survey'].query = grading_query
+
+    if params.get('create_form'):
+      params['create_form'].base_fields['student_survey'].query = student_query
+      params['create_form'].base_fields['grading_survey'].query = grading_query
+
 
 view = View()
 
 create = decorators.view(view.create)
+edit = decorators.view(view.edit)
 delete = decorators.view(view.delete)
-edit = decorators.view(view.edit)
 list = decorators.view(view.list)
 public = decorators.view(view.public)