Mini Shell

#!/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
import logging

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

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

from clconfigure import setup_logger

LOGS_PATH = '/var/log/cloudlinux/cloudlinux-edition-watcher.log'

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

    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()

    logging.debug('Executing with arguments "%s"', str(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)