views.py 文件源码

def upload_avatar(request):
        user = request.user
        user_avatar = request.FILES.get('user_avatar', None)
        if request.method == 'POST' and user_avatar:
            ext = os.path.splitext(user_avatar.name)[1]
            if ext.lower() in ['.jpg', '.jpeg', '.png']:
                filename = user.username + '.jpg'
                fs = FileSystemStorage()
                if fs.exists(filename):
                    fs.delete(filename)
                fs.save(filename, user_avatar)
                user_account = Account.get_by_user(user=user)
                request.session['user_avatar'] = user_account.user_avatar
                request.session.save()
        return redirect('account_settings')