external/mailinabox
1
0
Fork 0
mirror of https://github.com/mail-in-a-box/mailinabox.git synced 2025-04-04 00:17:06 +00:00
Code Issues Releases Wiki Activity

Add a hook in status_checks --show-changes

Browse Source
This commit is contained in:
downtownallday 2022-10-29 10:00:17 -04:00
parent aad837b75e
commit e4c20641c3
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
management/status_checks.py
View File

@ -1,4 +1,5 @@
#!/usr/local/lib/mailinabox/env/bin/python #!/usr/local/lib/mailinabox/env/bin/python
# -*- indent-tabs-mode: t; tab-width: 4; python-indent-offset: 4; -*-
##### #####
##### This file is part of Mail-in-a-Box-LDAP which is released under the ##### This file is part of Mail-in-a-Box-LDAP which is released under the
##### terms of the GNU Affero General Public License as published by the ##### terms of the GNU Affero General Public License as published by the
@ -28,6 +29,7 @@ from ssl_certificates import get_ssl_certificates, get_domain_ssl_files, check_c
from mailconfig import get_mail_domains, get_mail_aliases from mailconfig import get_mail_domains, get_mail_aliases
from utils import shell, sort_domains, load_env_vars_from_file, load_settings from utils import shell, sort_domains, load_env_vars_from_file, load_settings
import hooks
def get_services(): def get_services():
return [ return [
@ -75,6 +77,7 @@ def run_checks(rounded_values, env, output, pool, domains_to_check=None):
run_network_checks(env, output) run_network_checks(env, output)
run_domain_checks(rounded_values, env, output, pool, domains_to_check=domains_to_check) run_domain_checks(rounded_values, env, output, pool, domains_to_check=domains_to_check)
def get_ssh_port(): def get_ssh_port():
# Returns ssh port # Returns ssh port
try: try:
@ -1006,11 +1009,21 @@ def run_and_output_changes(env, pool):
out.add_heading(category) out.add_heading(category)
out.print_warning("This section was removed.") out.print_warning("This section was removed.")
# execute hooks
hook_data = {
'op':'output_changes_end',
'since': os.stat(cache_fn).st_mtime if os.path.exists(cache_fn) else 0,
'env':env,
'output':out
}
hooks.exec_hooks('status_checks', hook_data)
# Store the current status checks output for next time. # Store the current status checks output for next time.
os.makedirs(os.path.dirname(cache_fn), exist_ok=True) os.makedirs(os.path.dirname(cache_fn), exist_ok=True)
with open(cache_fn, "w") as f: with open(cache_fn, "w") as f:
json.dump(cur.buf, f, indent=True) json.dump(cur.buf, f, indent=True)
def normalize_ip(ip): def normalize_ip(ip):
# Use ipaddress module to normalize the IPv6 notation and # Use ipaddress module to normalize the IPv6 notation and
# ensure we are matching IPv6 addresses written in different # ensure we are matching IPv6 addresses written in different
@ -1097,6 +1110,10 @@ class BufferedOutput:
if __name__ == "__main__": if __name__ == "__main__":
from utils import load_environment from utils import load_environment
import logging
if os.environ.get('DEBUG','0') == '1':
logging.basicConfig(level=logging.DEBUG)
env = load_environment() env = load_environment()