import os, sys, django, csv

from django.contrib.auth import get_user_model

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

from achievements.models import Diploma, Achievements

if __name__ == '__main__':
    Diploma.objects.all().delete()
    Achievements.objects.all().delete()
    with open('./achievement/achievement.csv') as achievements_csv:
        achievements_reader = csv.DictReader(achievements_csv)
        for row in achievements_reader:
            Achievements.objects.create(**row)

    with open('./achievement/user_achievement.csv') as achievements_csv:
        achievements_reader = csv.DictReader(achievements_csv)
        for row in achievements_reader:
            a = Achievements.objects.get(id=row['id'])
            a.users.add(get_user_model().objects.get(id=row['student_id']))

    with open('./achievement/diploma.csv') as achievements_csv:
        achievements_reader = csv.DictReader(achievements_csv)
        for row in achievements_reader:
            Diploma.objects.create(**row)