OwlCyberSecurity - MANAGER

Edit File: cloudlinux-edition-watcher

#!/opt/cloudlinux/venv/bin/python3 -bb
# coding=utf-8
#
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2020 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENCE.TXT
#
"""
Detects edition change and performs predefined actions.
"""
import argparse

from clcommon.lib.cledition import CLEditions
from clsentry import init_sentry_client

from clconfigure.watcher import (
    PKG_VERSION_TINY,
    SENTRY_DSN,
    check,
    save_edition
)

if __name__ == "__main__":
    init_sentry_client('cloudlinux-edition-watcher', PKG_VERSION_TINY, SENTRY_DSN, custom_length=7000)

parser = argparse.ArgumentParser()

subparsers = parser.add_subparsers(dest='action')
    subparsers.add_parser('save')
    check_subparser = subparsers.add_parser('check')
    check_subparser.add_argument('--pre', action='store_true', default=False)
    check_subparser.add_argument('--edition', required=False, default=None)

args = parser.parse_args()
    if args.action == 'save':
        save_edition()
    elif args.action == 'check':
        new_edition = args.edition or CLEditions.get_cl_edition(skip_marker_check=True)
        check(new_edition, is_pre_check=args.pre)
    else:
        parser.print_help()
        exit(1)