Merge branch 'feature/http-cache' into dev

semihyonet · semihyonet · commit a514b2d31844 · 2022-07-17T13:16:49.000+02:00
diff --git a/.idea/plusclouds-cli.iml b/.idea/plusclouds-cli.iml
diff --git a/plusclouds/gateway/http_cache.py b/plusclouds/gateway/http_cache.py
@@ -0,0 +1,24 @@
+def cache_constructor(cache_dict: dict):
+	def cache_decorator(func):
+
+		def callback(*args, **kwargs):
+			url = kwargs.get("url", "")
+			if url == "":
+				url = "_"
+
+				if len(args) > 1:
+					url = args[1]
+
+			if url in cache_dict.keys():
+				print("returning cached dir")
+				return cache_dict[url]
+
+			resp = func(*args, **kwargs)
+			cache_dict[url] = resp
+			print("new cache on {}".format(url))
+
+			return resp
+
+		return callback
+
+	return cache_decorator
diff --git a/plusclouds/gateway/http_client.py b/plusclouds/gateway/http_client.py
@@ -2,6 +2,19 @@
 
 from plusclouds.settings import plusclouds_url
 from plusclouds.util.token_util import get_token
+from plusclouds.gateway.http_cache import cache_constructor
+
+get_dict = {}
+get_cache = cache_constructor(get_dict)
+
+post_dict = {}
+post_cache = cache_constructor(post_dict)
+
+delete_dict = {}
+delete_cache = cache_constructor(delete_dict)
+
+options_dict = {}
+options_cache = cache_constructor(options_dict)
 
 
 class HttpGateway:
@@ -11,11 +24,13 @@ def __init__(self, base_url: str = plusclouds_url):
 		self.latest_request = None
 		self.option_request = None
 
+	@get_cache
 	def get(self, url: str = "") -> requests.Response:
 		full_url = plusclouds_url + url
 		self.latest_request = requests.get(full_url, headers={"Authorization": "Bearer " + self.token.strip()})
 		return self.latest_request
 
+	@post_cache
 	def post(self, url: str = "", body=None) -> requests.Response:
 		if body is None:
 			body = {}
@@ -25,6 +40,7 @@ def post(self, url: str = "", body=None) -> requests.Response:
 											headers={"Authorization": "Bearer " + self.token.strip()})
 		return self.latest_request
 
+	@options_cache
 	def options(self, url: str = "") -> requests.Response:
 		self.option_request = requests.request("OPTIONS", self.base_url + url)
 		return self.option_request
