feat:新增投顾agent和nl2sqlagent
This commit is contained in:
@@ -0,0 +1,46 @@
|
||||
"""只读检查可用于 NL2SQL 真实验收的员工权限配置。"""
|
||||
from __future__ import annotations
|
||||
|
||||
import asyncio
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
|
||||
|
||||
from sqlalchemy import text
|
||||
|
||||
from config import database
|
||||
from config.database.mysql import get_session_factory
|
||||
|
||||
|
||||
async def main() -> None:
|
||||
queries = {
|
||||
"employees": text(
|
||||
"SELECT id, employee_role, status FROM sys_user "
|
||||
"WHERE user_type = 'EMPLOYEE' ORDER BY id LIMIT 50"
|
||||
),
|
||||
"roles": text(
|
||||
"SELECT id, employee_role, can_query, max_rows, daily_quota, status "
|
||||
"FROM nl2sql_query_role ORDER BY id"
|
||||
),
|
||||
"table_permissions": text(
|
||||
"SELECT role_id, table_name, permission, status "
|
||||
"FROM nl2sql_role_table_permission ORDER BY role_id, table_name LIMIT 200"
|
||||
),
|
||||
"column_permission_counts": text(
|
||||
"SELECT role_id, table_name, COUNT(*) AS column_count "
|
||||
"FROM nl2sql_role_column_permission WHERE status = 'active' "
|
||||
"GROUP BY role_id, table_name ORDER BY role_id, table_name"
|
||||
),
|
||||
}
|
||||
try:
|
||||
async with get_session_factory()() as session:
|
||||
for name, query in queries.items():
|
||||
rows = [dict(row) for row in (await session.execute(query)).mappings()]
|
||||
print(f"{name}: {rows}")
|
||||
finally:
|
||||
await database.mysql.dispose()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(main())
|
||||
Reference in New Issue
Block a user