vrr/agent-zero
2
0
Fork 0
mirror of https://github.com/agent0ai/agent-zero.git synced 2026-04-28 11:40:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

better filename check

Browse source
This commit is contained in:
linkliti 2026-01-24 21:54:21 +03:00
parent c4846e7e9d
commit e5669daf67
9 changed files with 75 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

9
python/helpers/attachment_manager.py
View file

@ -3,7 +3,8 @@ import io
import base64
from PIL import Image
from typing import Dict, List, Optional, Tuple
from werkzeug.utils import secure_filename
from python.helpers.security import safe_filename
from werkzeug.datastructures import FileStorage
from python.helpers.print_style import PrintStyle
@ -41,10 +42,10 @@ class AttachmentManager:
except AttributeError:
return False
def save_file(self, file, filename: str) -> Tuple[str, Dict]:
def save_file(self, file: FileStorage, name: str) -> Tuple[str, Dict]:
"""Save file and return path and metadata"""
try:
filename = secure_filename(filename)
filename = safe_filename(name)
if not filename:
raise ValueError("Invalid filename")
@ -68,7 +69,7 @@ class AttachmentManager:
return file_path, metadata
except Exception as e:
PrintStyle.error(f"Error saving file {filename}: {e}")
PrintStyle.error(f"Error saving file {name}: {e}")
return None, {} # type: ignore
def generate_image_preview(self, image_path: str, max_size: int = 800) -> Optional[str]: