mirror of
https://git.datalinker.icu/ltdrdata/ComfyUI-Manager
synced 2025-12-15 17:14:35 +08:00
added: middleware manager for security policy
This commit is contained in:
parent
043176168d
commit
adbeb527d6
@ -1,5 +1,6 @@
|
|||||||
import os
|
import os
|
||||||
import logging
|
import logging
|
||||||
|
from aiohttp import web
|
||||||
|
|
||||||
def prestartup():
|
def prestartup():
|
||||||
from . import prestartup_script # noqa: F401
|
from . import prestartup_script # noqa: F401
|
||||||
@ -43,3 +44,30 @@ def should_be_disabled(fullpath:str) -> bool:
|
|||||||
return True
|
return True
|
||||||
|
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
def get_client_ip(request):
|
||||||
|
peername = request.transport.get_extra_info("peername")
|
||||||
|
if peername is not None:
|
||||||
|
host, port = peername
|
||||||
|
return host
|
||||||
|
|
||||||
|
return "unknown"
|
||||||
|
|
||||||
|
|
||||||
|
def create_middleware():
|
||||||
|
connected_clients = set()
|
||||||
|
|
||||||
|
@web.middleware
|
||||||
|
async def manager_middleware(request: web.Request, handler):
|
||||||
|
nonlocal connected_clients
|
||||||
|
client_ip = get_client_ip(request)
|
||||||
|
connected_clients.add(client_ip)
|
||||||
|
|
||||||
|
handler_path = f"{handler.__module__}.{handler.__name__}"
|
||||||
|
|
||||||
|
response: web.Response = await handler(request)
|
||||||
|
return response
|
||||||
|
|
||||||
|
return manager_middleware
|
||||||
|
|
||||||
Loading…
x
Reference in New Issue
Block a user