Spaces:
Running
Running
Commit ·
8af1beb
1
Parent(s): 5a01a63
okay
Browse files- app/api/server.py +4 -0
- app/core/banner.py +24 -0
app/api/server.py
CHANGED
|
@@ -11,6 +11,7 @@ from app.api.v1.router import api_v1_router
|
|
| 11 |
from app.api.v1.system import is_maintenance
|
| 12 |
from app.config import get_settings
|
| 13 |
from app.core.auth.deps import init_auth_db
|
|
|
|
| 14 |
from app.core.database import pool_manager
|
| 15 |
from app.core.logger import get_logger
|
| 16 |
from app.core.redis_client import close_redis, create_redis_client
|
|
@@ -44,6 +45,9 @@ async def _self_ping():
|
|
| 44 |
|
| 45 |
@asynccontextmanager
|
| 46 |
async def lifespan(app: FastAPI):
|
|
|
|
|
|
|
|
|
|
| 47 |
if not _settings.tidb_instance_list:
|
| 48 |
_logger.error(
|
| 49 |
"No TiDB instances configured! Set TIDB_INSTANCES environment variable "
|
|
|
|
| 11 |
from app.api.v1.system import is_maintenance
|
| 12 |
from app.config import get_settings
|
| 13 |
from app.core.auth.deps import init_auth_db
|
| 14 |
+
from app.core.banner import get_banner
|
| 15 |
from app.core.database import pool_manager
|
| 16 |
from app.core.logger import get_logger
|
| 17 |
from app.core.redis_client import close_redis, create_redis_client
|
|
|
|
| 45 |
|
| 46 |
@asynccontextmanager
|
| 47 |
async def lifespan(app: FastAPI):
|
| 48 |
+
print(get_banner())
|
| 49 |
+
print("=" * 60)
|
| 50 |
+
print()
|
| 51 |
if not _settings.tidb_instance_list:
|
| 52 |
_logger.error(
|
| 53 |
"No TiDB instances configured! Set TIDB_INSTANCES environment variable "
|
app/core/banner.py
ADDED
|
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from __future__ import annotations
|
| 2 |
+
|
| 3 |
+
import pyfiglet
|
| 4 |
+
|
| 5 |
+
from app.config import get_settings
|
| 6 |
+
|
| 7 |
+
|
| 8 |
+
def get_banner() -> str:
|
| 9 |
+
settings = get_settings()
|
| 10 |
+
banner = pyfiglet.figlet_format("AgentDeck", font="slant")
|
| 11 |
+
lines = [
|
| 12 |
+
banner,
|
| 13 |
+
f" {settings.app_name} v{settings.app_version}",
|
| 14 |
+
f" Environment: {settings.environment or 'production'}",
|
| 15 |
+
]
|
| 16 |
+
if settings.tidb_instance_list:
|
| 17 |
+
lines.append(f" TiDB instances: {len(settings.tidb_instance_list)}")
|
| 18 |
+
else:
|
| 19 |
+
lines.append(" TiDB: not configured")
|
| 20 |
+
if settings.redis_url:
|
| 21 |
+
lines.append(" Redis: connected")
|
| 22 |
+
else:
|
| 23 |
+
lines.append(" Redis: not configured (degraded mode)")
|
| 24 |
+
return "\n".join(lines)
|