'"Daniel Diniz" <ajaksu@gmail.com>',

import time

from google.appengine.ext.db import djangoforms

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 decorators

from soc.views.helper import redirects

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']

    new_params['logic'] = survey_group_logic

    new_params['create_form'] = GroupForm

    new_params['edit_form'] = GroupForm

  @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

edit = decorators.view(view.edit)