allow sync and async function calls

This commit is contained in:
2026-04-03 14:10:38 +02:00
parent 51de53d01c
commit a82eaaaec5
+8 -2
View File
@@ -1,6 +1,7 @@
import aiofiles.os
import aiofiles
import os
import inspect
async def read_file(path: str, callback=None):
if not await aiofiles.os.path.exists(path):
@@ -8,7 +9,10 @@ async def read_file(path: str, callback=None):
async with aiofiles.open(path, "r") as f:
if callback:
return await callback(f)
result = callback(f)
if inspect.isawaitable(result):
return await result
return result
return await f.read()
@@ -19,6 +23,8 @@ async def save_file(path: str, data, callback=None, *args, **kwargs):
async with aiofiles.open(path, "w") as f:
if callback:
await callback(data, f, *args, **kwargs)
result = callback(data, f, *args, **kwargs)
if inspect.isawaitable(result):
await result
else:
await f.write(data)