mirror of
https://git.linux-kernel.at/oliver/ivatar.git
synced 2025-11-11 10:46:24 +00:00
94 lines
3.0 KiB
Python
94 lines
3.0 KiB
Python
#!/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 django_openid_auth.models import UserOpenID
|
|
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
|
|
UserOpenID.objects.get_or_create(
|
|
user_id=user.id,
|
|
claimed_id=openid["openid"],
|
|
display_id=openid["openid"],
|
|
)
|
|
except django.db.utils.IntegrityError:
|
|
print("%s not unique?" % openid["openid"])
|
|
|
|
fh.close()
|