regscale-cli 6.20.5.0__py3-none-any.whl → 6.20.6.0__py3-none-any.whl

regscale/models/regscale_models/__init__.py

@@ -1,10 +1,12 @@
 """RegScale models."""
 
+from .assessment_plan import *
 from .assessment import *
 from .asset import *
 from .asset_mapping import *
 from .business_impact_assessment import *
 from .catalog import *
+from .case import *
 from .cci import *
 from .change import *
 from .checklist import *

regscale/models/regscale_models/asset.py

@@ -8,7 +8,7 @@ from enum import Enum
 from typing import Any, List, Optional, Union, cast
 from urllib.parse import urljoin
 
-from pydantic import ConfigDict, Field
+from pydantic import Field
 from requests import Response
 from rich.progress import Progress
 

regscale/models/regscale_models/modules.py

@@ -4,7 +4,7 @@
 
 # standard python imports
 from dataclasses import asdict, dataclass
-from typing import Any
+from typing import Any, Optional
 
 from rich.console import Console
 from rich.table import Table
@@ -189,3 +189,90 @@ class Modules:
 
         # return the string
         return output
+
+    @staticmethod
+    def module_to_class(module: str) -> Optional[Any]:
+        """
+        Function to convert RegScale module to RegScale class
+
+        :param str module: RegScale module
+        :return: RegScale class if found in the mapping, else None
+        :rtype: Optional[Any]
+        """
+        from regscale.models import (
+            AssessmentPlan,
+            Assessment,
+            Asset,
+            Case,
+            Catalog,
+            Change,
+            Component,
+            Evidence,
+            Incident,
+            Issue,
+            Policy,
+            Project,
+            Questionnaires,
+            Requirement,
+            Risk,
+            SecurityPlan,
+            SupplyChain,
+            Task,
+            Threat,
+        )
+
+        regscale_models = {
+            "assessmentplans": AssessmentPlan,
+            "assessments": Assessment,
+            "assets": Asset,
+            "cases": Case,
+            "catalogues": Catalog,
+            "changes": Change,
+            "components": Component,
+            "evidence": Evidence,
+            "incidents": Incident,
+            "issues": Issue,
+            "policies": Policy,
+            "projects": Project,
+            "questionnaires": Questionnaires,
+            "requirements": Requirement,
+            "risks": Risk,
+            "securityplans": SecurityPlan,
+            "supplychain": SupplyChain,
+            "tasks": Task,
+            "threats": Threat,
+        }
+
+        return regscale_models.get(module) or None
+
+    @staticmethod
+    def get_module_to_id(module: str) -> Optional[int]:
+        """
+        Returns the id of a RegScale module
+
+        :return: id of RegScale module, if found in the mapping, else None
+        :rtype: Optional[int]
+        """
+        module_to_id = {
+            "assessmentplans": 33,
+            "assessments": 2,
+            "assets": 3,
+            "cases": 28,
+            "catalogues": 4,
+            "changes": 31,
+            "components": 27,
+            "evidence": 32,
+            "incidents": 8,
+            "issues": 10,
+            "policies": 11,
+            "projects": 12,
+            "questionnaires": 26,
+            "requirements": 13,
+            "risks": 14,
+            "securityplans": 16,
+            "supplychain": 25,
+            "tasks": 18,
+            "threats": 19,
+        }
+
+        return module_to_id.get(module)

regscale/models/regscale_models/regscale_model.py

