:gem: [Feature] DataProxyAPI: cache images

.gitignore

@@ -1 +1,2 @@
-__pycache__
+__pycache__
+cache

apis/data_proxy_api.py

@@ -9,7 +9,7 @@ from tclogger import logger
 from typing import Optional
 
 from apis.arg_parser import ArgParser
-from configs.envs import DATA_PROXY_APP_ENVS
+from configs.envs import DATA_PROXY_APP_ENVS, CACHE_ROOT
 
 
 class DataProxyAPI:
@@ -39,10 +39,36 @@ class DataProxyAPI:
             allow_headers=["*"],
         )
 
+    def get_img_path_by_url(self, url: str):
+        return CACHE_ROOT / url.split("/")[-1]
+
+    def save_img(self, img_url: str, img_bytes: bytes):
+        img_path = self.get_img_path_by_url(img_url)
+        if not img_path.parent.exists():
+            img_path.parent.mkdir(parents=True, exist_ok=True)
+        with open(img_path, "wb") as img_file:
+            img_file.write(img_bytes)
+        return img_path
+
+    def get_img_bytes_from_url(self, url: str):
+        try:
+            response = requests.get(url, headers=self.REQUESTS_HEADERS)
+            img_bytes = response.content
+        except Exception as e:
+            img_bytes = None
+
+        return img_bytes
+
     def get_img(self, url: str):
-        response = requests.get(url, headers=self.REQUESTS_HEADERS)
-        image_bytes = response.content
-        return Response(content=image_bytes, media_type="image/jpeg")
+        img_path = self.get_img_path_by_url(url)
+        if img_path.exists():
+            with open(img_path, "rb") as img_file:
+                img_bytes = img_file.read()
+        else:
+            img_bytes = self.get_img_bytes_from_url(url)
+            self.save_img(url, img_bytes)
+
+        return Response(content=img_bytes, media_type="image/jpeg")
 
     def setup_routes(self):
         self.app.get(

configs/envs.json

@@ -4,5 +4,6 @@
         "host": "0.0.0.0",
         "port": 22001,
         "version": "0.1"
-    }
+    },
+    "cache_root": "cache"
 }

configs/envs.py

@@ -1,9 +1,8 @@
 from pathlib import Path
-
 from tclogger import OSEnver
 
 configs_root = Path(__file__).parents[1] / "configs"
 envs_path = configs_root / "envs.json"
-
 ENVS_ENVER = OSEnver(envs_path)
 DATA_PROXY_APP_ENVS = ENVS_ENVER["data_proxy_app"]
+CACHE_ROOT = Path(__file__).parents[1] / ENVS_ENVER["cache_root"]