feat: add group field in team (#27)

* feat: add group field in team Signed-off-by: Dup4 <lyuzhi.pan@gmail.com> * feat: add set method Signed-off-by: Dup4 <lyuzhi.pan@gmail.com> --------- Signed-off-by: Dup4 <lyuzhi.pan@gmail.com>
xcpcio · May 19, 2024 · 2889120 · 2889120
1 parent ad3fea5
commit 2889120
Show file tree

Hide file tree

Showing 8 changed files with 20 additions and 11 deletions.
diff --git a/tests/spider/csg_cpc/v1/snapshots/csg_cpc_test/test_spider_csg_cpc_v1_2023_hncpc/teams b/tests/spider/csg_cpc/v1/snapshots/csg_cpc_test/test_spider_csg_cpc_v1_2023_hncpc/teams
diff --git a/tests/spider/domjudge/v2/snapshots/domjudge_test/test_spider_domjudge_v2_2023_hbcpc/teams b/tests/spider/domjudge/v2/snapshots/domjudge_test/test_spider_domjudge_v2_2023_hbcpc/teams
diff --git a/tests/spider/domjudge/v2/snapshots/domjudge_test/test_spider_domjudge_v2_47th_ec_final/teams b/tests/spider/domjudge/v2/snapshots/domjudge_test/test_spider_domjudge_v2_47th_ec_final/teams
diff --git a/.../spider/domjudge/v3/snapshots/domjudge_test/test_spider_domjudge_v3_9th_ccpc_guilin/teams b/.../spider/domjudge/v3/snapshots/domjudge_test/test_spider_domjudge_v3_9th_ccpc_guilin/teams
diff --git a/tests/spider/domjudge/v4/snapshots/domjudge_test/test_spider_domjudge_v4_2021_ec_final/teams b/tests/spider/domjudge/v4/snapshots/domjudge_test/test_spider_domjudge_v4_2021_ec_final/teams
diff --git a/tests/spider/ghost/v1/snapshots/ghost_test/test_spider_ghost_v1_2022_ec_final/teams b/tests/spider/ghost/v1/snapshots/ghost_test/test_spider_ghost_v1_2022_ec_final/teams
diff --git a/tests/spider/zoj/v2/snapshots/zoj_test/test_spider_zoj_v2_8th_ccpc_final/teams b/tests/spider/zoj/v2/snapshots/zoj_test/test_spider_zoj_v2_8th_ccpc_final/teams
diff --git a/xcpcio_board_spider/type/team.py b/xcpcio_board_spider/type/team.py
@@ -15,6 +15,7 @@ def __init__(self,
                  unofficial: bool = False,
                  girl: bool = False,
                  location: str = None,
+                 group: typing.List[str] = [],
                  extra: typing.Dict[str, typing.Any] = None):
         self.team_id = team_id
         self.name = name
@@ -29,10 +30,16 @@ def __init__(self,
 
         self.location = location
 
-        if extra is None:
-            self.extra = {}
-        else:
-            self.extra = extra
+        self.group = group
+        self.extra = extra if extra is not None else {}
+
+    def enable_group(self, group: str):
+        if group not in self.group:
+            self.group.append(group)
+
+    def disable_group(self, group: str):
+        if group in self.group:
+            self.group.remove(group)
 
     @property
     def get_dict(self):
@@ -57,6 +64,8 @@ def get_dict(self):
         if self.location is not None:
             obj["location"] = self.location
 
+        obj["group"] = self.group
+
         return obj
 
     @property