✨ first implemenation of line in file: append, ensure and absent. moremoban/moban#217, #1

Author: chfw

moban-ansible.yml

@@ -3,8 +3,8 @@ organisation: "moremoban"
 author: "chfw"
 contact: "[email protected]"
 company: "C.W."
-version: "0.0.1"
-current_version: "0.0.1"
+version: "0.0.2"
+current_version: "0.0.2"
 release: "0.0.1"
 copyright_year: 2020
 license: newbsd

moban_ansible/engines/__init__.py

@@ -0,0 +1,48 @@
+import re
+
+from moban.core.content_processor import ContentProcessor
+
+
+@ContentProcessor("lineinfile", "Checking", "Checked")
+def line_in_file(content: str, options: dict) -> str:
+    regexp = options.get("regexp")
+    create = options.get("create", False)
+    state = options.get("state", "present")
+
+    lines = content.split("\n")
+
+    if state == "present":
+        if regexp:
+            lines = present(lines, options)
+        elif create:
+            # regexp is None or ''
+            # append the line to the end
+            line = options.get("line")
+            if line:
+                lines += [line]
+
+    else:
+        lines = list(absent(lines, options))
+
+    return "\n".join(lines)
+
+
+def present(lines, options):
+    ensure_line = options.get("line")
+    regexp = options["regexp"]
+
+    for line in lines:
+        if re.match(regexp, line):
+            yield ensure_line
+        else:
+            yield line
+
+
+def absent(lines, options):
+    regexp = options["regexp"]
+
+    for line in lines:
+        if re.match(regexp, line):
+            continue
+
+        yield line

moban_ansible/engines/line_in_file.py

@@ -0,0 +1,38 @@
+from nose.tools import eq_
+
+from moban_ansible.engines.line_in_file import line_in_file
+
+
+def test_append_a_line():
+    options = {"create": "yes", "line": "192.168.1.1 foo.bar foo"}
+
+    content = "127.0.0.1 localhost"
+
+    new_content = line_in_file(content, options)
+    expected = [content, options["line"]]
+
+    eq_(new_content, "\n".join(expected))
+
+
+def test_absent():
+    absent_line = "192.168.1.1 foo.bar foo"
+    options = {"regexp": ".*foo$", "state": "absent"}
+
+    content = f"127.0.0.1 localhost\n{absent_line}"
+
+    new_content = line_in_file(content, options)
+    expected = "127.0.0.1 localhost"
+
+    eq_(new_content, expected)
+
+
+def test_present():
+    present_line = "192.168.1.1 foo.bar foo"
+    options = {"regexp": ".*foo$", "line": present_line}
+
+    content = f"127.0.0.1 localhost\n192.162.1.1 foo"
+
+    new_content = line_in_file(content, options)
+    expected = ["127.0.0.1 localhost", present_line]
+
+    eq_(new_content, "\n".join(expected))