mirror of
https://github.com/haiwen/seahub.git
synced 2025-05-09 08:28:42 +00:00
f9b3e692ec
* update * update2 * other * other * update * update * Update test_user_avatar.py * Update migrate_avatars_fs2db.py --------- Co-authored-by: 孙永强 <11704063+s-yongqiang@user.noreply.gitee.com> Co-authored-by: r350178982 <32759763+r350178982@users.noreply.github.com>
70 lines
3.0 KiB
Python
70 lines
3.0 KiB
Python
# Copyright (c) 2012-2016 Seafile Ltd.
|
|
import os
|
|
import logging
|
|
|
|
from rest_framework.authentication import SessionAuthentication
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework.response import Response
|
|
from rest_framework.views import APIView
|
|
from rest_framework import status
|
|
|
|
from django.utils.translation import gettext as _
|
|
from django.template.defaultfilters import filesizeformat
|
|
|
|
from seahub.api2.authentication import TokenAuthentication
|
|
from seahub.api2.throttling import UserRateThrottle
|
|
from seahub.api2.utils import api_error
|
|
|
|
from seahub.avatar.models import Avatar
|
|
from seahub.avatar.signals import avatar_updated
|
|
from seahub.avatar.settings import (AVATAR_MAX_AVATARS_PER_USER,
|
|
AVATAR_MAX_SIZE, AVATAR_ALLOWED_FILE_EXTS)
|
|
from seahub.avatar.templatetags.avatar_tags import api_avatar_url
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class UserAvatarView(APIView):
|
|
authentication_classes = (TokenAuthentication, SessionAuthentication)
|
|
permission_classes = (IsAuthenticated,)
|
|
throttle_classes = (UserRateThrottle,)
|
|
|
|
def post(self, request):
|
|
|
|
image_file = request.FILES.get('avatar', None)
|
|
|
|
if not image_file:
|
|
error_msg = 'avatar invalid.'
|
|
return api_error(status.HTTP_400_BAD_REQUEST, error_msg)
|
|
|
|
(root, ext) = os.path.splitext(image_file.name.lower())
|
|
if AVATAR_ALLOWED_FILE_EXTS and ext not in AVATAR_ALLOWED_FILE_EXTS:
|
|
error_msg = _("%(ext)s is an invalid file extension. Authorized extensions are : %(valid_exts_list)s") % {'ext' : ext, 'valid_exts_list' : ", ".join(AVATAR_ALLOWED_FILE_EXTS)}
|
|
return api_error(status.HTTP_400_BAD_REQUEST, error_msg)
|
|
|
|
if image_file.size > AVATAR_MAX_SIZE:
|
|
error_msg = _("Your file is too big (%(size)s), the maximum allowed size is %(max_valid_size)s") % { 'size' : filesizeformat(image_file.size), 'max_valid_size' : filesizeformat(AVATAR_MAX_SIZE)}
|
|
return api_error(status.HTTP_400_BAD_REQUEST, error_msg)
|
|
|
|
username = request.user.username
|
|
count = Avatar.objects.filter(emailuser=username).count()
|
|
if AVATAR_MAX_AVATARS_PER_USER > 1 and count >= AVATAR_MAX_AVATARS_PER_USER:
|
|
error_msg = _("You already have %(nb_avatars)d avatars, and the maximum allowed is %(nb_max_avatars)d.") % { 'nb_avatars' : count, 'nb_max_avatars' : AVATAR_MAX_AVATARS_PER_USER}
|
|
return api_error(status.HTTP_400_BAD_REQUEST, error_msg)
|
|
|
|
try:
|
|
avatar = Avatar(
|
|
emailuser = username,
|
|
primary = True,
|
|
)
|
|
avatar.avatar.save(image_file.name, image_file)
|
|
avatar.save()
|
|
avatar_updated.send(sender=Avatar, user=request.user, avatar=avatar)
|
|
avatar_url, is_default, date_uploaded = api_avatar_url(username)
|
|
|
|
except Exception as e:
|
|
logger.error(e)
|
|
error_msg = 'Internal Server Error'
|
|
return api_error(status.HTTP_500_INTERNAL_SERVER_ERROR, error_msg)
|
|
|
|
return Response({'avatar_url': avatar_url})
|