update: Add exception handling for write file operation.

This commit is contained in:
Stitch-z 2023-09-07 20:08:02 +08:00
parent d83d1e105c
commit b2b227391a

View file

@ -27,9 +27,13 @@ class File:
Returns:
The full filename of file, such as "/data/test.txt".
"""
root_path.mkdir(parents=True, exist_ok=True)
full_path = root_path / filename
async with aiofiles.open(full_path, mode="wb") as writer:
await writer.write(content)
logger.info(f"Successfully write docx: {full_path}")
return full_path
try:
root_path.mkdir(parents=True, exist_ok=True)
full_path = root_path / filename
async with aiofiles.open(full_path, mode="wb") as writer:
await writer.write(content)
logger.info(f"Successfully write docx: {full_path}")
return full_path
except Exception as e:
logger.error(f"Error writing file: {e}")
raise e