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 finance.models import Bill, Invoice
from courses.models import Course

if __name__ == '__main__':
    Bill.objects.all().delete()
    with open('./finance/bill.csv') as bill_csv:
        bill_reader = csv.DictReader(bill_csv)
        for row in bill_reader:
            bill_kwarg = dict()
            row = dict(row)
            bill_kwarg['id'] = row.pop('id', None)
            bill_kwarg['course'] = Course.objects.get(id=row.pop('course__id', None))
            opener_id = row.pop('opener__id', None)
            bill_kwarg['opener'] = get_user_model().objects.get(id=opener_id) if opener_id \
                      else get_user_model().objects.get(email="kate.gazukina@skillbox.ru")

            email = row.pop('user__email', None)
            try:
                email = email[:email.index("\n")]
            except ValueError:
                pass

            try:
                bill_kwarg['user'] = get_user_model().objects.get(email=email.lower())
            except get_user_model().DoesNotExist:
                print([email, bill_kwarg['id']])

            bill_kwarg['comment'] = row.pop('comment', None)
            bill_kwarg['description'] = row.pop('description', None)
            bill = Bill.objects.create(**bill_kwarg)
            try:
                price = int(row.pop('price', None))
            except ValueError:
                price = None

            try:
                real_price = int(row.pop('real_price', None))
            except ValueError:
                real_price = None
            Invoice.objects.create(bill=bill, price=price, real_price=real_price, **row)