def reset_collection(assignment_import, collection_type):
if collection_type == COHORT_COLLECTION_TYPE:
assignment = CohortAssignment()
assignment.override_previous = True
assignment.override_protected = True
assignment.cohort_number = assignment_import.cohort
elif collection_type == MAJOR_COLLECTION_TYPE:
assignment = MajorAssignment()
assignment.major_code = assignment_import.major
assignment.assignment_type = "file"
assignment.comments = assignment_import.comment
assignment.user = assignment_import.created_by
assignment.campus = assignment_import.campus
assignment.quarter = assignment_import.quarter
applicants_to_assign = []
for imp_assignment in assignment_import.get_assignments():
app = imp_assignment.get_application()
applicants_to_assign.append(app)
assignment.quarter = assignment_import.quarter
assignment.applicants = applicants_to_assign
client = AdSel()
if collection_type == COHORT_COLLECTION_TYPE:
return client.assign_cohorts_bulk(assignment)
elif collection_type == MAJOR_COLLECTION_TYPE:
return client.assign_majors(assignment)
def submit_collection(assignment_import):
(assignment_import, assignment) = _get_collection(assignment_import)
client = AdSel()
client.get_quarters()
if isinstance(assignment_import, SyskeyImport):
if assignment_import.cohort:
if assignment_import.upload_filename is not None:
return client.assign_cohorts_bulk(assignment)
else:
return client.assign_cohorts_manual(assignment)
elif assignment_import.major and len(assignment_import.major) > 0:
return client.assign_majors(assignment)
else:
return client.assign_purple_gold(assignment)