#!/usr/bin/python # coding=utf-8 # Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved # # Licensed under CLOUD LINUX LICENSE AGREEMENT # http://cloudlinux.com/docs/LICENSE.TXT import pwd import subprocess from clcommon.public_hooks.lib import ModifyUserHook from clcommon.lib.cledition import lve_supported_or_exit class LveLifecycleHook(ModifyUserHook): """ Apply limits on user creation and remove limits on user destroy """ LVECTL = '/usr/sbin/lvectl' @classmethod def exec_lvectl_cmd(cls, cmd, uid): """ This function doesn't return! Should be last call in code :param cmd: apply|destroy :param uid: lve id """ return subprocess.call([cls.LVECTL, cmd, str(uid)]) @lve_supported_or_exit def post_create_user(self, username, owner, **kwargs): uid = pwd.getpwnam(username).pw_uid self.exec_lvectl_cmd('apply', uid) @lve_supported_or_exit def pre_delete_user(self, username, **kwargs): uid = pwd.getpwnam(username).pw_uid self.exec_lvectl_cmd('delete', uid) self.exec_lvectl_cmd('destroy', uid)