mirror of
https://github.com/jazzband/django-avatar.git
synced 2026-03-16 22:20:30 +00:00
81ace968c7
* Add Avatar API support * extend rest_framework to INSTALLED_APPS * add api path to urls * add requirements.txt to api app * add self describe to assign_width_or_height function * add django-avatar api docs * remove unused files * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Johannes Wilm <mail@johanneswilm.org>
56 lines
2.4 KiB
Python
56 lines
2.4 KiB
Python
import os
|
|
|
|
from django.db.models import signals
|
|
|
|
from avatar.api.conf import settings as api_settings
|
|
from avatar.api.shortcut import get_object_or_none
|
|
from avatar.conf import settings
|
|
from avatar.models import Avatar, invalidate_avatar_cache
|
|
|
|
|
|
def create_default_thumbnails(sender, instance, created=False, **kwargs):
|
|
invalidate_avatar_cache(sender, instance)
|
|
|
|
if not created:
|
|
for size in settings.AVATAR_AUTO_GENERATE_SIZES:
|
|
if isinstance(size, int):
|
|
if not instance.thumbnail_exists(size, size):
|
|
instance.create_thumbnail(size, size)
|
|
else:
|
|
# Size is specified with height and width.
|
|
if not instance.thumbnail_exists(size[0, size[0]]):
|
|
instance.create_thumbnail(size[0], size[1])
|
|
|
|
|
|
def remove_previous_avatar_images_when_update(
|
|
sender, instance=None, created=False, update_main_avatar=True, **kwargs
|
|
):
|
|
if not created:
|
|
old_instance = get_object_or_none(Avatar, pk=instance.pk)
|
|
if old_instance and not old_instance.avatar == instance.avatar:
|
|
base_filepath = old_instance.avatar.name
|
|
path, filename = os.path.split(base_filepath)
|
|
# iterate through resized avatars directories and delete resized avatars
|
|
resized_path = os.path.join(path, "resized")
|
|
try:
|
|
resized_widths, _ = old_instance.avatar.storage.listdir(resized_path)
|
|
for width in resized_widths:
|
|
resized_width_path = os.path.join(resized_path, width)
|
|
resized_heights, _ = old_instance.avatar.storage.listdir(
|
|
resized_width_path
|
|
)
|
|
for height in resized_heights:
|
|
if old_instance.thumbnail_exists(width, height):
|
|
old_instance.avatar.storage.delete(
|
|
old_instance.avatar_name(width, height)
|
|
)
|
|
if update_main_avatar:
|
|
if old_instance.avatar.storage.exists(old_instance.avatar.name):
|
|
old_instance.avatar.storage.delete(old_instance.avatar.name)
|
|
except FileNotFoundError as e:
|
|
pass
|
|
|
|
|
|
if api_settings.API_AVATAR_CHANGE_IMAGE:
|
|
signals.pre_save.connect(remove_previous_avatar_images_when_update, sender=Avatar)
|
|
signals.post_save.connect(create_default_thumbnails, sender=Avatar)
|