allow sync and async function calls
This commit is contained in:
+8
-2
@@ -1,6 +1,7 @@
|
|||||||
import aiofiles.os
|
import aiofiles.os
|
||||||
import aiofiles
|
import aiofiles
|
||||||
import os
|
import os
|
||||||
|
import inspect
|
||||||
|
|
||||||
async def read_file(path: str, callback=None):
|
async def read_file(path: str, callback=None):
|
||||||
if not await aiofiles.os.path.exists(path):
|
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:
|
async with aiofiles.open(path, "r") as f:
|
||||||
if callback:
|
if callback:
|
||||||
return await callback(f)
|
result = callback(f)
|
||||||
|
if inspect.isawaitable(result):
|
||||||
|
return await result
|
||||||
|
return result
|
||||||
return await f.read()
|
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:
|
async with aiofiles.open(path, "w") as f:
|
||||||
if callback:
|
if callback:
|
||||||
await callback(data, f, *args, **kwargs)
|
result = callback(data, f, *args, **kwargs)
|
||||||
|
if inspect.isawaitable(result):
|
||||||
|
await result
|
||||||
else:
|
else:
|
||||||
await f.write(data)
|
await f.write(data)
|
||||||
|
|||||||
Reference in New Issue
Block a user