mirror of
https://git.linux-kernel.at/oliver/ivatar.git
synced 2025-11-11 18:56:23 +00:00
79 lines
2.3 KiB
Python
79 lines
2.3 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
Import a CSV - Format as follows:
|
|
<mailaddr>,<path_to_image>
|
|
|
|
Example:
|
|
myuser@mydomain.tld,myphoto.jpeg
|
|
|
|
This will create or update an existing user and assign the image
|
|
to the given address.
|
|
"""
|
|
|
|
import os
|
|
from os.path import isfile
|
|
import sys
|
|
from io import BytesIO
|
|
import csv
|
|
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.models import ConfirmedEmail
|
|
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 CSV" % sys.argv[0])
|
|
exit(-255)
|
|
|
|
if not isfile(sys.argv[1]):
|
|
print("First argument to '%s' must be a path to the CSV" % sys.argv[0])
|
|
exit(-255)
|
|
|
|
PATH = sys.argv[1]
|
|
with open(PATH, newline="") as csvfile:
|
|
contactreader = csv.reader(csvfile, delimiter=",")
|
|
for row in contactreader:
|
|
mailaddr = row[0]
|
|
image = row[1]
|
|
|
|
if not isfile(image):
|
|
print("File '%s' doesn't exist - cannot add" % image)
|
|
continue
|
|
|
|
print("Adding: %s" % mailaddr)
|
|
|
|
(user, created) = User.objects.get_or_create(username=mailaddr)
|
|
if not user.confirmedemail_set.count() < 1:
|
|
ConfirmedEmail.objects.get_or_create(
|
|
email=mailaddr,
|
|
user=user,
|
|
)
|
|
user.save()
|
|
with open(image, "rb") as avatar:
|
|
pilobj = Image.open(avatar)
|
|
out = BytesIO()
|
|
pilobj.save(out, pilobj.format, quality=JPEG_QUALITY)
|
|
out.seek(0)
|
|
photo = None
|
|
if user.photo_set.count() < 1:
|
|
photo = Photo()
|
|
photo.user = user
|
|
else:
|
|
photo = user.photo_set.first()
|
|
photo.ip_address = "0.0.0.0"
|
|
photo.format = file_format(pilobj.format)
|
|
photo.data = out.read()
|
|
photo.save()
|
|
print("xxx: %s" % user.confirmedemail_set.first())
|
|
confirmed_email = user.confirmedemail_set.first()
|
|
confirmed_email.photo_id = photo.id
|
|
confirmed_email.save()
|