regscale-cli 6.20.4.0__py3-none-any.whl → 6.20.4.1__py3-none-any.whl

regscale/__init__.py

@@ -1 +1 @@
-__version__ = "6.20.4.0"
+__version__ = "6.20.4.1"

regscale/integrations/public/fedramp/fedramp_cis_crm.py

@@ -610,6 +610,13 @@ def get_all_control_objectives(imps: List[ControlImplementation]) -> List[Contro
     return res
 
 
+def clean_key(key: str) -> str:
+    """
+    Clean the key by removing spaces
+    """
+    return key.replace(" ", "")
+
+
 def update_all_objectives(
     leveraged_auth_id: int,
     cis_data: Dict[str, Dict[str, str]],
@@ -634,7 +641,7 @@ def update_all_objectives(
         "[cyan]Processing control objectives...", total=len(EXISTING_IMPLEMENTATIONS.values())
     )
     # Create a combined dataset for easier access
-    combined_data = {key: {"cis": cis_data[key], "crm": crm_data.get(key, {})} for key in cis_data}
+    combined_data = {key: {"cis": cis_data[key], "crm": crm_data.get(clean_key(key), {})} for key in cis_data}
 
     # Process implementations in parallel
     with ThreadPoolExecutor(max_workers=30) as executor:
@@ -1075,7 +1082,7 @@ def parse_cis_worksheet(file_path: click.Path, cis_sheet_name: str) -> dict:
         results = list(executor.map(_process_row, [row for _, row in cis_df.iterrows()]))
 
     # iterate the results and index by control_id
-    return {result["control_id"]: result for result in results}
+    return {clean_key(result["control_id"]): result for result in results}
 
 
 def determine_skip_row(original_df: "pd.DataFrame", text_to_find: str, original_skip: int):