net_put: Handle template in src field (#103)

net_put: Handle template in src field

Reviewed-by: Nathaniel Case <this.is@nathanielca.se>
             https://github.com/Qalthos

Author: Qalthos

changelogs/fragments/103-net-put-handle-src.yaml

@@ -0,0 +1,3 @@
+---
+bugfixes:
+  - net_put - Restore missing function removed when action plugin stopped inheriting NetworkActionBase

plugins/action/net_put.py

@@ -77,8 +77,8 @@ def run(self, tmp=None, task_vars=None):
 
         if mode == "text":
             try:
-                self._handle_template(convert_data=False)
-            except ValueError as exc:
+                self._handle_src_option(convert_data=False)
+            except AnsibleError as exc:
                 return dict(failed=True, msg=to_text(exc))
 
             # Now src has resolved file write to disk in current diectory for scp
@@ -216,6 +216,46 @@ def _get_working_path(self):
             cwd = self._task._role._role_path
         return cwd
 
+    def _handle_src_option(self, convert_data=True):
+        src = self._task.args.get("src")
+        working_path = self._get_working_path()
+
+        if os.path.isabs(src) or urlsplit("src").scheme:
+            source = src
+        else:
+            source = self._loader.path_dwim_relative(
+                working_path, "templates", src
+            )
+            if not source:
+                source = self._loader.path_dwim_relative(working_path, src)
+
+        if not os.path.exists(source):
+            raise AnsibleError("path specified in src not found")
+
+        try:
+            with open(source, "r") as f:
+                template_data = to_text(f.read())
+        except IOError as e:
+            raise AnsibleError(
+                "unable to load src file {0}, I/O error({1}): {2}".format(
+                    source, e.errno, e.strerror
+                )
+            )
+
+        # Create a template search path in the following order:
+        # [working_path, self_role_path, dependent_role_paths, dirname(source)]
+        searchpath = [working_path]
+        if self._task._role is not None:
+            searchpath.append(self._task._role._role_path)
+            if hasattr(self._task, "_block:"):
+                dep_chain = self._task._block.get_dep_chain()
+                if dep_chain is not None:
+                    for role in dep_chain:
+                        searchpath.append(role._role_path)
+        searchpath.append(os.path.dirname(source))
+        self._templar.environment.loader.searchpath = searchpath
+        self._task.args["src"] = self._templar.template(template_data)
+
     def _get_network_os(self, task_vars):
         if "network_os" in self._task.args and self._task.args["network_os"]:
             display.vvvv("Getting network OS from task argument")

plugins/action/network.py

@@ -41,8 +41,8 @@ def run(self, task_vars=None):
         if config_module and self._task.args.get("src"):
             try:
                 self._handle_src_option()
-            except AnsibleError as e:
-                return {"failed": True, "msg": e.message, "changed": False}
+            except AnsibleError as exc:
+                return dict(failed=True, msg=to_text(exc))
 
         result = super(ActionModule, self).run(task_vars=task_vars)