Mini Shell

# -*- coding: utf-8 -*-

# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2022 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT
#

import re
import os
from typing import Optional

from clcommon.utils import get_mount_point, run_command, ExternalProgramFailed

USER_QUOTA_ON_PATTERN = re.compile(r'user quota on .+ \(.+\) is on')


def check_quota_enabled(path: str = '/home') -> Optional[str]:
    """
    Check if quotas enabled and initialised for specific path
    :return: string message with giagnostick information if disabled; None if enabled
    """
    if not os.path.isfile('/sbin/quotaon'):
        return 'Package "quota" not installed'

    try:
        mount_point = get_mount_point(path)
    except (OSError, ExternalProgramFailed) as e:
        return f'Error during mount point detection:\n{e}'

    try:
        quotaon_cmd = ['/sbin/quotaon', '-up', mount_point]
        quotaon_output = run_command(quotaon_cmd)
    except ExternalProgramFailed as e:
        quotaon_output = str(e)

    quotaon_output = quotaon_output.strip()
    # detect "user quota on / (/dev/mapper/VolGroup-lv_root) is on"
    if not USER_QUOTA_ON_PATTERN.search(quotaon_output):
        return quotaon_output
    return None