import csv
import os
import random
import string
import sys

import django

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

from storage.models import File

if __name__ == '__main__':
    File.objects.all().delete()
    with open('./course/storage.csv') as storage_csv:
        storage_reader = csv.DictReader(storage_csv)
        for row in storage_reader:
            if row['original']:
                key = ''.join(random.choice(string.ascii_letters) for _x in range(15))
                File.objects.create(original=row['original'], id=row['id'])