uw-it-aca / canvas-sis-provisioner / 4076774980

from django.core.management.base import BaseCommand, CommandError

from django.core.files.storage import default_storage

from django.utils.timezone import utc

from sis_provisioner.dao.canvas import (

from sis_provisioner.models.course import Course

from uw_canvas.terms import Terms

from dateutil.parser import parse

from logging import getLogger

import csv

logger = getLogger(__name__)

class Command(BaseCommand):

    help = "Insert courses from file."

    def add_arguments(self, parser):

        parser.add_argument(

    def get_all_terms(self):

        terms = {}

        for term in Terms().get_all_terms():

            terms[term.term_id] = term.sis_term_id or 'default'

        return terms

    def handle(self, *args, **options):

        course_file = options.get('course_file')

        terms = self.get_all_terms()

        with default_storage.open(course_file, mode='r') as csvfile:

            reader = csv.reader(csvfile)

            for row in reader:

                canvas_id = row[1]

                term_id = row[4]

                created_at = parse(row[8]).replace(tzinfo=utc)

                sis_source_id = row[13]

                workflow_state = row[14]

                term_sis_id = terms.get(term_id)

                if not sis_source_id:

                    sis_source_id = adhoc_course_sis_id(canvas_id)

                    try:

                        update_course_sis_id(canvas_id, sis_source_id)

                    except DataFailureException as ex:

                        logger.info('Add sis_id for course {}: {}'.format(

                        continue

                try:

                    course = Course.objects.get(course_id=sis_source_id)

                    course.created_date = created_at

                except Course.DoesNotExist:

                    course = Course(

                course.save()