1#@!#!123s

: / opt / dedrads /
Filename : cpmerge
back
#!/usr/lib/rads/venv/bin/python3
"""
This module provides command-line interface (CLI) functionalities for the
application.

It includes functions and classes to parse command-line arguments, handle user
input, and execute appropriate actions based on the commands and options
provided by the user.
"""
import sys
import logging
from argparse import ArgumentParser

# pylint: disable=wrong-import-position,import-error,import-self
sys.path.insert(0, '/usr/lib/dedrads')
from cpmerge import merge, revert_merge, check_target_user, check_users, split

## see merge.py for the majority of the notes.

## logging ##
logger = logging.getLogger("cpmerge")
logger.setLevel(logging.DEBUG)
file_handler = logging.FileHandler("/var/log/cpmerge.log")
file_handler.setLevel(logging.DEBUG)
console_handler = logging.StreamHandler(sys.stderr)
console_handler.setLevel(logging.INFO)
formatter = logging.Formatter(
    "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
)
console_formatter = logging.Formatter("%(message)s")
file_handler.setFormatter(formatter)
console_handler.setFormatter(console_formatter)
logger.addHandler(file_handler)
logger.addHandler(console_handler)
## logging ##


def main():
    """Main function"""
    version = '0.9.12'
    parser = ArgumentParser(
        description='Merge single or multiple cPanels into one cPanel account'
    )
    parser.add_argument('--target', '-t', help='User to merge into')
    parser.add_argument('--user', '-u', help='User to merge into target user ')
    parser.add_argument(
        '--all',
        '-a',
        help='Merge all users on the server into the target user',
        action='store_true',
    )
    parser.add_argument(
        '--owner', '-o', help='Merge all users owned by owner into target user'
    )
    parser.add_argument(
        '--list',
        '-l',
        help='list users to merge \n Example: --list \'user1 user2 user3\'',
    )
    parser.add_argument(
        '--no_backup',
        '-nb',
        help='''don\'t backup merged users prior to merging.\n
                cpmerge takes backups by default.''',
        action='store_true',
    )
    parser.add_argument(
        '--no_term',
        '-nt',
        help='don\'t terminate users after merging',
        action='store_true',
    )
    parser.add_argument(
        '--revert',
        '-r',
        help='''revert merge, or split from backups taken by cpmerge
                by target user.''',
    )
    parser.add_argument(
        '--version',
        '-v',
        help='display version',
        action='version',
        version=f'cpmerge {version}',
    )
    parser.add_argument(
        '--split', '-s', help='split a cpanel into multiple cpanels by domain'
    )
    parser.add_argument(
        '--domain',
        '-d',
        help='domain to split. this flag is only used with --split',
    )
    args = parser.parse_args()

    ## instantiate lists for appending ##
    users = []
    owner = []

    if args.split:
        if args.domain:
            domain = args.domain
        else:
            domain = False
        check_target_user(args.split)
        split(args.split, args.no_backup, domain)
    elif args.target:
        target_user = args.target
        check_target_user(target_user)
        if args.user:
            users.append(args.user)
            check_users(users)
            merge(
                target_user=target_user,
                merge_type='list',
                users=users,
                no_backup=args.no_backup,
                no_term=args.no_term,
            )
        elif args.all:
            merge(
                target_user=target_user,
                merge_type='all',
                no_backup=args.no_backup,
                no_term=args.no_term,
            )
        elif args.owner:
            owner.append(args.owner)
            check_users(owner)
            owner = args.owner
            merge(
                target_user=target_user,
                merge_type='owner',
                owner=owner,
                no_backup=args.no_backup,
                no_term=args.no_term,
            )
        elif args.list:
            for user in args.list.split(' '):
                users.append(user)
                check_users(users)
            merge(
                target_user=target_user,
                users=users,
                merge_type='list',
                no_backup=args.no_backup,
                no_term=args.no_term,
            )
    elif args.revert:
        revert_merge(args.revert)

    else:
        parser.print_help()
        sys.exit(1)


if __name__ == '__main__':
    main()