feat: add set method

Signed-off-by: Dup4 <lyuzhi.pan@gmail.com>
xcpcio · May 19, 2024 · 4f210a9 · 4f210a9
1 parent 333d6e8
commit 4f210a9
Show file tree

Hide file tree

Showing 8 changed files with 15 additions and 7 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
@@ -33,6 +33,14 @@ def __init__(self,
         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):
         obj = {}