validops-east-1 commited on
Commit
8af1beb
·
1 Parent(s): 5a01a63
Files changed (2) hide show
  1. app/api/server.py +4 -0
  2. 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)