ansible-collections/community.general
1
0
Fork 0
mirror of https://github.com/ansible-collections/community.general.git synced 2025-06-13 11:49:11 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

module_helper - fixed decorator cause_changes (#2203)

Browse source 
* fixed decorator cause_changes

* added changelog fragment

* typo
This commit is contained in:
Alexei Znamensky 2021-04-08 17:58:39 +12:00 • committed by GitHub
parent b6ae47c455
commit 0cd0f0eaf6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 68 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

33
plugins/module_utils/module_helper.py
View file

@ -101,18 +101,27 @@ class ArgFormat(object):
return [str(p) for p in func(value)]
def cause_changes(func, on_success=True, on_failure=False):
@wraps(func)
def wrapper(self, *args, **kwargs):
try:
func(*args, **kwargs)
if on_success:
self.changed = True
except Exception:
if on_failure:
self.changed = True
raise
return wrapper
def cause_changes(on_success=None, on_failure=None):
def deco(func):
if on_success is None and on_failure is None:
return func
@wraps(func)
def wrapper(*args, **kwargs):
try:
self = args[0]
func(*args, **kwargs)
if on_success is not None:
self.changed = on_success
except Exception:
if on_failure is not None:
self.changed = on_failure
raise
return wrapper
return deco
def module_fails_on_exception(func):