import os

import mysql.connector

config = {
    'user': 'yangchaojie',
    'password': 'P@ssw0rd',
    'host': 'rm-uf67t3vw28yp475tf1o.mysql.rds.aliyuncs.com',
    'database': 'dtok'
}

try:
    # 创建连接
    conn = mysql.connector.connect(**config)
    print("连接成功")

    # 创建cursor对象
    cursor = conn.cursor()

    # 执行查询SQL语句
    cursor.execute('SELECT * FROM script')

    # 获取查询结果
    rows = cursor.fetchall()
    print(f"共{len(rows)}个脚本")
    count = 0
    for row in rows:
        print(f"当前更新 {row[2]} ..")
        is_exists = os.path.exists(f"js/{row[2]}.js")
        if is_exists:
            print(f"找到脚本{row[2]}.js")
            file = open(f"js/{row[2]}.js", "r", encoding='utf-8')  # 第二个参数为模式（"r"表示只读）
            content = file.read()  # 将文件的全部内容读入字符串变量 content
            file.close()

            cursor.execute(f"update script set script_content =  where module = '{row[2]}'")
            print(f'更新完成{row[2]}')
            count += 1
            conn.commit()
            break
        else:
            print(f"[WARN]!!! 没有找到脚本{row[2]}.js")
    print(f"共更新{count}个脚本")
except mysql.connector.Error as error:
    print("发生错误: {}".format(error))

finally:
    if conn.is_connected():
        cursor.close()
        conn.close()
        print("连接已关闭")


def get_script_names():
    files_and_directories = os.listdir('js')

    # 列出所有文件
    files = [f.split('.')[0] for f in files_and_directories if os.path.isfile(os.path.join('js', f))]

    # 打印所有文件
    return files

# get_script_names()
#
# if __name__ == '__main__':
#     file = open("js/智能养号.js", "r",encoding='utf-8')  # 第二个参数为模式（"r"表示只读）
#     content = file.read()  # 将文件的全部内容读入字符串变量 content
#     print(content)  # 输出文件内容
#     file.close()