import os
import json
import pymysql
import logging
import logging.handlers
from database.database import *
if __name__ == '__main__':
conn = pymysql.connect(host=db_host, user=db_user, passwd=db_passwd, db=db_name, port=db_port, charset='utf8')
cursor = pymysql.cursors.SSCursor(conn)
cursor.execute("SELECT user_id, share_uid FROM anchor")
anchor_dict = {}
num = 0
while True:
num = num + 1
row = cursor.fetchone()
print(num, row)
try:
anchor_id = row[0]
word = row[1]
count = row[2]
weight = row[3]
if (weight > 0) and (num > 0):
if anchor_id in anchor_dict:
list1 = [word, count]
anchor_dict[anchor_id].append(list1)
else:
anchor_dict[anchor_id] = []
if (row == None) or (row == "None") or (num > 11253900):
break
except:
if (row == None) or (row == "None") or (num > 11253900):
break
pass
# 固化人物信息
with open("anchor_new.json", "w", encoding="utf-8") as f:
json.dump(anchor_dict, f)
os._exit(0)
cursor.close()
conn.close()