whoami7 - Manager
:
/
opt
/
dedrads
/
Upload File:
files >> //opt/dedrads/cpmerge
#!/opt/imh-python/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 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()
Copyright ©2021 || Defacer Indonesia