Работа с пользовательскими модулями grain, state и execution

# File: my_custom_state_module.py
import salt.exceptions

def enforce_custom_thing(name, foo, bar=True):
    """
    Принудительно установить состояние пользовательского объекта

    Этот модуль состояния выполняет пользовательское действие. Он вызывает
    модуль выполнения my_custom_module для проверки текущего состояния
    системы и выполнения необходимых изменений.

    name
        Объект, к которому нужно что-то сделать
    foo
        Обязательный аргумент
    bar : True
        Аргумент со значением по умолчанию
    """
    ret = {
        "name": name,
        "changes": {},
        "result": False,
        "comment": "",
    }

    # Начинать с базовой проверки ошибок. Все ли переданные параметры имеют смысл
    # и согласуются друг с другом?
    if bar == True and foo.startswith("Foo"):
        raise salt.exceptions.SaltInvocationError(
            'Аргумент "foo" не может начинаться с "Foo", если аргумент "bar" равен True.'
        )

    # Проверить текущее состояние системы. Нужно ли что-то менять?
    current_state = salt["my_custom_state_module.current_state"](name)

    if current_state == foo:
        ret["result"] = True
        ret["comment"] = "Система уже в корректном состоянии"
        return ret

    # Состояние системы нужно изменить. Проверка работы в корректном режиме
    # test=true.
    if opts["test"] == True:
        ret["comment"] = 'Состояние "{0}" будет изменено.'.format(name)
        ret["changes"] = {
            "old": current_state,
            "new": "Описание, разница, что угодно о новом состоянии",
        }

        # Возврат None при работе с test=true.
        ret["result"] = None

        return ret

    # Внесение фактического изменения и возврат результата.
    new_state = salt["my_custom_state_module.change_state"](name, foo)

    ret["comment"] = 'Состояние "{0}" было изменено!'.format(name)

    ret["changes"] = {
        "old": current_state,
        "new": new_state,
    }

    ret["result"] = True

    return ret

import subprocess

def some_function():
    # Инициализация словаря grains
    grains = {}

    # Логика, устанавливающая значения grains
    grains["os_version"] = subprocess.check_output("lsb_release -a | grep 'Description:' | cut -d ':' -f2- | xargs", shell=True).decode('utf-8').strip()
    grains["os_installation_time"] = subprocess.check_output("ls -l --full-time /etc/hostname | awk '{print $6" "$7" "$8}'", shell=True).decode('utf-8').strip()
    grains["os_uptime"] = subprocess.check_output("uptime --pretty", shell=True).decode('utf-8').strip()

    # Возврат словаря
    return grains

'''
Возвращает список директорий, к которым больше не привязаны пользователи.

Пример использования в CLI::

    salt '*' my_custom_execution_module.get_orphan_home_dirs
'''
def get_orphan_home_dirs():
    ret = dict()  # Словарь для хранения результатов
    root = '/home'  # Корневая директория для поиска
    dirs = os.listdir(root)  # Получаем список директорий в корне

    for dir_ in [d for d in dirs if os.path.isdir(os.path.join(root, d))]:  # Проходим по всем директориям
        found = False  # Флаг для отслеживания найденного пользователя
        for p in [u for u in pwd.getpwall() if (u.pw_uid >= 1000)]:  # Игнорируем системных пользователей
            if p.pw_dir == os.path.join(root, dir_):  # Проверяем, существует ли пользователь с данной директорией
                found = True
                break

        if not found:  # Если пользователь не найден
            ret[root + '/' + dir_] = 'Пользователь {} не найден!'.format(dir_)  # Добавляем результат в словарь

    return ret  # Возвращаем результаты