netbox-config-diff 2.2.0__tar.gz → 2.3.0__tar.gz

{netbox-config-diff-2.2.0/netbox_config_diff.egg-info → netbox-config-diff-2.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: netbox-config-diff
-Version: 2.2.0
+Version: 2.3.0
 Summary: Push rendered device configurations from NetBox to devices and apply them.
 Author: Artem Kotik
 Author-email: miaow2@yandex.ru
@@ -275,7 +275,7 @@ This is possible thanks to the scrapli_cfg. Read [Scrapli](https://github.com/sc
 
 | NetBox Version | Plugin Version |
 |----------------|----------------|
-|    3.5, 3.6    |    =>0.1.0     |
+|  3.5, 3.6, 3.7 |    =>0.1.0     |
 
 <!--install-start-->
 ## Installing

{netbox-config-diff-2.2.0 → netbox-config-diff-2.3.0}/README.md

@@ -44,7 +44,7 @@ This is possible thanks to the scrapli_cfg. Read [Scrapli](https://github.com/sc
 
 | NetBox Version | Plugin Version |
 |----------------|----------------|
-|    3.5, 3.6    |    =>0.1.0     |
+|  3.5, 3.6, 3.7 |    =>0.1.0     |
 
 <!--install-start-->
 ## Installing

{netbox-config-diff-2.2.0 → netbox-config-diff-2.3.0}/netbox_config_diff/__init__.py

@@ -2,7 +2,7 @@ from extras.plugins import PluginConfig
 
 __author__ = "Artem Kotik"
 __email__ = "miaow2@yandex.ru"
-__version__ = "2.2.0"
+__version__ = "2.3.0"
 
 
 class ConfigDiffConfig(PluginConfig):

{netbox-config-diff-2.2.0 → netbox-config-diff-2.3.0}/netbox_config_diff/configurator/base.py

@@ -126,7 +126,7 @@ class Configurator(SecretsMixin):
                 )
                 device.rendered_config = rendered_config
             else:
-                actual_config = await conn.get_config()
+                actual_config = await conn.get_config(config_template=device.rendered_config)
             device.actual_config = conn.clean_config(actual_config.result)
 
             device.diff = get_unified_diff(device.rendered_config, device.actual_config, device.name)

{netbox-config-diff-2.2.0 → netbox-config-diff-2.3.0}/netbox_config_diff/configurator/platforms.py

@@ -1,5 +1,5 @@
 import re
-from typing import Pattern
+from typing import Any, Pattern
 
 from scrapli_cfg.exceptions import TemplateError
 from scrapli_cfg.platform.core.arista_eos import AsyncScrapliCfgEOS
@@ -94,13 +94,17 @@ class CustomScrapliCfg:
         """
         self.logger.info("fetching configuration and replacing with provided substitutes")
 
-        source_config = await self.get_config(source=source)
+        source_config = await self.get_config(config_template=config_template, source=source)
         return source_config, self._render_substituted_config(
             config_template=config_template,
             substitutes=substitutes,
             source_config=source_config.result,
         )
 
+    async def get_config(self, **kwargs) -> ScrapliCfgResponse:
+        kwargs.pop("config_template", None)
+        return await super().get_config(**kwargs)
+
 
 class CustomAsyncScrapliCfgEOS(CustomScrapliCfg, AsyncScrapliCfgEOS):
     pass
@@ -119,4 +123,67 @@ class CustomAsyncScrapliCfgNXOS(CustomScrapliCfg, AsyncScrapliCfgNXOS):
 
 
 class CustomAsyncScrapliCfgJunos(CustomScrapliCfg, AsyncScrapliCfgJunos):
-    pass
+    is_set_config = False
+
+    async def get_config(self, config_template: str, source: str = "running") -> ScrapliCfgResponse:
+        response = self._pre_get_config(source=source)
+
+        command = "show configuration"
+        if re.findall(r"^set\s+", config_template, flags=re.I | re.M):
+            self.is_set_config = True
+            command += " | display set"
+
+        if self._in_configuration_session is True:
+            config_result = await self.conn.send_config(config=f"run {command}")
+        else:
+            config_result = await self.conn.send_command(command=command)
+
+        return self._post_get_config(
+            response=response,
+            source=source,
+            scrapli_responses=[config_result],
+            result=config_result.result,
+        )
+
+    async def load_config(self, config: str, replace: bool = False, **kwargs: Any) -> ScrapliCfgResponse:
+        """
+        Load configuration to a device
+
+        Supported kwargs:
+            set: bool indicating config is a "set" style config (ignored if replace is True)
+
+        Args:
+            config: string of the configuration to load
+            replace: replace the configuration or not, if false configuration will be loaded as a
+                merge operation
+            kwargs: additional kwargs that the implementing classes may need for their platform,
+                see above for junos supported kwargs
+
+        Returns:
+            ScrapliCfgResponse: response object
+
+        Raises:
+            N/A
+
+        """
+        response = self._pre_load_config(config=config)
+
+        config = self._prepare_load_config(config=config, replace=replace)
+
+        config_result = await self.conn.send_config(config=config, privilege_level="root_shell")
+
+        if self.is_set_config is True:
+            load_config = f"load set {self.filesystem}{self.candidate_config_filename}"
+        else:
+            if self._replace is True:
+                load_config = f"load override {self.filesystem}{self.candidate_config_filename}"
+            else:
+                load_config = f"load merge {self.filesystem}{self.candidate_config_filename}"
+
+        load_result = await self.conn.send_config(config=load_config)
+        self._in_configuration_session = True
+
+        return self._post_load_config(
+            response=response,
+            scrapli_responses=[config_result, load_result],
+        )

{netbox-config-diff-2.2.0 → netbox-config-diff-2.3.0/netbox_config_diff.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: netbox-config-diff
-Version: 2.2.0
+Version: 2.3.0
 Summary: Push rendered device configurations from NetBox to devices and apply them.
 Author: Artem Kotik
 Author-email: miaow2@yandex.ru
@@ -275,7 +275,7 @@ This is possible thanks to the scrapli_cfg. Read [Scrapli](https://github.com/sc
 
 | NetBox Version | Plugin Version |
 |----------------|----------------|
-|    3.5, 3.6    |    =>0.1.0     |
+|  3.5, 3.6, 3.7 |    =>0.1.0     |
 
 <!--install-start-->
 ## Installing

{netbox-config-diff-2.2.0 → netbox-config-diff-2.3.0}/tests/test_compliance_utils.py

@@ -56,4 +56,4 @@ def test_exclude_lines(regex: str, expected: str) -> None:
     ids=["diff", "no diff"],
 )
 def test_get_unified_diff(render: str, actual: str, expected: str) -> None:
-    assert get_unified_diff(render, actual, "test-1") == expected
+    assert get_unified_diff(actual, render, "test-1") == expected

{netbox-config-diff-2.2.0 → netbox-config-diff-2.3.0}/netbox_config_diff/compliance/utils.py

@@ -30,8 +30,8 @@ class CustomChoiceVar(ScriptVariable):
 
 def get_unified_diff(rendered_config: str, actual_config: str, device: str) -> str:
     diff = unified_diff(
-        rendered_config.strip().splitlines(),
         actual_config.splitlines(),
+        rendered_config.strip().splitlines(),
         fromfiledate=device,
         tofiledate=device,
         lineterm="",