#!/opt/cloudlinux/venv/bin/python3 -bb
# -*- coding: utf-8 -*-
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2018 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT
#
"""
User POST modification hook.
This script must be called AFTER different user-related
changes in control panel. See --help and docs for detailed
description of actions and parameters.
"""
import argparse
from clcommon.public_hooks.lib import ModifyUserHook
from clcommon.public_hooks import setup_logger_and_sentry
from clcommon.public_hooks.lib.helpers import valid_name as _valid_name
if __name__ == '__main__':
setup_logger_and_sentry()
parser = argparse.ArgumentParser(description=__doc__)
subparsers = parser.add_subparsers(dest='command')
create = subparsers.add_parser(
'create', help='call this command when new system user has been created')
create.add_argument('--username', '-u', required=True, type=_valid_name,
help='Name of a newly created account')
create.add_argument('--owner', '-o', required=True, type=_valid_name,
help='User owner: admin or reseller name')
modify = subparsers.add_parser(
'modify', help='call this command when user account has been modifed')
modify.add_argument('--username', '-u', required=True, type=_valid_name,
help='Name of the account')
modify.add_argument('--new-username', required=False, type=_valid_name,
help='Username after rename')
modify.add_argument('--new-owner', required=False, type=_valid_name,
help='Name of the new vendor of account. '
'Can be admin or reseller.')
restore = subparsers.add_parser(
'restore', help='call this command when user has been restored')
restore.add_argument('--username', '-u', required=True, type=_valid_name,
help='Name of the account')
delete = subparsers.add_parser(
'delete', help='call this command when user has been deleted')
delete.add_argument('--username', '-u', required=True, type=_valid_name,
help='Name of a deleted account')
suspend = subparsers.add_parser(
'suspend', help='call this command when user has been suspended')
suspend.add_argument('--username', '-u', required=True, type=_valid_name,
help='Name of a suspended account')
unsuspend = subparsers.add_parser(
'unsuspend', help='call this command when user has been unsuspend')
unsuspend.add_argument('--username', '-u', required=True, type=_valid_name,
help='Name of a unsuspended account')
args = parser.parse_args()
listener = ModifyUserHook()
if args.command == 'create':
listener.post_create_user(
username=args.username,
owner=args.owner)
elif args.command == 'modify':
listener.post_modify_user(
username=args.username,
new_name=args.new_username,
new_owner=args.new_owner)
elif args.command == 'restore':
listener.post_restore_user(
username=args.username)
elif args.command == 'delete':
listener.post_delete_user(
username=args.username)
elif args.command == 'unsuspend':
listener.post_unsuspend_user(
username=args.username)
elif args.command == 'suspend':
listener.post_suspend_user(
username=args.username
)
else:
raise NotImplementedError
|