Bennett/g0v0-server
1
0
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(storage): save MIME type into storage service

Browse Source
This commit is contained in:
MingxuanGame
2025-08-30 12:36:43 +00:00
parent c167cbcea6
commit 7d79f3cee7
6 changed files with 16 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/router/private/avatar.py
View File

@@ -35,7 +35,7 @@ async def upload_avatar(
"""
# check file
check_image(content, 5 * 1024 * 1024, 256, 256)
format_ = check_image(content, 5 * 1024 * 1024, 256, 256)
if url := current_user.avatar_url:
path = storage.get_file_name_by_url(url)
@@ -45,7 +45,7 @@ async def upload_avatar(
filehash = hashlib.sha256(content).hexdigest()
storage_path = f"avatars/{current_user.id}_{filehash}.png"
if not await storage.is_exists(storage_path):
await storage.write_file(storage_path, content)
await storage.write_file(storage_path, content, f"image/{format_}")
url = await storage.get_file_url(storage_path)
current_user.avatar_url = url
await session.commit()