63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
from fastapi import FastAPI, Request
|
||
from fastapi.responses import HTMLResponse
|
||
from .addons.base.models.configparser import Config
|
||
from .addons.base.models.configparser import DATABASE_URL
|
||
import psycopg2
|
||
from psycopg2 import sql
|
||
from fastapi.responses import JSONResponse
|
||
import logging
|
||
|
||
from fastapi.middleware.cors import CORSMiddleware
|
||
origins = [
|
||
"*",
|
||
]
|
||
|
||
logging.basicConfig(level=logging.INFO)
|
||
logger = logging.getLogger(__name__)
|
||
app = FastAPI()
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=origins,
|
||
allow_credentials=True,
|
||
allow_methods=["*"], # 允许所有 HTTP 方法
|
||
allow_headers=["*"], # 允许所有头部
|
||
)
|
||
|
||
config = Config()
|
||
def get_databases():
|
||
conn = None
|
||
try:
|
||
# 连接到默认的PostgreSQL数据库，通常为postgres
|
||
conn = psycopg2.connect(DATABASE_URL) # 使用DATABASE_URL连接
|
||
conn.autocommit = True # 设置自动提交模式
|
||
cursor = conn.cursor()
|
||
|
||
# 查询所有数据库，排除postgres
|
||
cursor.execute(sql.SQL("SELECT datname FROM pg_database WHERE datistemplate = false AND datname != 'postgres';"))
|
||
databases = [row[0] for row in cursor.fetchall()]
|
||
logger.info("获取已有数据库列表: %s", databases)
|
||
return databases
|
||
|
||
except Exception as e:
|
||
logger.error("Error getting databases: %s", e)
|
||
return []
|
||
finally:
|
||
if conn is not None:
|
||
conn.close()
|
||
|
||
@app.get("/", response_class=JSONResponse)
|
||
async def get_databases_route():
|
||
databases = get_databases()
|
||
return databases
|
||
|
||
from .addons.base.models import database
|
||
app.include_router(database.router, prefix="/web/database", tags=["数据库"])
|
||
|
||
if __name__ == '__main__':
|
||
try:
|
||
server_port = config.getint('options', 'http_port')
|
||
import uvicorn
|
||
uvicorn.run(app, host="0.0.0.0", port=server_port)
|
||
except Exception as e:
|
||
print(f"Failed to start the server: {e}")
|