"""
获取所有可用的文件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()