summaryrefslogtreecommitdiff
path: root/ATRI/utils/file.py
diff options
context:
space:
mode:
Diffstat (limited to 'ATRI/utils/file.py')
-rw-r--r--ATRI/utils/file.py27
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!")