65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
"""
|
||
获取所有可用的文件ID列表（用于测试）
|
||
"""
|
||
import pymysql
|
||
import os
|
||
|
||
# 数据库连接配置
|
||
DB_CONFIG = {
|
||
'host': os.getenv('DB_HOST', '152.136.177.240'),
|
||
'port': int(os.getenv('DB_PORT', 5012)),
|
||
'user': os.getenv('DB_USER', 'finyx'),
|
||
'password': os.getenv('DB_PASSWORD', '6QsGK6MpePZDE57Z'),
|
||
'database': os.getenv('DB_NAME', 'finyx'),
|
||
'charset': 'utf8mb4'
|
||
}
|
||
|
||
TENANT_ID = 615873064429507639
|
||
|
||
def get_available_file_configs():
|
||
"""获取所有可用的文件配置"""
|
||
conn = pymysql.connect(**DB_CONFIG)
|
||
cursor = conn.cursor(pymysql.cursors.DictCursor)
|
||
|
||
try:
|
||
sql = """
|
||
SELECT id, name, file_path, state
|
||
FROM f_polic_file_config
|
||
WHERE tenant_id = %s
|
||
AND state = 1
|
||
ORDER BY name
|
||
"""
|
||
cursor.execute(sql, (TENANT_ID,))
|
||
configs = cursor.fetchall()
|
||
|
||
print("="*80)
|
||
print("可用的文件配置列表（state=1）")
|
||
print("="*80)
|
||
print(f"\n共找到 {len(configs)} 个启用的文件配置:\n")
|
||
|
||
for i, config in enumerate(configs, 1):
|
||
print(f"{i}. ID: {config['id']}")
|
||
print(f" 名称: {config['name']}")
|
||
print(f" 文件路径: {config['file_path'] or '(空)'}")
|
||
print()
|
||
|
||
# 输出JSON格式，方便复制
|
||
print("\n" + "="*80)
|
||
print("JSON格式（可用于测试）:")
|
||
print("="*80)
|
||
print("[")
|
||
for i, config in enumerate(configs):
|
||
comma = "," if i < len(configs) - 1 else ""
|
||
print(f' {{"fileId": {config["id"]}, "fileName": "{config["name"]}.doc"}}{comma}')
|
||
print("]")
|
||
|
||
return configs
|
||
|
||
finally:
|
||
cursor.close()
|
||
conn.close()
|
||
|
||
if __name__ == '__main__':
|
||
get_available_file_configs()
|
||
|