Operation   : Linux premium131.web-hosting.com 4.18.0-553.44.1.lve.el8.x86_64 #1 SMP Thu Mar 13 14:29:12 UTC 2025 x86_64
Software     : Apache
Server IP    : 162.0.232.56 | Your IP: 216.73.216.111
Domains      : 1034 Domain(s)
Permission   : [ 0755 ]

#!/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)