47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
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)
|