Current File : /opt/cloudlinux/venv/lib/python3.11/site-packages/lve_utils/hooks//lve_control_hook.py
#!/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)
