# 根据sql命令的读取的clip_id列表
# 查找txt文件中对应clip_id的全内容
# 写入一个新的txt文件
import json
import pymysql
def get_clip_id():
conn = pymysql.connect(
host="ip",
port=port,
database='db_name',
charset='utf8',
user='db_user',
passwd='XXX'
)
sql = "SELECT XXX FROM YYY WHERE ZZZ"
cursor = conn.cursor()
cursor.execute(sql)
return cursor.fetchall()
def read_txt():
f2 = open("file1.txt",'w')
f = open("file2",'r',encoding="utf-8")
dict_clip = []
for line in f:
line.strip('\n')
a = json.loads(line)
dict_clip.append(a)
clip_id_list = get_clip_id()
print(len(clip_id_list))
for line in clip_id_list:
clip_id = line[0]
for item in dict_clip:
if item["clip_id"] == clip_id:
print(item)
js = json.dumps(item)
f2.writelines(js)
f2.write("\n")
if __name__ == '__main__':
# clip_id_list = get_clip_id()
# print(clip_id)
read_txt()