From 47435c6128b2821b1dc50df6fc2612c2908b45c3 Mon Sep 17 00:00:00 2001
From: profitroll <vozhd.kk@gmail.com>
Date: Thu, 22 Jun 2023 13:16:12 +0200
Subject: [PATCH] Async I/O implemented

---
 extensions/pages.py  | 25 +++++++++++++++++--------
 extensions/photos.py | 11 ++++++-----
 extensions/videos.py |  9 +++++----
 3 files changed, 28 insertions(+), 17 deletions(-)

diff --git a/extensions/pages.py b/extensions/pages.py
index b712a84..b88d117 100644
--- a/extensions/pages.py
+++ b/extensions/pages.py
@@ -1,3 +1,4 @@
+import aiofiles
 from os import path
 from modules.app import app
 from fastapi.responses import HTMLResponse, Response
@@ -5,27 +6,35 @@ from fastapi.responses import HTMLResponse, Response
 
 @app.get("/pages/matter.css", include_in_schema=False)
 async def page_matter():
-    with open(path.join("pages", "matter.css"), "r", encoding="utf-8") as f:
-        output = f.read()
+    async with aiofiles.open(
+        path.join("pages", "matter.css"), "r", encoding="utf-8"
+    ) as f:
+        output = await f.read()
     return Response(content=output)
 
 
 @app.get("/pages/{page}/{file}", include_in_schema=False)
 async def page_assets(page: str, file: str):
-    with open(path.join("pages", page, file), "r", encoding="utf-8") as f:
-        output = f.read()
+    async with aiofiles.open(
+        path.join("pages", page, file), "r", encoding="utf-8"
+    ) as f:
+        output = await f.read()
     return Response(content=output)
 
 
 @app.get("/", include_in_schema=False)
 async def page_home():
-    with open(path.join("pages", "home", "index.html"), "r", encoding="utf-8") as f:
-        output = f.read()
+    async with aiofiles.open(
+        path.join("pages", "home", "index.html"), "r", encoding="utf-8"
+    ) as f:
+        output = await f.read()
     return HTMLResponse(content=output)
 
 
 @app.get("/register", include_in_schema=False)
 async def page_register():
-    with open(path.join("pages", "register", "index.html"), "r", encoding="utf-8") as f:
-        output = f.read()
+    async with aiofiles.open(
+        path.join("pages", "register", "index.html"), "r", encoding="utf-8"
+    ) as f:
+        output = await f.read()
     return HTMLResponse(content=output)
diff --git a/extensions/photos.py b/extensions/photos.py
index 3077ada..448ccad 100644
--- a/extensions/photos.py
+++ b/extensions/photos.py
@@ -1,3 +1,4 @@
+import aiofiles
 import re
 import pickle
 from secrets import token_urlsafe
@@ -130,7 +131,7 @@ async def photo_upload(
             ".".join(base_name) + f"_{int(datetime.now().timestamp())}." + extension
         )
 
-    with open(
+    async with aiofiles.open(
         path.join("data", "users", current_user.user, "albums", album, filename), "wb"
     ) as f:
         f.write(await file.read())
@@ -266,8 +267,8 @@ if configGet("media_token_access") is True:
 
         mime = Magic(mime=True).from_file(image_path)
 
-        with open(image_path, "rb") as f:
-            image_file = f.read()
+        async with aiofiles.open(image_path, "rb") as f:
+            image_file = await f.read()
 
         return Response(image_file, media_type=mime)
 
@@ -293,8 +294,8 @@ async def photo_get(
 
     mime = Magic(mime=True).from_file(image_path)
 
-    with open(image_path, "rb") as f:
-        image_file = f.read()
+    async with aiofiles.open(image_path, "rb") as f:
+        image_file = await f.read()
 
     return Response(image_file, media_type=mime)
 
diff --git a/extensions/videos.py b/extensions/videos.py
index e19d2d7..bd0402a 100644
--- a/extensions/videos.py
+++ b/extensions/videos.py
@@ -1,3 +1,4 @@
+import aiofiles
 import re
 import pickle
 from secrets import token_urlsafe
@@ -59,10 +60,10 @@ async def video_upload(
             ".".join(base_name) + f"_{int(datetime.now().timestamp())}." + extension
         )
 
-    with open(
+    async with aiofiles.open(
         path.join("data", "users", current_user.user, "albums", album, filename), "wb"
     ) as f:
-        f.write(await file.read())
+        await f.write(await file.read())
 
     # Hashing and duplicates check should be here
 
@@ -112,8 +113,8 @@ async def video_get(
 
     mime = Magic(mime=True).from_file(video_path)
 
-    with open(video_path, "rb") as f:
-        video_file = f.read()
+    with aiofiles.open(video_path, "rb") as f:
+        video_file = await f.read()
 
     return Response(video_file, media_type=mime)