fix: Improve file extension handling in image upload utility

2025-03-06 09:30:46 +08:00 · 2025-03-06 09:30:46 +08:00 · 9f5be12b2e
commit 9f5be12b2e
parent 9e32de9922
1 changed files with 1 additions and 0 deletions
--- a/utils.py
+++ b/utils.py
@ -276,6 +276,7 @@ def upload_image(image_path, file_name=None):
            file_extension = os.path.splitext(image_path)[1]
            object_key = f'dotabot/{timestamp}_{random_id}{file_extension}'
        else:
+            file_extension = os.path.splitext(file_name)[1]
            object_key = f'dotabot/{file_name}'
        
        # 设置文件的Content-Type