+
−#!/usr/bin/python2.5
+
−#
+
−# Copyright 2009 the Melange authors.
+
−#
+
−# Licensed under the Apache License, Version 2.0 (the "License");
+
−# you may not use this file except in compliance with the License.
+
−# You may obtain a copy of the License at
+
−#
+
−#   http://www.apache.org/licenses/LICENSE-2.0
+
−#
+
−# Unless required by applicable law or agreed to in writing, software
+
−# distributed under the License is distributed on an "AS IS" BASIS,
+
−# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+
−# See the License for the specific language governing permissions and
+
−# limitations under the License.
+
−
+
−"""SurveyRecord (Model) query functions.
+
−"""
+
−
+
−__authors__ = [
+
−  '"Lennard de Rijk" <ljvderijk@gmail.com>',
+
−  ]
+
−
+
−
+
−from google.appengine.ext import db
+
−
+
−from soc.logic.models import work
+
−from soc.models.survey_record import SurveyRecord
+
−from soc.models.grading_project_survey_record import GradingProjectSurveyRecord
+
−from soc.models.project_survey_record import ProjectSurveyRecord
+
−
+
−
+
−class Logic(work.Logic):
+
−  """Logic methods for listing results for Surveys.
+
−  """
+
−
+
−  def __init__(self, model=SurveyRecord,
+
−               base_model=None, scope_logic=None):
+
−    """Defines the name, key_name and model for this entity.
+
−    """
+
−
+
−    super(Logic, self).__init__(model=model, base_model=base_model,
+
−                                scope_logic=scope_logic)
+
−
+
−  def updateSurveyRecord(self, user, survey, survey_record, fields):
+
−    """ Create a new survey record, or get an existing one.
+
−
+
−    params:
+
−      user = user taking survey
+
−      survey = survey entity
+
−      survey_record = existing record, if one exists
+
−      fields = submitted responses to survey fields
+
−    """
+
−
+
−    if survey_record:
+
−      create = False
+
−      for prop in survey_record.dynamic_properties():
+
−        delattr(survey_record, prop)
+
−    else:
+
−      create = True
+
−      Record = self.getModel()
+
−      survey_record = Record(user=user, survey=survey)
+
−
+
−    schema = eval(survey.survey_content.schema)
+
−
+
−    for name, value in fields.items():
+
−      # TODO(ajaksu) logic below can be improved now we have different models
+
−      if name == 'project':
+
−        project = student_project.StudentProject.get(value)
+
−        survey_record.project = project
+
−      elif name == 'grade':
+
−        survey_record.grade = GRADES[value]
+
−      else:
+
−        pick_multi = name in schema and schema[name]['type'] == 'pick_multi'
+
−        if pick_multi and hasattr(fields, 'getlist'): # it's a multidict
+
−          setattr(survey_record, name, ','.join(fields.getlist(name)))
+
−        else:
+
−          setattr(survey_record, name, value)
+
−
+
−    # if creating evaluation record, set SurveyRecordGroup
+
−    db.put(survey_record)
+
−    return survey_record
+
−
+
−
+
−class ProjectLogic(Logic):
+
−  """Logic class for ProjectSurveyRecord.
+
−  """
+
−
+
−  def __init__(self, model=ProjectSurveyRecord,
+
−               base_model=SurveyRecord, scope_logic=None):
+
−    """Defines the name, key_name and model for this entity.
+
−    """
+
−
+
−    super(Logic, self).__init__(model=model, base_model=base_model,
+
−                                scope_logic=scope_logic)
+
−
+
−
+
−class GradingProjectLogic(ProjectLogic):
+
−  """Logic class for GradingProjectSurveyRecord
+
−  """
+
−
+
−  def __init__(self, model=GradingProjectSurveyRecord,
+
−               base_model=ProjectSurveyRecord, scope_logic=None):
+
−    """Defines the name, key_name and model for this entity.
+
−    """
+
−
+
−    super(Logic, self).__init__(model=model, base_model=base_model,
+
−                                scope_logic=scope_logic)
+
−
+
−
+
−logic = Logic()
+
−project_logic = ProjectLogic()
+
−grading_logic = GradingProjectLogic()
+
−
+
−
+
−def updateSurveyRecord(user, survey, survey_record, fields):
+
−  """Create a new survey record, or get an existing one.
+
−
+
−  params:
+
−    user = user taking survey
+
−    survey = survey entity
+
−    survey_record = existing record, if one exists
+
−    fields = submitted responses to survey fields
+
−  """
+
−
+
−  # TODO(ajaksu) We should use class information here, but being careful about
+
−  # compatibility with existent records should the class change.
+
−  if hasattr(survey_record, 'grade'):
+
−    record_logic = grading_logic
+
−  elif hasattr(survey_record, 'project'):
+
−    record_logic = grading_logic
+
−  else:
+
−    record_logic = logic
+
−
+
−  return record_logic.updateSurveyRecord(user, survey, survey_record, fields)