def test_survey_clone(db, create_event, dummy_event):
survey = Survey(event=dummy_event, title='test')
first_section = SurveySection(title='section 1',
display_as_section=True,
position=2)
survey.items.append(first_section)
first_section.children.append(
SurveyQuestion(title='question in s1',
field_type='text',
is_required=True))
second_section = SurveySection(title='section 2',
display_as_section=True,
position=1)
survey.items.append(second_section)
second_section.children.append(SurveyText(description='My text'))
second_section.children.append(
SurveyQuestion(title='What is your name?',
field_type='text',
is_required=False))
db.session.flush()
new_event = create_event()
cloner = EventSurveyCloner(dummy_event)
cloner.run(new_event, {}, {}, False)
assert len(new_event.surveys) == 1
assert len(new_event.surveys[0].items) == len(survey.items)
for i, item in enumerate(new_event.surveys[0].items):
for attr in get_simple_column_attrs(SurveyItem):
assert getattr(item, attr) == getattr(survey.items[i], attr)
def migrate_survey(self, evaluation, event):
survey = Survey(event_id=int(event.id))
if evaluation.title and not evaluation.title.startswith(
'Evaluation for '):
survey.title = _sanitize(evaluation.title)
if not survey.title:
survey.title = "Evaluation"
survey.introduction = _sanitize(evaluation.announcement)
if evaluation.contactInfo:
contact_text = "Contact: ".format(_sanitize(
evaluation.contactInfo))
survey.introduction += "\n\n{}".format(
contact_text) if survey.introduction else contact_text
survey.submission_limit = evaluation.submissionsLimit if evaluation.submissionsLimit else None
survey.anonymous = evaluation.anonymous
# Require the user to login if the survey is not anonymous or if logging in was required before
survey.require_user = not survey.anonymous or evaluation.mandatoryAccount
if evaluation.startDate.date() == date.min or evaluation.endDate.date(
) == date.min:
survey.start_dt = event.endDate
survey.end_dt = survey.start_dt + timedelta(days=7)
else:
survey.start_dt = localize_as_utc(evaluation.startDate, event.tz)
survey.end_dt = localize_as_utc(evaluation.endDate, event.tz)
if survey.end_dt < survey.start_dt:
survey.end_dt = survey.end_dt + timedelta(days=7)
for kind, notification in evaluation.notifications.iteritems():
survey.notifications_enabled = True
recipients = set(notification._toList) | set(notification._ccList)
if kind == 'evaluationStartNotify':
survey.start_notification_emails = list(recipients)
elif kind == 'newSubmissionNotify':
survey.new_submission_emails = list(recipients)
self.print_success(cformat('%{cyan}{}%{reset}').format(survey),
always=True,
event_id=event.id)
question_map = {}
section = SurveySection(survey=survey, display_as_section=False)
for position, old_question in enumerate(evaluation._questions):
question = self.migrate_question(old_question, position)
question_map[old_question] = question
section.children.append(question)
for old_submission in evaluation._submissions:
submission = self.migrate_submission(old_submission, question_map,
event.tz)
survey.submissions.append(submission)
return survey
def migrate_survey(self, evaluation):
survey = Survey(event_new=self.event)
title = convert_to_unicode(evaluation.title)
if title and not title.startswith('Evaluation for '):
survey.title = sanitize_user_input(title)
if not survey.title:
survey.title = "Evaluation"
survey.introduction = sanitize_user_input(evaluation.announcement)
if evaluation.contactInfo:
contact_text = "Contact: ".format(
sanitize_user_input(evaluation.contactInfo))
survey.introduction += "\n\n{}".format(
contact_text) if survey.introduction else contact_text
survey.submission_limit = evaluation.submissionsLimit if evaluation.submissionsLimit else None
survey.anonymous = evaluation.anonymous
# Require the user to login if the survey is not anonymous or if logging in was required before
survey.require_user = not survey.anonymous or evaluation.mandatoryAccount
if evaluation.startDate.date() == date.min or evaluation.endDate.date(
) == date.min:
survey.start_dt = self.event.end_dt
survey.end_dt = survey.start_dt + timedelta(days=7)
else:
survey.start_dt = self._naive_to_aware(evaluation.startDate)
survey.end_dt = self._naive_to_aware(evaluation.endDate)
if survey.end_dt < survey.start_dt:
survey.end_dt = survey.end_dt + timedelta(days=7)
for kind, notification in evaluation.notifications.iteritems():
survey.notifications_enabled = True
recipients = set(notification._toList) | set(notification._ccList)
if kind == 'evaluationStartNotify':
survey.start_notification_emails = list(recipients)
elif kind == 'newSubmissionNotify':
survey.new_submission_emails = list(recipients)
self.print_success('%[cyan]{}%[reset]'.format(survey))
question_map = {}
section = SurveySection(survey=survey, display_as_section=False)
for position, old_question in enumerate(evaluation._questions):
question = self.migrate_question(old_question, position)
question_map[old_question] = question
section.children.append(question)
for i, old_submission in enumerate(evaluation._submissions, 1):
submission = self.migrate_submission(old_submission, question_map,
i)
survey.submissions.append(submission)
survey._last_friendly_submission_id = len(survey.submissions)
return survey
def _process(self):
form = SurveyForm(obj=FormDefaults(require_user=True), event=self.event)
if form.validate_on_submit():
survey = Survey(event=self.event)
# add a default section so people can start adding questions right away
survey.items.append(SurveySection(display_as_section=False))
form.populate_obj(survey)
db.session.add(survey)
db.session.flush()
flash(_('Survey created'), 'success')
logger.info('Survey %s created by %s', survey, session.user)
return jsonify_data(flash=False)
return jsonify_template('events/surveys/management/edit_survey.html', event=self.event, form=form,
survey=None)
def _clone_surveys(self, new_event):
survey_attrs = get_simple_column_attrs(Survey) - {'uuid', 'start_dt', 'end_dt', '_last_friendly_submission_id'}
for old_survey in self.old_event.surveys:
if old_survey.is_deleted:
continue
survey = Survey()
survey.populate_from_attrs(old_survey, survey_attrs)
item_map = {}
for old_item in old_survey.items:
item = self._clone_item(survey, old_item)
if old_item.parent:
assert old_item.parent != old_item
try:
item.parent = item_map[old_item.parent]
except KeyError:
item.parent = item_map[old_item.parent] = self._clone_item(survey, old_item.parent)
item_map[old_item] = item
new_event.surveys.append(survey)
def _process(self):
form = SurveyForm(obj=FormDefaults(require_user=True),
event=self.event)
if form.validate_on_submit():
survey = Survey(event_new=self.event.as_event)
# add a default section so people can start adding questions right away
survey.items.append(SurveySection(display_as_section=False))
form.populate_obj(survey)
db.session.add(survey)
db.session.flush()
flash(_('Survey created'), 'success')
logger.info('Survey {} created by {}'.format(survey, session.user))
return redirect(url_for('.manage_survey', survey))
return WPManageSurvey.render_template('management/edit_survey.html',
self.event,
event=self.event,
form=form,
survey=None)
