update

utils/portary_div_exposure.py

@@ -363,10 +363,9 @@ def from_id_get_tag(card_id_dict):
     return query_count
 
 
-def save_data_to_csv(all_tags,user_portrait_dict,word_count_exposure):
+def save_data_to_csv(user_portrait_dict,word_count_exposure):
     all_data = []
-
-    for tag in all_tags:
+    for tag in user_portrait_dict:
         data_type = ""
         data_count = ""
         diary_exposure = 0
@@ -375,12 +374,12 @@ def save_data_to_csv(all_tags,user_portrait_dict,word_count_exposure):
         user_portrait = user_portrait_dict.get(tag)
         if user_portrait:
             data_type, data_count = user_portrait
-        if word_count_exposure["diary"].get("tag"):
-            diary_exposure = word_count_exposure["diary"].get("tag")
-        if word_count_exposure["answer"].get("tag"):
-            answer_exposure = word_count_exposure["answer"].get("tag")
-        if word_count_exposure["tractate"].get("tag"):
-            tractate_exposure = word_count_exposure["tractate"].get("tag")
+        if word_count_exposure["diary"].get(tag):
+            diary_exposure = word_count_exposure["diary"].get(tag)
+        if word_count_exposure["answer"].get(tag):
+            answer_exposure = word_count_exposure["answer"].get(tag)
+        if word_count_exposure["tractate"].get(tag):
+            tractate_exposure = word_count_exposure["tractate"].get(tag)
         all_data.append((data_type,data_count,diary_exposure,answer_exposure,tractate_exposure))
         print(all_data[-1])
 
@@ -408,7 +407,7 @@ def parse_data():
     # 获取曝光id对应的标签
     word_count_exposure = from_id_get_tag(card_id_dict)
     print(word_count_exposure)
-    save_data_to_csv(all_tags,user_portrait_dict,word_count_exposure)
+    save_data_to_csv(user_portrait_dict,word_count_exposure)
 
 if __name__ == "__main__":
     parse_data()