diff options
Diffstat (limited to 'ATRI/utils/file.py')
-rw-r--r-- | ATRI/utils/file.py | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/ATRI/utils/file.py b/ATRI/utils/file.py new file mode 100644 index 0000000..4699f5d --- /dev/null +++ b/ATRI/utils/file.py @@ -0,0 +1,27 @@ +import aiofiles +from pathlib import Path + +from ATRI.exceptions import WriteError + + +async def write_file(path: Path, text, encoding='utf-8') -> None: + try: + async with aiofiles.open(path, 'w', encoding=encoding) as target: + await target.write(text) + except WriteError: + raise WriteError("Writing file failed!") + + +async def open_file(path: Path, method, encoding='utf-8'): + try: + async with aiofiles.open(path, 'r', encoding=encoding) as target: + if method == "read": + return target.read() + elif method == "readlines": + return await target.readlines() + elif method == "readline": + return await target.readline() + else: + return target.readable() + except EOFError: + raise EOFError("File not fond!") |