Account for empty string regexp in lineinfile (#41451)

* Use context managers for interacting with files

* Account for empty string as regexp

Rather than explicitly testing for None, also test for an empty string which will evaluate to False. An empty string regexp matches every line, which ends up replacing the incorrect line.

* Store line parameter in a variable

* Add tests
pull/41936/head
Sam Doran 7 years ago committed by GitHub
parent 3d78dd6677
commit 4b5b4a760c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -193,9 +193,8 @@ from ansible.module_utils._text import to_bytes, to_native
def write_changes(module, b_lines, dest): def write_changes(module, b_lines, dest):
tmpfd, tmpfile = tempfile.mkstemp() tmpfd, tmpfile = tempfile.mkstemp()
f = os.fdopen(tmpfd, 'wb') with open(tmpfile, 'wb') as f:
f.writelines(b_lines) f.writelines(b_lines)
f.close()
validate = module.params.get('validate', None) validate = module.params.get('validate', None)
valid = not validate valid = not validate
@ -247,14 +246,13 @@ def present(module, dest, regexp, line, insertafter, insertbefore, create,
b_lines = [] b_lines = []
else: else:
f = open(b_dest, 'rb') with open(b_dest, 'rb') as f:
b_lines = f.readlines() b_lines = f.readlines()
f.close()
if module._diff: if module._diff:
diff['before'] = to_native(b('').join(b_lines)) diff['before'] = to_native(b('').join(b_lines))
if regexp is not None: if regexp:
bre_m = re.compile(to_bytes(regexp, errors='surrogate_or_strict')) bre_m = re.compile(to_bytes(regexp, errors='surrogate_or_strict'))
if insertafter not in (None, 'BOF', 'EOF'): if insertafter not in (None, 'BOF', 'EOF'):
@ -270,7 +268,7 @@ def present(module, dest, regexp, line, insertafter, insertbefore, create,
m = None m = None
b_line = to_bytes(line, errors='surrogate_or_strict') b_line = to_bytes(line, errors='surrogate_or_strict')
for lineno, b_cur_line in enumerate(b_lines): for lineno, b_cur_line in enumerate(b_lines):
if regexp is not None: if regexp:
match_found = bre_m.search(b_cur_line) match_found = bre_m.search(b_cur_line)
else: else:
match_found = b_line == b_cur_line.rstrip(b('\r\n')) match_found = b_line == b_cur_line.rstrip(b('\r\n'))
@ -406,21 +404,20 @@ def absent(module, dest, regexp, line, backup):
'before_header': '%s (content)' % dest, 'before_header': '%s (content)' % dest,
'after_header': '%s (content)' % dest} 'after_header': '%s (content)' % dest}
f = open(b_dest, 'rb') with open(b_dest, 'rb') as f:
b_lines = f.readlines() b_lines = f.readlines()
f.close()
if module._diff: if module._diff:
diff['before'] = to_native(b('').join(b_lines)) diff['before'] = to_native(b('').join(b_lines))
if regexp is not None: if regexp:
bre_c = re.compile(to_bytes(regexp, errors='surrogate_or_strict')) bre_c = re.compile(to_bytes(regexp, errors='surrogate_or_strict'))
found = [] found = []
b_line = to_bytes(line, errors='surrogate_or_strict') b_line = to_bytes(line, errors='surrogate_or_strict')
def matcher(b_cur_line): def matcher(b_cur_line):
if regexp is not None: if regexp:
match_found = bre_c.search(b_cur_line) match_found = bre_c.search(b_cur_line)
else: else:
match_found = b_line == b_cur_line.rstrip(b('\r\n')) match_found = b_line == b_cur_line.rstrip(b('\r\n'))
@ -480,13 +477,15 @@ def main():
backrefs = params['backrefs'] backrefs = params['backrefs']
path = params['path'] path = params['path']
firstmatch = params['firstmatch'] firstmatch = params['firstmatch']
regexp = params['regexp']
line = params.get('line', None)
b_path = to_bytes(path, errors='surrogate_or_strict') b_path = to_bytes(path, errors='surrogate_or_strict')
if os.path.isdir(b_path): if os.path.isdir(b_path):
module.fail_json(rc=256, msg='Path %s is a directory !' % path) module.fail_json(rc=256, msg='Path %s is a directory !' % path)
if params['state'] == 'present': if params['state'] == 'present':
if backrefs and params['regexp'] is None: if backrefs and not regexp:
module.fail_json(msg='regexp= is required with backrefs=true') module.fail_json(msg='regexp= is required with backrefs=true')
if params.get('line', None) is None: if params.get('line', None) is None:
@ -500,13 +499,13 @@ def main():
line = params['line'] line = params['line']
present(module, path, params['regexp'], line, present(module, path, regexp, line,
ins_aft, ins_bef, create, backup, backrefs, firstmatch) ins_aft, ins_bef, create, backup, backrefs, firstmatch)
else: else:
if params['regexp'] is None and params.get('line', None) is None: if not regexp and line is None:
module.fail_json(msg='one of line= or regexp= is required with state=absent') module.fail_json(msg='one of line= or regexp= is required with state=absent')
absent(module, path, params['regexp'], params.get('line', None), backup) absent(module, path, regexp, line, backup)
if __name__ == '__main__': if __name__ == '__main__':

@ -717,3 +717,37 @@
assert: assert:
that: that:
- result.stat.checksum == 'eca8d8ea089d4ea57a3b87d4091599ca8b60dfd2' - result.stat.checksum == 'eca8d8ea089d4ea57a3b87d4091599ca8b60dfd2'
###################################################################
# Issue 29443
- name: Deploy the test file for lineinfile
copy:
src: test.txt
dest: "{{ output_dir }}/test.txt"
register: result
- name: Assert that the test file was deployed
assert:
that:
- result is changed
- result.checksum == '5feac65e442c91f557fc90069ce6efc4d346ab51'
- result.state == 'file'
- name: Insert a line in the file using an empty string as a regular expression
lineinfile:
path: "{{ output_dir }}/test.txt"
regexp: ''
line: This is line 6
register: insert_empty_regexp
- name: Stat the file
stat:
path: "{{ output_dir }}/test.txt"
register: result
- name: Assert that the file contents match what is expected
assert:
that:
- insert_empty_regexp is changed
- result.stat.checksum == '3baeade8eb2ecf4b01d70d541e9b8258b67c7f9f'
Loading…
Cancel
Save