如何确定fabfile中的结构角色以将特定文件发送到主机

Question

我有一个看起来像这样的环境：

env.roledefs = {
    'cisco-collectors': ['hosta', 'hostb', 'hostc'],
    'brocade-collectors': ['hosta', 'hostd']
}

我有一些特定文件需要发送给具有特定角色的主机：

files = {
    'cisco-collectors': ['/path/to/filea', '/path/to/fileb'],
    'brocade-collectors': ['/path/to/filec', '/path/to/filed']
}

我该如何编写sendFiles（）函数，以便在命令行上指定角色时，甚至使用@roles()装饰器也可以获取正确的文件列表？

这个问题显示了一种确定主机是否属于角色的方法，但是我需要获取当前正在执行的角色，以便知道要发送的文件列表。

理想的情况是这样的：

@roles('cisco-collectors', 'brocade-collectors')    
def sendFiles():
  for file in files[env.current_role]:
    put(file)

Answer 1

env.host_string.role包含最新的fabric源中的当前角色（未发布）。

提交

Answer 2

在我的测试中，我认为您无法可靠地获得1.5中的当前角色。 请参阅帕维尔（Pavel）有关此情况将如何变化的答案。 尽管这似乎不方便，但我认为原因是Fabric合并了主机列表，因此角色不会延续。

但是，如果您不需要使用角色功能，则有一个解决方法。 您可以创建一个更改主机列表的任务。 不过，这仅适用于一个角色！

from fabric.api import task, local, env

@task
def role1():
    env.current_role = 'role1'
    env.hosts.extend(['example.com'])

@task
def role2():
    env.current_role = 'role2'
    env.hosts.extend(['test.example.com'])

@task
def test():
    print env.current_role
    print env.hosts

如果现在运行fab role1 test您将看到：

[localhost] Executing task 'test'
role1
['localhost', 'example.com']
[example.com] Executing task 'test'
role1
['localhost', 'example.com']

不完全是您所追求的……但是可能会帮助您找到可行的方法。