#!/usr/bin/env python
'''
Import the whole libravatar export
'''

import os
from os.path import isfile, isdir, join
import sys
import base64
from io import BytesIO
import django
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "ivatar.settings")  # pylint: disable=wrong-import-position
django.setup()  # pylint: disable=wrong-import-position
from django.contrib.auth.models import User
from PIL import Image
from ivatar.settings import JPEG_QUALITY
from ivatar.ivataraccount.read_libravatar_export import read_gzdata as libravatar_read_gzdata
from ivatar.ivataraccount.models import ConfirmedEmail
from ivatar.ivataraccount.models import ConfirmedOpenId
from ivatar.ivataraccount.models import Photo
from ivatar.ivataraccount.models import file_format

if len(sys.argv) < 2:
    print("First argument to '%s' must be the path to the exports" % sys.argv[0])
    exit(-255)

if not isdir(sys.argv[1]):
    print("First argument to '%s' must be a directory containing the exports" % sys.argv[0])
    exit(-255)

PATH = sys.argv[1]
for file in os.listdir(PATH):
    if not file.endswith('.xml.gz'):
        continue
    if isfile(join(PATH, file)):
        fh = open(join(PATH, file), 'rb')
        items = libravatar_read_gzdata(fh.read())
        print('Adding user "%s"' % items['username'])
        (user, created) = User.objects.get_or_create(username=items['username'])
        user.password = items['password']
        user.save()

        saved_photos = {}
        for photo in items['photos']:
            photo_id = photo['id']
            data = base64.decodebytes(bytes(photo['data'], 'utf-8'))
            pilobj = Image.open(BytesIO(data))
            out = BytesIO()
            pilobj.save(out, pilobj.format, quality=JPEG_QUALITY)
            out.seek(0)
            photo = Photo()
            photo.user = user
            photo.ip_address = '0.0.0.0'
            photo.format = file_format(pilobj.format)
            photo.data = out.read()
            photo.save()
            saved_photos[photo_id] = photo

        for email in items['emails']:
            try:
                ConfirmedEmail.objects.get_or_create(email=email['email'], user=user,
                                                     photo=saved_photos.get(email['photo_id']))
            except django.db.utils.IntegrityError:
                print('%s not unique?' % email['email'])

        for openid in items['openids']:
            try:
                ConfirmedOpenId.objects.get_or_create(openid=openid['openid'], user=user,
                                                      photo=saved_photos.get(openid['photo_id'])) # pylint: disable=no-member
            except django.db.utils.IntegrityError:
                print('%s not unique?' % openid['openid'])

        fh.close()