update

utils/func_from_es_get_article.py

@@ -354,23 +354,23 @@ def get_user_post_from_mysql(sql):
     tags_v3_tractate_dict = {}
     sql_res = con_sql_doris_prod(sql)
     for card_id in sql_res:
-        print(type(card_id),card_id)
-        es_res = get_diary_from_mysql(card_id,id_type='diary')
-        second_demands_tractate_dict[card_id] = es_res["_source"]["second_demands"]
-        tags_v3_tractate_dict[card_id] = es_res["_source"]["tags_v3"]
+        # print(type(card_id[0]),card_id)
+        es_res = get_diary_from_mysql(card_id[0],id_type='diary')
+        second_demands_tractate_dict[card_id[0]] = es_res["_source"]["second_demands"]
+        tags_v3_tractate_dict[card_id[0]] = es_res["_source"]["tags_v3"]
 
         for position in es_res["_source"]["second_demands"]:
             if position in second_demands_count_dict:
                 second_demands_count_dict[position] += 1
             else:
                 second_demands_count_dict[position] = 1
-                second_demands_card_id_list.append(int(card_id))
+                second_demands_card_id_list.append(int(card_id[0]))
         for tag in es_res["_source"]["tags_v3"]:
             if tag in tags_v3_count_dict:
                 tags_v3_count_dict[tag] += 1
             else:
                 tags_v3_count_dict[tag] = 1
-                tags_v3_card_id_list.append(int(card_id))
+                tags_v3_card_id_list.append(int(card_id[0]))
     # print(card_id_list)
     return second_demands_count_dict, tags_v3_count_dict, second_demands_card_id_list,tags_v3_card_id_list,second_demands_tractate_dict,tags_v3_tractate_dict
 
@@ -386,7 +386,7 @@ def get_diary_from_mysql(sql,id_type='tractate'):
     sql_res = con_sql_doris_prod(sql)
     for card_id in sql_res:
         # print(type(card_id[0]),card_id)
-        es_res = get_tractate_tags_from_es(card_id[0],id_type='diary')
+        es_res = get_tractate_tags_from_es(card_id[0],id_type=id_type)
         second_demands_tractate_dict[card_id[0]] = es_res["_source"]["second_demands"]
         tags_v3_tractate_dict[card_id[0]] = es_res["_source"]["tags_v3"]