AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-04-23 06:09:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

add the filename length limit (#5326)

Browse Source
This commit is contained in:
Jyong 2024-06-17 20:36:54 +08:00 committed by GitHub
parent 147a39b984
commit 07387e9586
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/services/file_service.py
View File

@ -31,7 +31,10 @@ class FileService:
@staticmethod @staticmethod
def upload_file(file: FileStorage, user: Union[Account, EndUser], only_image: bool = False) -> UploadFile: def upload_file(file: FileStorage, user: Union[Account, EndUser], only_image: bool = False) -> UploadFile:
filename = file.filename
extension = file.filename.split('.')[-1] extension = file.filename.split('.')[-1]
if len(filename) > 200:
filename = filename.split('.')[0][:200] + '.' + extension
etl_type = current_app.config['ETL_TYPE'] etl_type = current_app.config['ETL_TYPE']
allowed_extensions = UNSTRUSTURED_ALLOWED_EXTENSIONS + IMAGE_EXTENSIONS if etl_type == 'Unstructured' \ allowed_extensions = UNSTRUSTURED_ALLOWED_EXTENSIONS + IMAGE_EXTENSIONS if etl_type == 'Unstructured' \
else ALLOWED_EXTENSIONS + IMAGE_EXTENSIONS else ALLOWED_EXTENSIONS + IMAGE_EXTENSIONS
@ -75,7 +78,7 @@ class FileService:
tenant_id=current_tenant_id, tenant_id=current_tenant_id,
storage_type=config['STORAGE_TYPE'], storage_type=config['STORAGE_TYPE'],
key=file_key, key=file_key,
name=file.filename, name=filename,
size=file_size, size=file_size,
extension=extension, extension=extension,
mime_type=file.mimetype, mime_type=file.mimetype,
@ -93,6 +96,8 @@ class FileService:
@staticmethod @staticmethod
def upload_text(text: str, text_name: str) -> UploadFile: def upload_text(text: str, text_name: str) -> UploadFile:
if len(text_name) > 200:
text_name = text_name[:200]
# user uuid as file name # user uuid as file name
file_uuid = str(uuid.uuid4()) file_uuid = str(uuid.uuid4())
file_key = 'upload_files/' + current_user.current_tenant_id + '/' + file_uuid + '.txt' file_key = 'upload_files/' + current_user.current_tenant_id + '/' + file_uuid + '.txt'