From 1f035c4456fb3e1d87841acac4427a564a283650 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?F=C3=A9lix=20Pi=C3=A9dallu?= <felix@piedallu.me>
Date: Sun, 9 Jun 2024 23:22:03 +0200
Subject: [PATCH] Use a wrapper for file-backed cached functions

---
 package_linter.py | 33 +++++++++++++++++----------------
 1 file changed, 17 insertions(+), 16 deletions(-)

diff --git a/package_linter.py b/package_linter.py
index 1fd51ef..e18773f 100755
--- a/package_linter.py
+++ b/package_linter.py
@@ -310,26 +310,27 @@ def file_exists(file_path):
     return os.path.isfile(file_path) and os.stat(file_path).st_size > 0
 
 
+def cache_file(cachefile: str, ttl_s: int):
+    def cache_is_fresh():
+        return os.path.exists(cachefile) and time.time() - os.path.getmtime(cachefile) < ttl_s
+    def decorator(function):
+        def wrapper(*args, **kwargs):
+            if not cache_is_fresh():
+                with open(cachefile, "w+") as outfile:
+                    outfile.write(function(*args, **kwargs))
+            return open(cachefile).read()
+        return wrapper
+    return decorator
+
+
+@cache_file(".spdx_licenses", 3600)
 def spdx_licenses():
-    cachefile = ".spdx_licenses"
-    if os.path.exists(cachefile) and time.time() - os.path.getmtime(cachefile) < 3600:
-        return open(cachefile).read()
-
-    url = "https://spdx.org/licenses/"
-    content = urlopen(url)["content"]
-    open(cachefile, "w").write(content)
-    return content
+    return urlopen("https://spdx.org/licenses/")["content"]
 
 
+@cache_file(".manifest.v2.schema.json", 3600)
 def manifest_v2_schema():
-    cachefile = ".manifest.v2.schema.json"
-    if os.path.exists(cachefile) and time.time() - os.path.getmtime(cachefile) < 3600:
-        return json.loads(open(cachefile).read())
-
-    url = "https://raw.githubusercontent.com/YunoHost/apps/master/schemas/manifest.v2.schema.json"
-    content = urlopen(url)["content"]
-    open(cachefile, "w").write(content)
-    return json.loads(content)
+    return urlopen("https://raw.githubusercontent.com/YunoHost/apps/master/schemas/manifest.v2.schema.json")["content"]
 
 
 tests = {}