设置单个Salt状态的umask

58次阅读

问题描述

在使用Salt时遇到了一个问题，他想知道是否有方法可以为pip状态模块设置umask，或者是否有一种更通用的方法可以在运行特定状态模块之前设置umask。在Ansible中，pip模块可以接受umask参数，但在Salt中的pip状态模块中并没有类似的选项。用户使用umask为007来运行salt-call，但需要以更开放的权限安装某些Python包，他想知道是否有方法可以实现这一点。

解决方案

请注意以下操作注意版本差异及修改前做好备份。

方案1

在较新的Salt版本中（3006及以上），umask现在是一个全局状态参数，可以在任何状态中指定。你可以直接在各个状态中设置umask参数来满足你的需求。

方案2

在版本低于3006的情况下，虽然Salt本身没有直接支持在单个状态中设置umask，但你可以通过创建一个自定义状态模块来实现这个目标。以下是一个具体的操作步骤：

创建一个名为umask.py的自定义状态模块文件，放置在Salt状态目录的_states子目录下。例如：/srv/salt/_states/umask.py。
在umask.py文件中，编写一个包装函数wrap，该函数会接受要包装的状态名称、umask值以及其他参数。
在包装函数中，临时更改系统的umask，然后调用原始的状态函数，并在状态函数执行后恢复之前的umask。
在使用pip状态模块时，使用自定义的umask.wrap状态来包装，并传递需要的参数。

以下是具体的示例代码和操作步骤：

创建umask.py文件：

# /srv/salt/_states/umask.py
import os

def wrap(wrapped_state, mask, *args, **kwargs):
    old_umask = None
    try:
        if not __opts__["test"]:
            old_umask = os.umask(int(mask, base=8))
        return __states__[wrapped_state](*args, **kwargs)
    finally:
        if old_umask is not None:
            os.umask(old_umask)

在使用pip模块时，使用umask.wrap状态来包装：

# Your Salt state file
"install python requests package":
  umask.wrap:
    - wrapped_state: pip.installed
    - mask: '022'
    - name: requests

在上述示例中，我们创建了一个自定义状态模块umask.py，其中定义了一个wrap函数来实现umask的临时更改。然后，在需要设置umask的状态中，使用umask.wrap状态来包装原始的状态。

请注意，这个方法在较新的Salt版本中可能已经不再需要，因为umask现在可以在全局状态中直接设置。如果你的Salt版本高于3006，可以直接在各个状态中设置umask参数。

方案3

另一种方法是在低于3006版本的情况下，使用cmd.run状态来运行pip命令，并在命令中设置umask。这可以在运行命令时指定特定的umask值。以下是一个示例：

# Your Salt state file
"install python module":
  cmd.run:
    - name: pip install foo
    - umask: '077'
    - unless:
      - fun: pip.is_installed
        pkgname: foo
"install python module at version":
  cmd.run:
    - name: pip install bar==4.2
    - umask: '077'
    - unless:
      - pip freeze | grep bar==4.2

在上述示例中，我们使用了cmd.run状态来运行pip命令，并通过umask参数设置了所需的umask值。这允许你在安装Python包时设置不同的umask值。