mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-24 22:52:12 -03:00
refactor: Enhance logging configuration by adding a filter for non-critical connection reset errors
This commit is contained in:
@@ -32,7 +32,22 @@ class LoraManager:
|
|||||||
|
|
||||||
# Configure aiohttp access logger to be less verbose
|
# Configure aiohttp access logger to be less verbose
|
||||||
logging.getLogger('aiohttp.access').setLevel(logging.WARNING)
|
logging.getLogger('aiohttp.access').setLevel(logging.WARNING)
|
||||||
logging.getLogger("asyncio").setLevel(logging.WARNING)
|
|
||||||
|
# Add specific suppression for connection reset errors
|
||||||
|
class ConnectionResetFilter(logging.Filter):
|
||||||
|
def filter(self, record):
|
||||||
|
# Filter out connection reset errors that are not critical
|
||||||
|
if "ConnectionResetError" in str(record.getMessage()):
|
||||||
|
return False
|
||||||
|
if "_call_connection_lost" in str(record.getMessage()):
|
||||||
|
return False
|
||||||
|
if "WinError 10054" in str(record.getMessage()):
|
||||||
|
return False
|
||||||
|
return True
|
||||||
|
|
||||||
|
# Apply the filter to asyncio logger
|
||||||
|
asyncio_logger = logging.getLogger("asyncio")
|
||||||
|
asyncio_logger.addFilter(ConnectionResetFilter())
|
||||||
|
|
||||||
added_targets = set() # Track already added target paths
|
added_targets = set() # Track already added target paths
|
||||||
|
|
||||||
@@ -141,10 +156,6 @@ class LoraManager:
|
|||||||
async def _initialize_services(cls):
|
async def _initialize_services(cls):
|
||||||
"""Initialize all services using the ServiceRegistry"""
|
"""Initialize all services using the ServiceRegistry"""
|
||||||
try:
|
try:
|
||||||
# Ensure aiohttp access logger is configured with reduced verbosity
|
|
||||||
logging.getLogger('aiohttp.access').setLevel(logging.WARNING)
|
|
||||||
logging.getLogger("asyncio").setLevel(logging.WARNING)
|
|
||||||
|
|
||||||
# Initialize CivitaiClient first to ensure it's ready for other services
|
# Initialize CivitaiClient first to ensure it's ready for other services
|
||||||
await ServiceRegistry.get_civitai_client()
|
await ServiceRegistry.get_civitai_client()
|
||||||
|
|
||||||
|
|||||||
@@ -105,7 +105,22 @@ logger = logging.getLogger("lora-manager-standalone")
|
|||||||
|
|
||||||
# Configure aiohttp access logger to be less verbose
|
# Configure aiohttp access logger to be less verbose
|
||||||
logging.getLogger('aiohttp.access').setLevel(logging.WARNING)
|
logging.getLogger('aiohttp.access').setLevel(logging.WARNING)
|
||||||
logging.getLogger("asyncio").setLevel(logging.WARNING)
|
|
||||||
|
# Add specific suppression for connection reset errors
|
||||||
|
class ConnectionResetFilter(logging.Filter):
|
||||||
|
def filter(self, record):
|
||||||
|
# Filter out connection reset errors that are not critical
|
||||||
|
if "ConnectionResetError" in str(record.getMessage()):
|
||||||
|
return False
|
||||||
|
if "_call_connection_lost" in str(record.getMessage()):
|
||||||
|
return False
|
||||||
|
if "WinError 10054" in str(record.getMessage()):
|
||||||
|
return False
|
||||||
|
return True
|
||||||
|
|
||||||
|
# Apply the filter to asyncio logger
|
||||||
|
asyncio_logger = logging.getLogger("asyncio")
|
||||||
|
asyncio_logger.addFilter(ConnectionResetFilter())
|
||||||
|
|
||||||
# Now we can import the global config from our local modules
|
# Now we can import the global config from our local modules
|
||||||
from py.config import config
|
from py.config import config
|
||||||
@@ -126,7 +141,6 @@ class StandaloneServer:
|
|||||||
async def _configure_access_logger(self, app):
|
async def _configure_access_logger(self, app):
|
||||||
"""Configure access logger to reduce verbosity"""
|
"""Configure access logger to reduce verbosity"""
|
||||||
logging.getLogger('aiohttp.access').setLevel(logging.WARNING)
|
logging.getLogger('aiohttp.access').setLevel(logging.WARNING)
|
||||||
logging.getLogger("asyncio").setLevel(logging.WARNING)
|
|
||||||
|
|
||||||
# If using aiohttp>=3.8.0, configure access logger through app directly
|
# If using aiohttp>=3.8.0, configure access logger through app directly
|
||||||
if hasattr(app, 'access_logger'):
|
if hasattr(app, 'access_logger'):
|
||||||
@@ -223,7 +237,6 @@ class StandaloneLoraManager(LoraManager):
|
|||||||
|
|
||||||
# Configure aiohttp access logger to be less verbose
|
# Configure aiohttp access logger to be less verbose
|
||||||
logging.getLogger('aiohttp.access').setLevel(logging.WARNING)
|
logging.getLogger('aiohttp.access').setLevel(logging.WARNING)
|
||||||
logging.getLogger("asyncio").setLevel(logging.WARNING)
|
|
||||||
|
|
||||||
added_targets = set() # Track already added target paths
|
added_targets = set() # Track already added target paths
|
||||||
|
|
||||||
@@ -376,7 +389,6 @@ async def main():
|
|||||||
|
|
||||||
# Explicitly configure aiohttp access logger regardless of selected log level
|
# Explicitly configure aiohttp access logger regardless of selected log level
|
||||||
logging.getLogger('aiohttp.access').setLevel(logging.WARNING)
|
logging.getLogger('aiohttp.access').setLevel(logging.WARNING)
|
||||||
logging.getLogger("asyncio").setLevel(logging.WARNING)
|
|
||||||
|
|
||||||
# Create the server instance
|
# Create the server instance
|
||||||
server = StandaloneServer()
|
server = StandaloneServer()
|
||||||
|
|||||||
Reference in New Issue
Block a user