[fix] 비례대표 국회의원까지 함께 계산에 추가

analysis/diversity_db.py

@@ -277,9 +277,12 @@ def save_to_mongo_national(factor: str, stair=0, opts=True) -> None:
     data = [
         councilor[factor_field[factor]]
         for councilor in client["council"]["national_councilor"].find()
+    ] + [
+        councilor[factor_field[factor]]
+        for councilor in client["council"]["national_councilor_global"].find()
     ]
-    # print(f"{metroId} {factor}")
-    # print(data)
+    print(f"National {factor}")
+    print(data)
     client["stats"].get_collection("diversity_index").update_one(
         {"national": True},
         {"$set": {f"{factor}DiversityIndex": gini_simpson(data, stair, opts)}},
@@ -348,17 +351,17 @@ def calculate_age_diversity_rank_history_national() -> None:
 
 
 def main():
-    for localId in range(1, 227):
-        save_to_mongo_local(localId, "age", stair=10)
-        save_to_mongo_local(localId, "gender")
-        save_to_mongo_local(localId, "party")
-    calculate_rank_local("age")
-    calculate_rank_local("gender")
-    calculate_rank_local("party")
-    calculate_age_diversity_rank_history_local()
+    # for localId in range(1, 227):
+    #     save_to_mongo_local(localId, "age", stair=10)
+    #     save_to_mongo_local(localId, "gender")
+    #     save_to_mongo_local(localId, "party")
+    # calculate_rank_local("age")
+    # calculate_rank_local("gender")
+    # calculate_rank_local("party")
+    # calculate_age_diversity_rank_history_local()
 
     for metroId in range(1, 18):
-        if metroId in [8, 17]:
+        if metroId in [17]:
             continue
         save_to_mongo_metro(metroId, "age", stair=10)
         save_to_mongo_metro(metroId, "gender")
@@ -368,10 +371,10 @@ def main():
     calculate_rank_metro("party")
     calculate_age_diversity_rank_history_metro()
 
-    save_to_mongo_national("age", stair=10)
-    save_to_mongo_national("gender")
-    save_to_mongo_national("party")
-    calculate_age_diversity_rank_history_national()
+    # save_to_mongo_national("age", stair=10)
+    # save_to_mongo_national("gender")
+    # save_to_mongo_national("party")
+    # calculate_age_diversity_rank_history_national()
 
 
 if __name__ == "__main__":