From c65d8f618df978919005b01142e70dc753551de0 Mon Sep 17 00:00:00 2001 From: lonkaars Date: Sun, 14 Mar 2021 18:55:15 +0100 Subject: WIP avatar upload --- api/user/avatar.py | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'api/user') diff --git a/api/user/avatar.py b/api/user/avatar.py index b0d2701..d3c86b8 100644 --- a/api/user/avatar.py +++ b/api/user/avatar.py @@ -3,6 +3,7 @@ from db import cursor from auth.login_token import token_login from user.info import valid_user_id from os.path import exists +from codecs import decode default_avatar = open("database/avatars/default.png", "rb").read() @@ -25,6 +26,15 @@ def get_avatar(): @avatar.route('/avatar', methods = ["POST"]) #TODO: pillow image size validation (client side resize) def update_avatar(): + token = request.cookies.get("token") or "" + if not token: return "", 401 + if not request.data: return "", 400 + + login = token_login(token) or "" + if not login: return "", 403 + + open(f"database/avatars/{login}.png", "wb").write(decode(request.data, "base64")) + return "", 200 dynamic_route = ["/user", avatar] -- cgit v1.2.3