import redis
import utils
import mysql.connector
from config import * 

def insert_measure_vector(conn,cursor):
    
    redis_client = redis.Redis(host=REDIS_HOST, port=REDIS_PORT, password=REDIS_PASSWORD, db=6)
    # redis_client = redis.Redis(host=REDIS_HOST, port=REDIS_PORT, db=6)
    # 执行SQL语句，更新数据
    select_query = '''
                SELECT ori_measure_id,ori_measure_name,measure_name FROM measure_config_half_year where year='2025'
                '''
    # select_query = '''
    #                 SELECT ori_measure_id,ori_measure_name,measure_name FROM measure_config where year='2024'
    #                '''
    cursor.execute(select_query)
    records = cursor.fetchall()
    print(records[:8])
    #return 
    index = 1
    for record in records:
        if redis_client.hexists('measure_config_new', record[0]):
            measure_vector = redis_client.hget('measure_config_new', record[0])
        else:
            print('新增指标',record[1])
            vector = utils.embed_with_str_local(record[1])
            measure_vector = str(vector)
        #print(f'新增指标{index} 对应归一化指标为{record[2]}',record[1])
        #index += 1
        #vector = utils.embed_with_str_local(record[1])
        #measure_vector = str(vector)
        redis_client.hset('measure_config_new', record[0], measure_vector)
    redis_client.close()
    conn.close()

if __name__ == "__main__":
    conn = mysql.connector.connect(
        host=MYSQL_HOST,
        user=MYSQL_USER,
        password=MYSQL_PASSWORD,
        database="financial_report_prod"
    )
    cursor = conn.cursor()
    
    insert_measure_vector(conn,cursor)