Add org to state file

SURFscz · Jul 24, 2024 · bb3f9e4 · bb3f9e4
1 parent ca411a1
commit bb3f9e4
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/SRAMsync/json_file.py b/SRAMsync/json_file.py
@@ -124,14 +124,14 @@ def add_user(self, user: str, co: str) -> None:
         self._new_state["users"][user] = cast(StateUser, {"CO": co})
 
     def add_groups(
-        self, dest_group_names: list[str], co: str, sram_group: str, group_attributes: list[str]
+        self, dest_group_names: list[str], co: str, org: str, sram_group: str, group_attributes: list[str]
     ) -> None:
         for dest_group_name in dest_group_names:
             if dest_group_name not in self._new_state["groups"]:
                 self._new_state["groups"][dest_group_name] = {
                     "members": [],
                     "attributes": group_attributes,
-                    "sram": {"CO": co, "sram-group": sram_group, "org": ""},
+                    "sram": {"CO": co, "sram-group": sram_group, "org": org},
                     "graced_users": {},
                 }
 

diff --git a/SRAMsync/state.py b/SRAMsync/state.py
@@ -73,7 +73,7 @@ def add_user(self, user: str, co: str) -> None:
 
     @abstractmethod
     def add_groups(
-        self, dest_group_names: list[str], co: str, sram_group: str, group_attributes: list[str]
+        self, dest_group_names: list[str], co: str, org: str, sram_group: str, group_attributes: list[str]
     ) -> None:
         """Add a new group."""
 

diff --git a/SRAMsync/sync_with_sram.py b/SRAMsync/sync_with_sram.py
@@ -468,6 +468,7 @@ def process_group_data(cfg: Config, fq_co: str, org: str, co: str) -> None:
             cfg.state.add_groups(
                 dest_group_names=dest_group_names,
                 co=co,
+                org=org,
                 sram_group=sram_group,
                 group_attributes=group_attributes,
             )