dee4387b0b
- Add ALWRITY_FEATURE_PROFILE env var (precedence over ALWRITY_ROUTER_PROFILE) - Add OPTIONAL_MODULE_MATRIX defining 'all' and 'default' profiles - Add get_feature_profile_status() to RouterManager - Add GET /api/feature-profile/status endpoint in main.py and app.py - Returns active profile and enabled optional modules
173 lines
10 KiB
Python
173 lines
10 KiB
Python
"""
|
|
Router Manager Module
|
|
Handles FastAPI router inclusion and management.
|
|
"""
|
|
|
|
from importlib import import_module
|
|
from typing import Any, Dict, List, Optional
|
|
|
|
import os
|
|
|
|
from fastapi import FastAPI
|
|
from loguru import logger
|
|
|
|
|
|
CORE_ROUTER_REGISTRY = [
|
|
{"name": "component_logic", "module": "api.component_logic", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "subscription", "module": "api.subscription", "attr": "router", "profiles": {"all", "default", "podcast"}},
|
|
{"name": "step3_research", "module": "api.onboarding_utils.step3_routes", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "step4_assets", "module": "api.onboarding_utils.step4_asset_routes", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "step4_persona", "module": "api.onboarding_utils.step4_persona_routes_optimized", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "gsc_auth", "module": "routers.gsc_auth", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "wordpress_oauth", "module": "routers.wordpress_oauth", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "bing_oauth", "module": "routers.bing_oauth", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "bing_analytics", "module": "routers.bing_analytics", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "bing_analytics_storage", "module": "routers.bing_analytics_storage", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "seo_tools", "module": "routers.seo_tools", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "facebook_writer", "module": "api.facebook_writer.routers", "attr": "facebook_router", "profiles": {"all", "default"}},
|
|
{"name": "linkedin", "module": "routers.linkedin", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "linkedin_image", "module": "api.linkedin_image_generation", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "brainstorm", "module": "api.brainstorm", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "hallucination_detector", "module": "api.hallucination_detector", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "writing_assistant", "module": "api.writing_assistant", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "content_planning", "module": "api.content_planning.api.router", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "user_data", "module": "api.user_data", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "user_environment", "module": "api.user_environment", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "strategy_copilot", "module": "api.content_planning.strategy_copilot", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "error_logging", "module": "routers.error_logging", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "frontend_env_manager", "module": "routers.frontend_env_manager", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "platform_analytics", "module": "routers.platform_analytics", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "bing_insights", "module": "routers.bing_insights", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "background_jobs", "module": "routers.background_jobs", "attr": "router", "profiles": {"all", "default"}},
|
|
]
|
|
|
|
OPTIONAL_ROUTER_REGISTRY = [
|
|
{"name": "blog_writer", "module": "api.blog_writer.router", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "story_writer", "module": "api.story_writer.router", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "wix", "module": "api.wix_routes", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "blog_seo_analysis", "module": "api.blog_writer.seo_analysis", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "persona", "module": "api.persona_routes", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "video_studio", "module": "api.video_studio.router", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "stability", "module": "routers.stability", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "stability_advanced", "module": "routers.stability_advanced", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "stability_admin", "module": "routers.stability_admin", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "images", "module": "api.images", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "image_studio", "module": "routers.image_studio", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "product_marketing", "module": "routers.product_marketing", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "campaign_creator", "module": "routers.campaign_creator", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "content_assets", "module": "api.content_assets.router", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "podcast", "module": "api.podcast.router", "attr": "router", "profiles": {"all", "default", "podcast"}},
|
|
{"name": "youtube", "module": "api.youtube.router", "attr": "router", "profiles": {"all", "default"}, "include_kwargs": {"prefix": "/api"}},
|
|
{"name": "research_config", "module": "api.research_config", "attr": "router", "profiles": {"all", "default"}, "include_kwargs": {"prefix": "/api/research", "tags": ["research"]}},
|
|
{"name": "research_engine", "module": "api.research.router", "attr": "router", "profiles": {"all", "default"}, "include_kwargs": {"tags": ["Research Engine"]}},
|
|
{"name": "scheduler_dashboard", "module": "api.scheduler_dashboard", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "oauth_token_monitoring", "module": "api.oauth_token_monitoring_routes", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "agents", "module": "api.agents_api", "attr": "router", "profiles": {"all", "default"}},
|
|
{"name": "today_workflow", "module": "api.today_workflow", "attr": "router", "profiles": {"all", "default"}},
|
|
]
|
|
|
|
OPTIONAL_MODULE_MATRIX = {
|
|
"all": [entry["name"] for entry in OPTIONAL_ROUTER_REGISTRY],
|
|
"default": [entry["name"] for entry in OPTIONAL_ROUTER_REGISTRY],
|
|
}
|
|
|
|
|
|
class RouterManager:
|
|
"""Manages FastAPI router inclusion and organization."""
|
|
|
|
def __init__(self, app: FastAPI):
|
|
self.app = app
|
|
self.included_routers = []
|
|
self.failed_routers = []
|
|
|
|
def _is_verbose(self) -> bool:
|
|
return os.getenv("ALWRITY_VERBOSE", "false").lower() == "true"
|
|
|
|
def _get_profile(self) -> str:
|
|
return os.getenv("ALWRITY_FEATURE_PROFILE", os.getenv("ALWRITY_ROUTER_PROFILE", os.getenv("ALWRITY_FEATURE_TO_ENABLE", "all"))).strip().lower() or "all"
|
|
|
|
def _should_include_router(self, registry_entry: Dict[str, Any], profile: str) -> bool:
|
|
profiles = registry_entry.get("profiles", {"all", "default"})
|
|
return profile in profiles or profile in {"all", "default"}
|
|
|
|
def _load_router_from_registry(self, registry_entry: Dict[str, Any]):
|
|
module = import_module(registry_entry["module"])
|
|
return getattr(module, registry_entry["attr"])
|
|
|
|
def include_router_safely(self, router, router_name: Optional[str] = None, include_kwargs: Optional[Dict[str, Any]] = None) -> bool:
|
|
"""Include a router safely with error handling."""
|
|
verbose = self._is_verbose()
|
|
router_name = router_name or getattr(router, 'prefix', 'unknown')
|
|
|
|
try:
|
|
self.app.include_router(router, **(include_kwargs or {}))
|
|
self.included_routers.append(router_name)
|
|
if verbose:
|
|
logger.info(f"✅ Router included successfully: {router_name}")
|
|
return True
|
|
except Exception as e:
|
|
router_name = router_name or 'unknown'
|
|
self.failed_routers.append({"name": router_name, "error": str(e)})
|
|
if verbose:
|
|
logger.warning(f"❌ Router inclusion failed: {router_name} - {e}")
|
|
return False
|
|
|
|
@staticmethod
|
|
def _demo_release_mode_enabled() -> bool:
|
|
"""Return True when demo-release safety mode is enabled."""
|
|
return os.getenv("ALWRITY_DEMO_RELEASE", "false").lower() in {"1", "true", "yes", "on"}
|
|
|
|
def _include_registry_group(self, registry: List[Dict[str, Any]], group_name: str) -> bool:
|
|
verbose = self._is_verbose()
|
|
profile = self._get_profile()
|
|
|
|
try:
|
|
if verbose:
|
|
logger.info(f"Including {group_name} routers for profile '{profile}'...")
|
|
|
|
for entry in registry:
|
|
if not self._should_include_router(entry, profile):
|
|
continue
|
|
|
|
try:
|
|
router = self._load_router_from_registry(entry)
|
|
self.include_router_safely(router, entry["name"], entry.get("include_kwargs"))
|
|
except Exception as e:
|
|
logger.warning(f"{entry['name']} router not mounted: {e}")
|
|
|
|
logger.info(f"✅ {group_name.capitalize()} routers processed for profile '{profile}'")
|
|
return True
|
|
|
|
except Exception as e:
|
|
logger.error(f"❌ Error including {group_name} routers: {e}")
|
|
return False
|
|
|
|
def include_core_routers(self) -> bool:
|
|
"""Include core application routers."""
|
|
return self._include_registry_group(CORE_ROUTER_REGISTRY, "core")
|
|
|
|
def include_optional_routers(self) -> bool:
|
|
"""Include optional routers with error handling."""
|
|
return self._include_registry_group(OPTIONAL_ROUTER_REGISTRY, "optional")
|
|
|
|
def get_router_status(self) -> Dict[str, Any]:
|
|
"""Get the status of router inclusion."""
|
|
return {
|
|
"active_profile": self._get_profile(),
|
|
"included_routers": self.included_routers,
|
|
"failed_routers": self.failed_routers,
|
|
"total_included": len(self.included_routers),
|
|
"total_failed": len(self.failed_routers)
|
|
}
|
|
|
|
def get_feature_profile_status(self) -> Dict[str, Any]:
|
|
"""Get feature profile status and enabled modules."""
|
|
profile = self._get_profile()
|
|
enabled_modules = OPTIONAL_MODULE_MATRIX.get(profile, OPTIONAL_MODULE_MATRIX.get("all", []))
|
|
|
|
return {
|
|
"active_profile": profile,
|
|
"enabled_modules": enabled_modules,
|
|
"available_profiles": list(OPTIONAL_MODULE_MATRIX.keys())
|
|
}
|