mirror of
https://git.linux-kernel.at/oliver/ivatar.git
synced 2025-11-17 13:38:03 +00:00
146 lines
4.4 KiB
Python
Executable File
146 lines
4.4 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
# Copyright (C) 2018 Oliver Falk <oliver@linux-kernel.at>
|
|
#
|
|
# This file is part of Libravatar
|
|
#
|
|
# Libravatar is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published
|
|
# by the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Libravatar is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with Libravatar. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import base64
|
|
import gzip
|
|
import json
|
|
import os
|
|
import sys
|
|
from xml.sax import saxutils
|
|
import hashlib
|
|
|
|
# pylint: disable=relative-import
|
|
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "libravatar.settings")
|
|
import libravatar.settings as settings
|
|
from libravatar.utils import create_logger, is_hex
|
|
import django
|
|
django.setup()
|
|
from django.contrib.auth.models import User
|
|
|
|
os.umask(022)
|
|
LOGGER = create_logger('exportaccount')
|
|
|
|
SCHEMA_ROOT = 'https://www.libravatar.org/schemas/export/0.2'
|
|
SCHEMA_XSD = '%s/export.xsd' % SCHEMA_ROOT
|
|
|
|
|
|
def xml_header():
|
|
return '''<?xml version="1.0" encoding="UTF-8"?>
|
|
<user xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
|
xsi:schemaLocation="%s %s"
|
|
xmlns="%s">\n''' % (SCHEMA_ROOT, SCHEMA_XSD, SCHEMA_ROOT)
|
|
|
|
|
|
def xml_footer():
|
|
return '</user>\n'
|
|
|
|
|
|
def xml_account(username):
|
|
escaped_username = saxutils.quoteattr(username)
|
|
escaped_site_url = saxutils.quoteattr(settings.SITE_URL)
|
|
return ' <account username=%s site=%s/>\n' % (escaped_username, escaped_site_url)
|
|
|
|
def xml_email(emails):
|
|
returnstring = " <emails>\n"
|
|
for email in emails:
|
|
returnstring += ' <email photo_id="' + str(email.photo_id) + '">' + email.email.encode('utf-8') + '</email>' + "\n"
|
|
returnstring += " </emails>\n"
|
|
return returnstring
|
|
|
|
def xml_openid(openids):
|
|
returnstring = " <openids>\n"
|
|
for openid in openids:
|
|
returnstring += ' <openid photo_id="' + str(openid.photo_id) + '">' + openid.openid.encode('utf-8') + '</openid>' + "\n"
|
|
returnstring += " </openids>\n"
|
|
return returnstring
|
|
|
|
|
|
def xml_photos(photos):
|
|
s = ' <photos>\n'
|
|
for photo in photos:
|
|
(photo_filename, photo_format, id) = photo
|
|
encoded_photo = encode_photo(photo_filename, photo_format)
|
|
if encoded_photo:
|
|
s += ''' <photo id="%s" encoding="base64" format=%s>
|
|
%s
|
|
</photo>\n''' % (id, saxutils.quoteattr(photo_format), encoded_photo)
|
|
s += ' </photos>\n'
|
|
return s
|
|
|
|
|
|
def encode_photo(photo_filename, photo_format):
|
|
filename = settings.USER_FILES_ROOT + photo_filename + '.' + photo_format
|
|
if not os.path.isfile(filename):
|
|
LOGGER.warning('Photo not found: %s', filename)
|
|
return None
|
|
|
|
photo_content = None
|
|
with open(filename) as photo:
|
|
photo_content = photo.read()
|
|
|
|
if not photo_content:
|
|
LOGGER.warning('Could not read photo: %s', filename)
|
|
return None
|
|
|
|
return base64.b64encode(photo_content)
|
|
|
|
|
|
def main(argv=None, dryrun=False):
|
|
if argv is None:
|
|
argv = sys.argv
|
|
|
|
if(len(sys.argv) > 1):
|
|
userobjs = User.objects.filter(username=sys.argv[1])
|
|
else:
|
|
userobjs = User.objects.all()
|
|
if(len(sys.argv) > 2):
|
|
dryrun = True
|
|
|
|
for user in userobjs:
|
|
hash_object = hashlib.new('sha256')
|
|
hash_object.update(user.username + user.password)
|
|
file_hash = hash_object.hexdigest()
|
|
photos = []
|
|
for photo in user.photos.all():
|
|
photo_details = (photo.filename, photo.format, photo.id)
|
|
photos.append(photo_details)
|
|
username = user.username
|
|
|
|
dest_filename = settings.EXPORT_FILES_ROOT + file_hash + '.xml.gz'
|
|
data = ''
|
|
data += xml_header()
|
|
data += xml_account(username)
|
|
data += xml_email(user.confirmed_emails.all())
|
|
data += xml_openid(user.confirmed_openids.all())
|
|
data += xml_photos(photos)
|
|
data += xml_footer()
|
|
|
|
if not dryrun:
|
|
destination = gzip.open(dest_filename, 'w')
|
|
destination.write(data)
|
|
destination.close()
|
|
print(dest_filename)
|
|
else:
|
|
print(data)
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|