feat: support location

Signed-off-by: Dup4 <lyuzhi.pan@gmail.com>
xcpcio · Oct 15, 2023 · bbab4ac · bbab4ac
1 parent 0200825
commit bbab4ac
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/xcpcio_board_spider/spider/csg_cpc/v1/csg_cpc.py b/xcpcio_board_spider/spider/csg_cpc/v1/csg_cpc.py
@@ -60,7 +60,7 @@ def parse_teams(self):
             name = raw_team["name"]
             school = raw_team["school"]
             members = raw_team["tmember"].split("、")
-            coach = raw_team["coach"]
+            coach = str(raw_team["coach"])
             kind = int(raw_team["tkind"])
             room = str(raw_team["room"])
 
@@ -69,7 +69,9 @@ def parse_teams(self):
             team.organization = school
             team.members = members
             team.coach = coach
-            team.location = room
+
+            if len(room) > 0:
+                team.location = room
 
             if kind == 0:
                 team.official = True

diff --git a/xcpcio_board_spider/type/team.py b/xcpcio_board_spider/type/team.py
@@ -14,7 +14,7 @@ def __init__(self,
                  official: bool = False,
                  unofficial: bool = False,
                  girl: bool = False,
-                 location: str = "",
+                 location: str = None,
                  extra: typing.Dict[str, typing.Any] = None):
         self.team_id = team_id
         self.name = name
@@ -54,6 +54,9 @@ def get_dict(self):
         if self.girl is not None:
             obj[constants.TEAM_TYPE_GIRL] = self.girl
 
+        if self.location is not None:
+            obj["location"] = self.location
+
         return obj
 
     @property