Merge pull request #56 from NewWays-TechForImpactKAIST/API-model

Add API Model

API/MongoDB.py

@@ -0,0 +1,32 @@
+from dataclasses import dataclass
+
+
+@dataclass
+class Councilor:
+    sdName: str
+    sggName: str
+    wiwName: str
+    name: str
+    gender: str
+    birthday: str
+    age: int
+    jobId: int
+    job: str
+    eduId: int
+    edu: str
+
+    @classmethod
+    def from_dict(cls, data: dict):
+        return cls(
+            sdName=data.get("sdName"),
+            sggName=data.get("sggName"),
+            wiwName=data.get("wiwName"),
+            name=data.get("name"),
+            gender=data.get("gender"),
+            birthday=data.get("birthday"),
+            age=data.get("age"),
+            jobId=data.get("jobId"),
+            job=data.get("job"),
+            eduId=data.get("eduId"),
+            edu=data.get("edu"),
+        )

API/candidate.py

@@ -3,6 +3,8 @@
 import os, requests, sys
 import xml.etree.ElementTree as ET
 from typing import List
+from .MongoDB import Councilor
+import argparse
 
 from configurations.secrets import OpenDataPortalSecrets
 from .utils import save_to_excel
@@ -39,11 +41,10 @@ def fetch_data(
     data_list = []
     for item in root.findall(".//item"):
         data_entry = {child.tag: child.text for child in item}
-
         for column in drop_columns:
             data_entry.pop(column)
 
-        data_list.append(data_entry)
+        data_list.append(Councilor.from_dict(data_entry))
 
     return data_list