Edit: /opt/dedrads/cpmerge (4631B)
#!/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()