129 lines
4.7 KiB
Python
129 lines
4.7 KiB
Python
"""
|
|
Logging configuration for ALwrity backend.
|
|
Provides clean logging setup for end users vs developers.
|
|
"""
|
|
|
|
import logging
|
|
import os
|
|
import sys
|
|
from loguru import logger
|
|
|
|
|
|
def setup_clean_logging():
|
|
"""Set up clean logging for end users."""
|
|
verbose_mode = os.getenv("ALWRITY_VERBOSE", "false").lower() == "true"
|
|
|
|
# Always remove all existing handlers first to prevent conflicts
|
|
logger.remove()
|
|
|
|
if not verbose_mode:
|
|
# Suppress verbose logging for end users - be more aggressive
|
|
logging.getLogger('sqlalchemy.engine').setLevel(logging.CRITICAL)
|
|
logging.getLogger('sqlalchemy.pool').setLevel(logging.CRITICAL)
|
|
logging.getLogger('sqlalchemy.dialects').setLevel(logging.CRITICAL)
|
|
logging.getLogger('sqlalchemy.orm').setLevel(logging.CRITICAL)
|
|
logging.getLogger('sqlalchemy').setLevel(logging.CRITICAL)
|
|
logging.getLogger('sqlalchemy.engine.Engine').setLevel(logging.CRITICAL)
|
|
|
|
# Suppress service initialization logs
|
|
logging.getLogger('services').setLevel(logging.WARNING)
|
|
logging.getLogger('api').setLevel(logging.WARNING)
|
|
logging.getLogger('models').setLevel(logging.WARNING)
|
|
|
|
# Suppress specific noisy loggers
|
|
noisy_loggers = [
|
|
'linkedin_persona_service',
|
|
'facebook_persona_service',
|
|
'core_persona_service',
|
|
'persona_analysis_service',
|
|
'ai_service_manager',
|
|
'ai_engine_service',
|
|
'website_analyzer',
|
|
'competitor_analyzer',
|
|
'keyword_researcher',
|
|
'content_gap_analyzer',
|
|
'onboarding_data_service',
|
|
'comprehensive_user_data',
|
|
'strategy_data',
|
|
'gap_analysis_data',
|
|
'phase1_steps',
|
|
'daily_schedule_generator',
|
|
'gsc_service',
|
|
'wordpress_oauth',
|
|
'data_filter',
|
|
'source_mapper',
|
|
'grounding_engine',
|
|
'blog_content_seo_analyzer',
|
|
'linkedin_service',
|
|
'citation_manager',
|
|
'content_analyzer',
|
|
'linkedin_prompt_generator',
|
|
'linkedin_image_storage',
|
|
'hallucination_detector',
|
|
'writing_assistant',
|
|
'onboarding_data_service',
|
|
'enhanced_linguistic_analyzer',
|
|
'persona_quality_improver',
|
|
'logging_middleware',
|
|
'exa_service',
|
|
'step3_research_service',
|
|
'sitemap_service',
|
|
'linkedin_image_generator',
|
|
'linkedin_prompt_generator',
|
|
'linkedin_image_storage',
|
|
'router_manager',
|
|
'frontend_serving',
|
|
'database',
|
|
'user_business_info',
|
|
'auth_middleware',
|
|
'pricing_service',
|
|
'create_billing_tables'
|
|
]
|
|
|
|
for logger_name in noisy_loggers:
|
|
logging.getLogger(logger_name).setLevel(logging.WARNING)
|
|
|
|
# Configure loguru to be less verbose (only show warnings and errors)
|
|
def warning_only_filter(record):
|
|
return record["level"].name in ["WARNING", "ERROR", "CRITICAL"]
|
|
|
|
logger.add(
|
|
sys.stdout.write,
|
|
level="WARNING",
|
|
format="{time:HH:mm:ss} | {level: <8} | {name}:{function}:{line} - {message}\n",
|
|
filter=warning_only_filter
|
|
)
|
|
# Add a focused sink to surface Story Video Generation INFO logs in console
|
|
def video_generation_filter(record):
|
|
msg = record.get("message", "")
|
|
name = record.get("name", "")
|
|
service = record.get("extra", {}).get("service")
|
|
return (
|
|
"[StoryVideoGeneration]" in msg
|
|
or "services.story_writer.video_generation_service" in name
|
|
or "[video_gen]" in msg
|
|
or service == "video_generation_service"
|
|
or "services.llm_providers.main_video_generation" in name
|
|
)
|
|
logger.add(
|
|
sys.stdout.write,
|
|
level="INFO",
|
|
format="{time:HH:mm:ss} | {level: <8} | {name}:{function}:{line} - {message}\n",
|
|
filter=video_generation_filter
|
|
)
|
|
else:
|
|
# In verbose mode, show all log levels with detailed formatting
|
|
logger.add(
|
|
sys.stdout.write,
|
|
level="DEBUG",
|
|
format="{time:HH:mm:ss} | {level: <8} | {name}:{function}:{line} - {message}\n"
|
|
)
|
|
|
|
return verbose_mode
|
|
|
|
|
|
def get_uvicorn_log_level():
|
|
"""Get appropriate uvicorn log level based on verbose mode."""
|
|
verbose_mode = os.getenv("ALWRITY_VERBOSE", "false").lower() == "true"
|
|
return "debug" if verbose_mode else "warning"
|