import os, sys, django, csv

sys.path.append("../")
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "lms.settings")
django.setup()

from courses.models import Course

if __name__ == '__main__':
    with open('./course/course.csv') as user_csv:
        user_reader = csv.DictReader(user_csv)
        for row in user_reader:
            row = dict(row)
            teachers = row.pop('teachers', None).split("[")[1].split("]")[0].split(", ")
            mentors = row.pop('mentors', None).split("[")[1].split("]")[0].split(", ")
            course, _is_create = Course.objects.get_or_create(**row)

            for teacher in teachers:
                if teacher:
                    course.teachers.add(teacher)

            for mentor in mentors:
                if mentor:
                    course.mentors.add(mentor)