@@ -1150,13 +1150,18 @@ class RegScaleModel(BaseModel, ABC):
         :return: The saved object
         :rtype: T
         """
-        if self.has_changed():
+        # Check if the model has change tracking and if there are changes
+        has_change_tracking = hasattr(self, "has_changed") and callable(getattr(self, "has_changed", None))
+        should_save = not has_change_tracking or self.has_changed()
+
+        if should_save:
             if bulk:
                 logger.debug(f"Adding {self.__class__.__name__} {self.id} to pending updates")
                 self._get_pending_updates().add(self._get_cache_key(self))
                 self.cache_object(self)  # Update the cache with the current state
                 return self
             else:
+                logger.debug(f"Saving {self.__class__.__name__} {self.id}")
                 return self._perform_save()
         else:
             logger.debug(f"No changes detected for {self.__class__.__name__} {self.id}")
@@ -1284,6 +1289,7 @@ class RegScaleModel(BaseModel, ABC):
         endpoint = self.get_endpoint("update").format(id=self.id)
         response = self._get_api_handler().put(endpoint=endpoint, data=self.dict(), headers=self._get_headers())
         if hasattr(response, "ok") and response.ok:
+            logger.debug(f"Successfully saved {self.__class__.__name__} {self.id}")
             obj = self.__class__(**response.json())
             self.cache_object(obj)
             return obj

regscale/models/regscale_models/vulnerability.py

@@ -14,9 +14,9 @@ from regscale.core.app.api import Api
 from regscale.core.app.application import Application
 from regscale.core.app.utils.app_utils import get_current_datetime
 from regscale.models import regscale_models
+from regscale.models.regscale_models.issue import IssueStatus
 from regscale.models.regscale_models.regscale_model import RegScaleModel
 
-
 logger = logging.getLogger("regscale")
 
 
@@ -68,7 +68,7 @@ class Vulnerability(RegScaleModel):
     firstSeen: Optional[str] = None
     daysOpen: Optional[int] = None
     dns: Optional[str] = None
-    ipAddress: Optional[str] = ""
+    ipAddress: Optional[str] = None
     mitigated: Optional[bool] = None
     operatingSystem: Optional[str] = None
     port: Optional[Union[str, int]] = None
@@ -86,7 +86,7 @@ class Vulnerability(RegScaleModel):
     tenantsId: int = Field(default=0)
     isPublic: bool = Field(default=False)
     dateClosed: Optional[str] = None
-    status: Optional[VulnerabilityStatus] = VulnerabilityStatus.Open
+    status: Optional[Union[str, IssueStatus]] = Field(default_factory=IssueStatus.Open)
     buildVersion: Optional[str] = None
     cvsSv3BaseVector: Optional[str] = None
     cvsSv2BaseVector: Optional[str] = None

regscale/models/regscale_models/vulnerability_mapping.py

@@ -23,7 +23,7 @@ class VulnerabilityMapping(RegScaleModel):
     vulnerabilityId: int
     assetId: int
     scanId: Optional[int] = None
-    securityPlansId: Optional[int] = None
+    securityPlanId: Optional[int] = None
     status: Optional[str] = Field(max_length=450)
     firstSeen: str
     lastSeen: str
@@ -33,7 +33,7 @@ class VulnerabilityMapping(RegScaleModel):
     lastUpdatedById: str = Field(default_factory=RegScaleModel.get_user_id)
     dateCreated: str
     dateLastUpdated: Optional[str] = None
-    tenantsId: int = Field(default=0)
+    tenantsId: int = Field(default=1)
 
     @staticmethod
     def _get_additional_endpoints() -> ConfigDict:

regscale/regscale.py

@@ -127,6 +127,7 @@ file_upload = import_command_with_timing("regscale.core.app.internal.file_upload
 migrations = import_command_with_timing(INTERNAL, "migrations")
 model = import_command_with_timing(INTERNAL, "model")
 issues = import_command_with_timing(INTERNAL, "issues")
+set_permissions = import_command_with_timing(INTERNAL, "set_permissions")
 
 ############################################################
 # Public Integrations
@@ -757,6 +758,7 @@ cli.add_command(evidence)  # add Evidence Feature
 cli.add_command(migrations)  # add data migration support
 cli.add_command(issues)  # add POAM(Issues) Editor Feature
 cli.add_command(model)  # add POAM(Issues) Editor Feature
+cli.add_command(set_permissions)  # add builk editor for record permissions
 
 ############################################################
 # Add Commercial Integrations

{regscale_cli-6.20.5.0.dist-info → regscale_cli-6.20.6.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: regscale-cli
-Version: 6.20.5.0
+Version: 6.20.6.0
 Summary: Command Line Interface (CLI) for bulk processing/loading data into RegScale
 Home-page: https://github.com/RegScale/regscale-cli
 Author: Travis Howerton

{regscale_cli-6.20.5.0.dist-info → regscale_cli-6.20.6.0.dist-info}/RECORD

@@ -1,5 +1,6 @@
-regscale/__init__.py,sha256=NG2YqNT-SOqEul9gYB4T2_-wp5yQlbqTCwORHAiQc9E,25
-regscale/regscale.py,sha256=2mNIkLGtQjIvl0m255iV20_3DK4oRSTlXYxe-8XQpeo,30764
+regscale/__init__.py,sha256=ZygAIkX6Nbjag1czWdQa-yP-GM1mBE_9ss21Xh__JFc,34
+regscale/_version.py,sha256=x6wIid4ymuFQH9uE0D5dtIycsH6xYzb1mpib16IO1ps,1198
+regscale/regscale.py,sha256=xcxnTwEwWgfO3Fnp0LVo32SZCJzAswq3WDZgm21nHnI,30914
 regscale/airflow/__init__.py,sha256=yMwN0Bz4JbM0nl5qY_hPegxo_O2ilhTOL9PY5Njhn-s,270
 regscale/airflow/click_dags.py,sha256=H3SUR5jkvInNMv1gu-VG-Ja_H-kH145CpQYNalWNAbE,4520
 regscale/airflow/click_mixins.py,sha256=BTwKWsEu6KVtlrzFbXpD_RuEpMzc-CSnCCySUzqzCgQ,3571
@@ -37,7 +38,7 @@ regscale/core/app/__init__.py,sha256=nGcCN1vWBAnZzoccIlt0jwWQdegCOrBWOB7LPhQkQSs
 regscale/core/app/api.py,sha256=_hwh1HWIIrPfL4jHDKlEmW5pQEr2Iq3idn4bYGtfPQo,23148
 regscale/core/app/application.py,sha256=TalC51f30Ox6o7LBsh2RcCdU_YqFuCbGw1zhK2vSS7s,25982
 regscale/core/app/logz.py,sha256=8AdBKmquv45JGi5fCMc38JqQg6-FpUONGmqfd5EGwrI,2583
-regscale/core/app/internal/__init__.py,sha256=2ZJGwSuQBoQa1euFJ2pQsyEBMqWqXv7qq8yu02eQcYQ,4436
+regscale/core/app/internal/__init__.py,sha256=rod4nmE7rrYDdbuYF4mCWz49TK_2r-v4tWy1UHW63r0,4808
 regscale/core/app/internal/admin_actions.py,sha256=hOdma7QGwFblmxEj9UTjiWWmZGUS5_ZFtxL2qZdhf-Q,7443
 regscale/core/app/internal/assessments_editor.py,sha256=UozEaQR1v3eXbkFgOY11CkjMLEAF27ej5vlMQ4O9XVA,31477
 regscale/core/app/internal/catalog.py,sha256=svBAgFZaGrkNHh2gZEo1ex2vdjllAIc4940soGykHOc,11088
@@ -51,15 +52,16 @@ regscale/core/app/internal/login.py,sha256=GsFaBwmSc32liWoRnMFy78m8SuB9pfUV0c1ly
 regscale/core/app/internal/migrations.py,sha256=wHIzb1uglLVOXawnb-K3Yt70Z5QyfQYb8ZZOMDrNRU4,7983
 regscale/core/app/internal/model_editor.py,sha256=0sxfg92TJYyECU9KwgHooXeY3ZdaF2i5gGWe1gButFY,61696
 regscale/core/app/internal/poam_editor.py,sha256=3PtpSMpV7bqKFuTHD2ACYcTyB3EEtTRKuYv-XArjd0A,22929
+regscale/core/app/internal/set_permissions.py,sha256=MbUxe-zaAz5hK3BZ4IeY7Xbr8-FZOuJwkCbs_yVBC-k,5666
 regscale/core/app/internal/workflow.py,sha256=SpgYk1QyzdilVLOK1fFzaKhdLspumaugf5VezgboxhQ,4007
 regscale/core/app/utils/XMLIR.py,sha256=M_RrCsbjznihatkucCKw6dPgHTPQczXyqIdUXWhuCLI,8328
 regscale/core/app/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 regscale/core/app/utils/api_handler.py,sha256=T1meKw6Yi3ZAgRbQ1xuKDVh9Q9B8mbMqqN_LrSwIlAM,11765
 regscale/core/app/utils/app_utils.py,sha256=zLEHHHWR4likFQYY_bZsWLTAoRajyyaWrESkotVrFlM,36240
-regscale/core/app/utils/file_utils.py,sha256=URKWVEiR9aFnwoW3-Io7R22tBVeROTC3sX1wOZuhqXw,8912
+regscale/core/app/utils/file_utils.py,sha256=x7cqYPxJ-PgSb2lX_K6zhdqyGQjOCmr04lnqdBQV3hI,9269
 regscale/core/app/utils/parser_utils.py,sha256=aBEgcFwbJMD-ARf3wzf-tyWwR6NHvzEcdYcPMm8hGqo,2533
 regscale/core/app/utils/pickle_file_handler.py,sha256=iMdv4N8z00TB5LyPdxIcLKNRpDQVWQ8ZQWAqCKpqmF0,1695
-regscale/core/app/utils/regscale_utils.py,sha256=krtwcA6ohhF0iPDz3dMbzYv-nAqe-N5ENp2SZrg-JpY,11583
+regscale/core/app/utils/regscale_utils.py,sha256=KuXTAfEO10DQTnhQadIBIQ02iuLD0D0xKUzf5Yv8yP4,7400
 regscale/core/app/utils/report_utils.py,sha256=rKn3nmiSPq8LTKM16UiKBSizTsZKh-7JwdOlfX6YSpQ,4210
 regscale/core/app/utils/variables.py,sha256=h29Qrr6jg-Feo74IeqHz28wYLhgtzZ2pCDMf6g0WJFU,8989
 regscale/core/app/utils/catalog_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -92,7 +94,7 @@ regscale/core/static/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 regscale/core/static/regex.py,sha256=KLg-do2oPuUD6bQG9usqqSy5sxqlSEAEuUUbGVkZ6eU,384
 regscale/core/utils/__init__.py,sha256=Eyvc210t-KRqvTcV5DNWRjKO99CdWy_dC57wfggFWcw,3876
 regscale/core/utils/click_utils.py,sha256=y6vAbaPoT0Abb8siS1tvJxJQTm4sbWQTeq5nTDIgBtI,375
-regscale/core/utils/date.py,sha256=C5J4xwiYihzWmlZYlSPeOFiEDOo94CFU-OOSGYqY6rs,9308
+regscale/core/utils/date.py,sha256=_45gTGY0NrnnC49ajuYLGU8QLcTUKYm09C_z93O1oKs,10005
 regscale/core/utils/graphql.py,sha256=oXGBcAuDa0uasMnD4CokIzzKKa-IgExdKXLwtaK2_tA,8386
 regscale/core/utils/urls.py,sha256=ZcU9OJqDmVQXgu6BrLESIp2KMkkUuzTZZ_wHsZMzfA4,687
 regscale/dev/__init__.py,sha256=wVP59B1iurp36ul8pD_CjmunJLHOdKWWodz1r5loNvw,190
@@ -111,7 +113,7 @@ regscale/integrations/api_paginator.py,sha256=73rjaNM9mGv8evHAeoObXEjZPg-bJuGPo6
 regscale/integrations/api_paginator_example.py,sha256=lEuYI-xEGcjnXuIzbCobCP0YRuukLF0s8S3d382SAH4,12119
 regscale/integrations/integration_override.py,sha256=PH7t_bf-RCe_it3FJ61tlKX5UghqHuSEQNJWDfCamAg,5480
 regscale/integrations/jsonl_scanner_integration.py,sha256=l8nq_T3rE1XX-6HxrNHm3xzxCNWbIjxQvGMdtZWs7KQ,57003
-regscale/integrations/scanner_integration.py,sha256=PMP-BjSLDvBvdbc2z13zb5w3-HWfv0O41SSyj3TrLQE,135563
+regscale/integrations/scanner_integration.py,sha256=lijwTk_VD_mhNN04lMgf_Wh3IKhv_INWESFgtjt_f8Q,135805
 regscale/integrations/variables.py,sha256=AyaG286U-ek2uQ4qeYBAG3eh_QnlBlth1PKZI5Je5sc,2212
 regscale/integrations/commercial/__init__.py,sha256=sVbzBnLbKRhnAmRsIw8kRvfbyJojPfVqzH3OlgoxvqM,13927
 regscale/integrations/commercial/ad.py,sha256=YXSmK8vRf6yi2GnREGa5GrE6GelhFrLj44SY8AO1pK0,15509
@@ -226,16 +228,16 @@ regscale/integrations/commercial/trivy/commands.py,sha256=YGQFtpQGkmcLT3X2OVp7lt
 regscale/integrations/commercial/trivy/scanner.py,sha256=iiwTvjqRlLRgQvCs_FP0j83B7ApOta0MSXyO0-iHfSk,11309
 regscale/integrations/commercial/wizv2/WizDataMixin.py,sha256=s7F_rVrP9IZa_x_vh3MswR7W_UBHRfd4kHGVsNX4ips,3606
 regscale/integrations/commercial/wizv2/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-regscale/integrations/commercial/wizv2/click.py,sha256=hpexL0xHIW5WBUJu-hWQHI2vK9s6i1TI2mD6YWZOafs,12217
-regscale/integrations/commercial/wizv2/constants.py,sha256=xpcO8W9645MncNKbRFspD7m05fVhPK0dzwzToAm0ReQ,23141
+regscale/integrations/commercial/wizv2/click.py,sha256=DPis8cMST7zVXMzajYUGqI5WNg6nh08afiQhExm54_c,12276
+regscale/integrations/commercial/wizv2/constants.py,sha256=YamxNum-GHZgkTiCsKD_rii7B9chlFOeVR56FBKvZ80,23418
 regscale/integrations/commercial/wizv2/issue.py,sha256=qj5_g-SoOz6f5QDgs34GdQEyxFwJLKAWl1EVx7vjrAw,13407
 regscale/integrations/commercial/wizv2/models.py,sha256=hZ6557LJfcp1_NRbMM0V_G1erz1jEFmsuKPn86kXE54,5667
-regscale/integrations/commercial/wizv2/parsers.py,sha256=YcFOGdYZZ17hj2pcRMC9Ho2wPY94dfJ4hHwqTR5BB6c,11095
+regscale/integrations/commercial/wizv2/parsers.py,sha256=BD0i-80NhfQAHYPk0EAbUClXfzxLT3aHzb2Gs2ObVAk,11811
 regscale/integrations/commercial/wizv2/sbom.py,sha256=QcGaYiBGtZ3mBcbo-KGl-I2u6QHKAIinTk26LPy0Kng,4466
-regscale/integrations/commercial/wizv2/scanner.py,sha256=ndUQxQ-_STzLTEtqSkVBsQYLMuFu0cxc1KYIxjUv9IA,21180
-regscale/integrations/commercial/wizv2/utils.py,sha256=Na0HRlymccXPzC1lybEPqHVuvCTbfw7r4LDi98_4mUg,34252
-regscale/integrations/commercial/wizv2/variables.py,sha256=W_m_TSLUDaqyKJZHukp6i8nftFuhQKb4WRVu9lIyMAs,2485
-regscale/integrations/commercial/wizv2/wiz_auth.py,sha256=BpQTYJn3u0QiWC2IAw-bunZpBPsJtDJgOsC2zOL_UU4,5554
+regscale/integrations/commercial/wizv2/scanner.py,sha256=N-YVSbQJvTrg-tmxnxY_w7gJ7qS4VyBTuOiMMy3TXCo,23876
+regscale/integrations/commercial/wizv2/utils.py,sha256=ELJZ9sgi2h9lEHFF7mlOQQ5f8MugM-sH72BZSRjF_nk,37550
+regscale/integrations/commercial/wizv2/variables.py,sha256=TxyhBGgKUC1bpg-2gCNiU2txqDGiQosbguAFLecP69E,2591
+regscale/integrations/commercial/wizv2/wiz_auth.py,sha256=qGcUhpC9eRkJIngmv3i9yHdp0q9rv6AvA5gHkypYzCE,5549
 regscale/integrations/integration/__init__.py,sha256=WJgPLnEahD94QLE8NR8QCzlf8xk2ix76_WPDlf98ezU,70
 regscale/integrations/integration/integration.py,sha256=pr_fbqBieYbqp3PdBjuqKuZCYFf0kF4GkFdlViTKG54,586
 regscale/integrations/integration/inventory.py,sha256=gHL1a6VSV3zf4DTmksEy8qXNrqhIun8TwLxhRpuEMqY,341
@@ -255,7 +257,7 @@ regscale/integrations/public/fedramp/components.py,sha256=z6PMObm-kjRR42bT04Efnj
 regscale/integrations/public/fedramp/docx_parser.py,sha256=EA9g1iTlB6-GtOzV9JwGW8x_SruhbaIMOzstCBvjiq8,10526
 regscale/integrations/public/fedramp/fedramp_cis_crm.py,sha256=IVASI6W0uEOg_SyxJSqyHev1KzSwEnF1RTWYT3qfViw,64576
 regscale/integrations/public/fedramp/fedramp_common.py,sha256=Mdy3_WdCEcTwSXEEKXiODmr2YJTWcTg6jfyWZJWfruQ,115406
-regscale/integrations/public/fedramp/fedramp_docx.py,sha256=GnRjuWEgE9XSe9egPOQSZ8lMjY4CpqcD2IS5paI-xQc,13742
+regscale/integrations/public/fedramp/fedramp_docx.py,sha256=eKkRwfcIi4aHJp4ajKDUGJECItwrZwYfCiKzmfB2W1Q,13703
 regscale/integrations/public/fedramp/fedramp_five.py,sha256=F4KIOdnYaRjJrJHJYsN9Bmf6i5s0nus-t_GaAh8A3ow,92645
 regscale/integrations/public/fedramp/fedramp_traversal.py,sha256=BkgwsFluZO5g0Rc0WujYgbc1_YLofamJzFP7Z5UYfao,4528
 regscale/integrations/public/fedramp/import_fedramp_r4_ssp.py,sha256=Opld0vEZ4b71cIDogu6ykTUL86tXZVTSnwQzjz8w_4U,9925
@@ -309,7 +311,7 @@ regscale/models/integration_models/azure_alerts.py,sha256=2etrpvcxa7jVQrc98bJlVG
 regscale/models/integration_models/base64.py,sha256=sxV6O5qY1_TstJENX5jBPsSdQwmA83-NNhgJFunXiZE,570
 regscale/models/integration_models/burp.py,sha256=FBEBkH3U0Q8vq71FFoWnvgLRF5Hkr9GYmQFmNNHFrVk,16932
 regscale/models/integration_models/burp_models.py,sha256=UytDTAcCaxyu-knFkm_mEUH6UmWK3OTXKSC9Sc6OjVs,3669
-regscale/models/integration_models/cisa_kev_data.json,sha256=UNYylFQOW9KHp5YTozdUTC8NTNe-2xcNf6manT8MGF0,1228128
+regscale/models/integration_models/cisa_kev_data.json,sha256=0-CT_plLwQFlIsFgJ-RZxQIi_uCeyIsrWiB2XNwXctA,1231803
 regscale/models/integration_models/defender_data.py,sha256=jsAcjKxiGmumGerj7xSWkFd6r__YpuKDnYX5o7xHDiE,2844
 regscale/models/integration_models/defenderimport.py,sha256=OFwEH0Xu-HFLIZJZ8hP60Ov3lS8RR7KHEsw4wI8QnoE,5766
 regscale/models/integration_models/drf.py,sha256=Aq7AdLa_CH97NrnR-CxaFI22JjVN9uCxVN7Z-BBUaNU,18896
@@ -339,7 +341,7 @@ regscale/models/integration_models/flat_file_importer/__init__.py,sha256=V_P3R0j
 regscale/models/integration_models/sbom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 regscale/models/integration_models/sbom/cyclone_dx.py,sha256=0pFR0BWBrF5c8_cC_8mj2MXvNOMHOdHbBYXvTVfFAh8,4058
 regscale/models/integration_models/synqly_models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-regscale/models/integration_models/synqly_models/capabilities.json,sha256=A5IBt8dDS4g8Kztgos9xugSyvuvuE5YZZE8wfnRkZnU,349623
+regscale/models/integration_models/synqly_models/capabilities.json,sha256=zGoXgG919b6Gh63Mdd6md4Js4z9c088z7UdAcQvyqXM,351522
 regscale/models/integration_models/synqly_models/connector_types.py,sha256=8nxptkTexpskySnmL0obNAff_iu_fx6tJ7i1-4hJvao,461
 regscale/models/integration_models/synqly_models/ocsf_mapper.py,sha256=e2kTOhWSNRnzbgMchMx-7c21pCgSv2DqWnxvajKEKJM,16960
 regscale/models/integration_models/synqly_models/param.py,sha256=Xt5Zm6lC_VkLj7LF2qXo72TJZHysqttsp5ai0NCf1po,2643
@@ -353,11 +355,11 @@ regscale/models/integration_models/synqly_models/connectors/vulnerabilities.py,s
 regscale/models/integration_models/tenable_models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 regscale/models/integration_models/tenable_models/integration.py,sha256=lplL8zmjTFuhLreW-4y7G1fiCOBgzNAaATq800jgTQc,10271
 regscale/models/integration_models/tenable_models/models.py,sha256=dmG7btkN4YkDWwnfW5Ldc3tWEAGjPiaRgJjrqMOkPEU,15846
-regscale/models/regscale_models/__init__.py,sha256=f3i0UQgyaWxWxyQrNrIaV55KOEhMSGWZ6b_6OO7VngQ,2141
+regscale/models/regscale_models/__init__.py,sha256=ejw7TPJPwS3BwF01-JzL6-R5DLO1zR7DVW9HHcMNCJ8,2192
 regscale/models/regscale_models/assessment.py,sha256=ekzNlcsfDGBu97PMCi7hBRGbzVgxk7Ij0RfrdGh1Rfw,20440
 regscale/models/regscale_models/assessment_plan.py,sha256=qo2YA5ckSbUKDHnC_2BUc2I9kMTje9Gq-qTCXqvEKCY,1716
 regscale/models/regscale_models/assessment_result.py,sha256=K48yjYKwgY1-d_Y3aQUDcCvaqcTIVYdbKV5Wgicf4Ts,1283
-regscale/models/regscale_models/asset.py,sha256=-ws6o6hRl2sMLNcT8CIw0SQvOwLtrVSeEp5kF80ydAQ,19840
+regscale/models/regscale_models/asset.py,sha256=MDYIFlPfngjjr6obP0wLlQPFzNw8CIFeYVAKFnpdqR4,19828
 regscale/models/regscale_models/asset_mapping.py,sha256=HFlkAoPZHy2xPYq28cXuzLpFoP36SI08HTL8mH_Q-uE,6851
 regscale/models/regscale_models/business_impact_assessment.py,sha256=tkuvw1ssdrVJJcsHT43nFjFhAFTSSFRNfWRIg5W-d2k,2214
 regscale/models/regscale_models/case.py,sha256=hLVTwZXzusnXR9avqh7xSVLJwPJ1rPI_Nla_VAkpZcg,1192
@@ -402,7 +404,7 @@ regscale/models/regscale_models/link.py,sha256=lAY4Ig3Menm1EqfcAbVJ7jsCsRO5tWtJI
 regscale/models/regscale_models/master_assessment.py,sha256=t03-8vQJ7hnPNgEU0GSC1XGKOgAqHKuIupbt5IjJvtI,7322
 regscale/models/regscale_models/meta_data.py,sha256=Fg8rrWSTx3K00QkF4glH9UdY9OFWJ4_UqxleLSSbx8I,2482
 regscale/models/regscale_models/module.py,sha256=a7lalHmVTQ04ZILnJxuFWdHYDtyusBMTtonD4ICL-Wc,7272
-regscale/models/regscale_models/modules.py,sha256=CkR03zJhrBjhw50d-Kpyem7Ryh-QWz_RLHEMlx2e8VA,6150
+regscale/models/regscale_models/modules.py,sha256=yeva_tct88o2NFt93_zmqUcXZ3LucVbAv5FvQPru3cQ,8551
 regscale/models/regscale_models/objective.py,sha256=aJIpmkZ-NscqV2y8SGB4HYzm615b6mklyHnW9bL2ljk,249
 regscale/models/regscale_models/organization.py,sha256=E3Y5bwtgaIRL4rbKxU3DyfFVYUzP3d-1umG_CIUG3gE,714
 regscale/models/regscale_models/parameter.py,sha256=WS5onxkvz119brxugoBidK7K_3CeNVzchLTvW8aEdJs,3859
@@ -419,7 +421,7 @@ regscale/models/regscale_models/questionnaire.py,sha256=QMSXfNpSoaS8tkeo80C7OWLj
 regscale/models/regscale_models/questionnaire_instance.py,sha256=1LgGTwFACmWx2xOOqUKhAUOms4_3L6NZySzuteg0_sI,8241
 regscale/models/regscale_models/rbac.py,sha256=oHzKqwL4bkH2XT4WaslbNlMnWayrSKP9zYbG72e2ijk,4522
 regscale/models/regscale_models/reference.py,sha256=P_7jT6H-NZIa7TyH3j98N-ZHlB6FsjpZVRZCCpms-D4,3253
-regscale/models/regscale_models/regscale_model.py,sha256=z-SbjQpwE67sM9pfDg2oiyIJSBEGIEwqfFvCAhzM3bA,68806
+regscale/models/regscale_models/regscale_model.py,sha256=-TLI7FX63s0qHHERES33c7ho9_qHKb3X7-3dihFBd9E,69210
 regscale/models/regscale_models/requirement.py,sha256=-8PnMbuWAZHol5X1w-fzm-moD784Et0oevSVbz6xLhU,767
 regscale/models/regscale_models/risk.py,sha256=lZFDYPpTt0aEYCrYz5FBKk1Y4y2CKXYU1A8eEKT3zzg,8661
 regscale/models/regscale_models/risk_issue_mapping.py,sha256=j6uR0x6xymJ0BsIwgnadeUQDshHNhU3VnpyItZp4dc4,2393
@@ -441,8 +443,8 @@ regscale/models/regscale_models/task.py,sha256=le3N2GIUCEJrFnNh0DLU7RAcle4ULr8OP
 regscale/models/regscale_models/threat.py,sha256=4TNZcRnTgmlDwBsYu5Pbh9GRd8ZWAtqqr0Xph3uPNAA,7255
 regscale/models/regscale_models/user.py,sha256=wiU2qKwp3aYtmaHEmTtv8BbMEgFb913dHgc2VnmsAkg,7186
 regscale/models/regscale_models/user_group.py,sha256=vzlXHvPNsgJd38H0R3osi46Oj19QO5oPx0qXntQBKWI,1891
-regscale/models/regscale_models/vulnerability.py,sha256=VKgWMVvAQ74Sq46myYsivGfIeW63dHq3bv3cRCkDcbg,10960
-regscale/models/regscale_models/vulnerability_mapping.py,sha256=cq44xkkCH7rfp0BJxavr4DLiEAHouyyPmi5EaizH6NI,6261
+regscale/models/regscale_models/vulnerability.py,sha256=klN8MYatevFGVaulex6WeLw-uvxT07ty-cXU9CxS_-Q,11042
+regscale/models/regscale_models/vulnerability_mapping.py,sha256=k14azHU7VwryHfXaJdvAMWXs3AqHDr7rnj602bscWQM,6260
 regscale/models/regscale_models/workflow.py,sha256=uMNVVAn5qR8wxjMP5PHbcvMmvPRe1gn1oclVUpbv52s,1873
 regscale/models/regscale_models/workflow_action.py,sha256=zMHrvUkTQJlkCj6Xx6wadozBwsVPpObj5GD81LI2VZc,925
 regscale/models/regscale_models/workflow_instance.py,sha256=15QPr25bVdB3ERPXmQfy2REflVsOZAq6-wZF0GV0MPo,10895
@@ -487,6 +489,7 @@ tests/mocks/xml.py,sha256=WYeZRZfyYmOi4TTvF7I53l3gKF4x1DE1dqy08szPkfQ,261
 tests/regscale/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/regscale/test_about.py,sha256=32YZC8XJW5QkIYIfwEVGIdIxQNUIbyzn3OzVcX_5X24,635
 tests/regscale/test_authorization.py,sha256=fls5ODCYiu0DdkwXFepO_GM-BP6tRaPmMCZWX6VD2e8,1899
+tests/regscale/test_init.py,sha256=W4deASP6ebAfheqaU5OaiGAfMP6vHGXkJBI4iO98cMQ,3998
 tests/regscale/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/regscale/core/test_api.py,sha256=25AsIT-jg8SrlkPeynUwwm4PvTOrHcRO2Ba_omnNLUY,7512
 tests/regscale/core/test_app.py,sha256=8HvRRD37pI_x4VjXyjs1g7JWXr-jFIbK_lYst4tLXjI,18922
@@ -514,9 +517,9 @@ tests/regscale/models/test_regscale_model.py,sha256=ZsrEZkC4EtdIsoQuayn1xv2gEGcV
 tests/regscale/models/test_report.py,sha256=eiSvS_zS0aVeL0HBvtmHVvEzcfF9ZFVn2twj5g8KttY,970
 tests/regscale/models/test_tenable_integrations.py,sha256=PNJC2Zu6lv1xj7y6e1yOsz5FktSU3PRKb5x3n5YG3w0,4072
 tests/regscale/models/test_user_model.py,sha256=e9olv28qBApgnvK6hFHOgXjUC-pkaV8aGDirEIWASL4,4427
-regscale_cli-6.20.5.0.dist-info/LICENSE,sha256=ytNhYQ9Rmhj_m-EX2pPq9Ld6tH5wrqqDYg-fCf46WDU,1076
-regscale_cli-6.20.5.0.dist-info/METADATA,sha256=uhXNlWUvaELYMD0YsHh_2qCqD_OXwj17t9lEeB8jrBw,34899
-regscale_cli-6.20.5.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-regscale_cli-6.20.5.0.dist-info/entry_points.txt,sha256=cLOaIP1eRv1yZ2u7BvpE3aB4x3kDrDwkpeisKOu33z8,269
-regscale_cli-6.20.5.0.dist-info/top_level.txt,sha256=Uv8VUCAdxRm70bgrD4YNEJUmDhBThad_1aaEFGwRByc,15
-regscale_cli-6.20.5.0.dist-info/RECORD,,
+regscale_cli-6.20.6.0.dist-info/LICENSE,sha256=ytNhYQ9Rmhj_m-EX2pPq9Ld6tH5wrqqDYg-fCf46WDU,1076
+regscale_cli-6.20.6.0.dist-info/METADATA,sha256=4wWenW-x93Egn3RCqEkEP59SYaDsBVsq-KPzsB8ZsXs,34899
+regscale_cli-6.20.6.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+regscale_cli-6.20.6.0.dist-info/entry_points.txt,sha256=cLOaIP1eRv1yZ2u7BvpE3aB4x3kDrDwkpeisKOu33z8,269
+regscale_cli-6.20.6.0.dist-info/top_level.txt,sha256=Uv8VUCAdxRm70bgrD4YNEJUmDhBThad_1aaEFGwRByc,15
+regscale_cli-6.20.6.0.dist-info/RECORD,,

tests/regscale/test_init.py

@@ -0,0 +1,94 @@
+import sys
+import types
+import unittest
+from unittest.mock import patch, mock_open, MagicMock
+
+airflow_exceptions = types.ModuleType("airflow.exceptions")
+airflow_exceptions.AirflowException = Exception
+sys.modules["airflow.exceptions"] = airflow_exceptions
+
+airflow_settings = types.ModuleType("airflow.settings")
+airflow_settings.initialize = lambda: None
+airflow_settings.LAZY_LOAD_PROVIDERS = True
+sys.modules["airflow.settings"] = airflow_settings
+
+sys.modules["airflow.configuration"] = types.ModuleType("airflow.configuration")
+
+airflow_operators = types.ModuleType("airflow.operators")
+airflow_operators_python = types.ModuleType("airflow.operators.python")
+
+
+class DummyPythonOperator:
+    def __init__(self, *args, **kwargs):
+        pass
+
+
+airflow_operators_python.PythonOperator = DummyPythonOperator
+sys.modules["airflow.operators"] = airflow_operators
+sys.modules["airflow.operators.python"] = airflow_operators_python
+
+from regscale.airflow.tasks.init import get_shared_keys, set_shared_config_values  # noqa: E402
+
+
+class TestInitPy(unittest.TestCase):
+    @patch("regscale.airflow.tasks.init.execute_click_command")
+    @patch("yaml.safe_load", return_value={"foo": "bar"})
+    @patch("pathlib.Path.open", new_callable=mock_open, read_data="foo: bar\n")
+    def test_get_shared_keys(self, mock_file, mock_yaml, mock_exec):
+        """
+        If the YAML and dag_run.conf share a key, we should get that key back and execute the click command once.
+        """
+        dag_context = {"dag_run": MagicMock(conf={"foo": "baz"})}
+        shared_keys = get_shared_keys("dummy.yaml", **dag_context)
+        self.assertEqual(shared_keys, ["foo"])
+        mock_exec.assert_called_once()
+
+    @patch("regscale.airflow.tasks.init.execute_click_command")
+    def test_set_shared_config_values(self, mock_exec):
+        """
+        When shared keys are found in xcom, set_shared_config_values should call execute_click_command for each.
+        """
+        dag_context = {
+            "dag_run": MagicMock(conf={"foo": "bar", "op_kwargs": {"shared_keys_task": "task1"}}),
+            "ti": MagicMock(xcom_pull=MagicMock(return_value=["foo"])),
+        }
+        set_shared_config_values(shared_keys_task=None, **dag_context)
+        mock_exec.assert_called_once()
+
+    def test_set_shared_config_values_raises(self):
+        """
+        If op_kwargs is missing from dag_run.conf, set_shared_config_values should raise an AirflowException.
+        """
+        from airflow.exceptions import AirflowException
+
+        dag_context = {"dag_run": MagicMock(conf={})}
+        with self.assertRaises(AirflowException):
+            set_shared_config_values(shared_keys_task=None, **dag_context)
+
+    @patch("regscale.airflow.tasks.init.execute_click_command")
+    def test_set_shared_config_values_warns(self, mock_exec):
+        """
+        If xcom_pull returns None, set_shared_config_values should log a warning and not call execute_click_command.
+        """
+        dag_context = {
+            "dag_run": MagicMock(conf={"foo": "bar", "op_kwargs": {"shared_keys_task": "task1"}}),
+            "ti": MagicMock(xcom_pull=MagicMock(return_value=None)),
+        }
+        with self.assertLogs(level="WARNING") as log:
+            set_shared_config_values(shared_keys_task=None, **dag_context)
+        self.assertTrue(any("No shared keys found" in msg for msg in log.output))
+        mock_exec.assert_not_called()
+
+    @patch("yaml.safe_load", return_value={"foo": "bar"})
+    @patch("pathlib.Path.open", new_callable=mock_open, read_data="foo: bar\n")
+    def test_get_shared_keys_logs_error(self, mock_file, mock_yaml):
+        """
+        If dag_run is missing from the context, get_shared_keys should log an error and raise KeyError.
+        """
+        with self.assertLogs(level="ERROR") as log, self.assertRaises(KeyError):
+            get_shared_keys("dummy.yaml")
+        self.assertTrue(any("context contains" in msg for msg in log.output))
+
+
+if __name__ == "__main__":
+    unittest.main()