FilesExpand file tree

master

/

cache_middleware.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

53 lines (44 loc) · 1.67 KB

master

/

cache_middleware.py

Copy path

Top

File metadata and controls

• Code
• Blame

53 lines (44 loc) · 1.67 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

"""Cache control middleware for ComfyUI server"""

from aiohttp import web

from typing import Callable, Awaitable

# Time in seconds

ONE_HOUR: int = 3600

ONE_DAY: int = 86400

IMG_EXTENSIONS = (

".jpg",

".jpeg",

".png",

".ppm",

".bmp",

".pgm",

".tif",

".tiff",

".webp",

)

@web.middleware

async def cache_control(

request: web.Request, handler: Callable[[web.Request], Awaitable[web.Response]]

) -> web.Response:

"""Cache control middleware that sets appropriate cache headers based on file type and response status"""

response: web.Response = await handler(request)

path_filename = request.path.rsplit("/", 1)[-1]

is_entry_point = path_filename.startswith("index") and path_filename.endswith(

".json"

)

if request.path.endswith(".js") or request.path.endswith(".css") or is_entry_point:

response.headers.setdefault("Cache-Control", "no-store")

return response

# Early return for non-image files - no cache headers needed

if not request.path.lower().endswith(IMG_EXTENSIONS):

return response

# Handle image files

if response.status == 404:

response.headers.setdefault("Cache-Control", f"public, max-age={ONE_HOUR}")

elif response.status in (200, 201, 202, 203, 204, 205, 206, 301, 308):

# Success responses and permanent redirects - cache for 1 day

response.headers.setdefault("Cache-Control", f"public, max-age={ONE_DAY}")

elif response.status in (302, 303, 307):

# Temporary redirects - no cache

response.headers.setdefault("Cache-Control", "no-cache")

# Note: 304 Not Modified falls through - no cache headers set

return response