import csv
import django
import os
import sys

from django.contrib.auth import get_user_model
from django.db import IntegrityError

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

from yandex_money.models import Payment
from finance.models import Bill, Invoice
from course_service.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)
            try:
                bill = Bill.objects.create(**bill_kwarg)
                method = row.pop('bill_method', None)
                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

                inv = Invoice.objects.create(bill=bill, method=method, price=price, real_price=real_price, **row)

                if method == 'Y':
                    yandex_pay, _is_create = Payment.objects.get_or_create(
                        order_amount=price,
                        order_number=inv.id,
                        customer_number=bill.user.id,
                        user=bill.user,
                        cps_email=bill.user.email,
                        shop_amount=inv.real_price,
                    )
                    inv.yandex_pay = yandex_pay

                inv.date = row['date']
                inv.save()

            except IntegrityError:
                pass