feat: set cache to 5 minutes

r3tr0ananas · Jul 12, 2024 · 15d7eb0 · 15d7eb0
1 parent 9f62806
commit 15d7eb0
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/api/__init__.py b/api/__init__.py
@@ -1 +1 @@
-__version__ = "1.1.3"
+__version__ = "1.1.4"
diff --git a/api/image.py b/api/image.py
@@ -78,7 +78,7 @@ def to_dict(self) -> ImageData:
             "sources": self.sources
         }
 
-    def to_file_response(self, raw: bool = False) -> FileResponse:
+    def to_file_response(self, raw: bool = False, expire: str = "300") -> FileResponse:
         """Returns file response object."""
         if raw is False:
             downscaled_image = self.downscale_image()
@@ -87,15 +87,15 @@ def to_file_response(self, raw: bool = False) -> FileResponse:
                 downscaled_image,
                 media_type="image/webp",
                 headers = {
-                    "Expires": "0",
+                    "Expires": expire,
                     "x-image-id": self.id
                 }
             )
 
         return FileResponse(
             self.path,
             headers = {
-                "Expires": "0",
+                "Expires": expire,
                 "x-image-id": self.id
             }   
         )
diff --git a/api/main.py b/api/main.py
@@ -187,11 +187,11 @@ async def get_metadata(id: str):
         }
     },
 )
-@limiter.limit(f"{RATE_LIMIT}/second")  
+@limiter.limit(f"{RATE_LIMIT}/second")
 async def random_image(request: Request, category: str = None, raw: bool = False):
     image = agac.get_random(category)
 
-    return image.to_file_response(raw)
+    return image.to_file_response(raw, expire="0")
 
 @app.get(
     "/search",