0

我正在使用fabric 编写我的rsync 包装器，变量env.host_string 将由execute() 设置以运行任务。要获取 env.host_string，我首先运行（'pwd'），然后运行 ​​rsync。

是否可以在某个检查点到达之前确保用户设置 env.hosts，例如条件 src == env.host_string ？

from fabric.api import run, env, task, abort
from string import Template

@task
def sync_to_same_dir(src, path):
    env.host_string
    cmd_template = Template("""rsync --dry-run -e ssh -avr $user@$src_host:$path/ $path/""")
    if path.endswith("/"):
        path = path[:-1]

    run("pwd") # it's a work around
    if src == env.host_string:
        abort("cannot rysnc to the same host")
    cmd = cmd_template.substitute(path=path, user=env.user, src_host=src)
    run(cmd)

我从fabric的源代码中找到了答案。有一个简单的想法：如何根据需要运行检查主机？

@needs_host
def put(local_path=None, remote_path=None, use_sudo=False,
    mirror_local_mode=False, mode=None):
    """
    Upload one or more files to a remote host.
    """

当用户没有分配任何主机时，我会跟踪需求主机，它会提示询问主机：

No hosts found. Please specify (single) host string for connection:

我们可以将任务重写为：

from fabric.network import needs_host

@task
@needs_host
def the_task_needs_host(): pass
4

1 回答 1

1

你想做什么？一个任务知道它正在使用什么主机而没有任何其他结构调用：

fab -f host-test.py foo
[98.98.98.98] Executing task 'foo'
98.98.98.98
98.98.98.98

Done.

这是脚本，例如：

#!/user/bin/env python

from fabric.api import *

env.user = 'mgoose'

@task
@hosts("98.98.98.98")
def foo():
    print(env.host)
    print(env.host_string)

所以你不需要做任何特别的事情来知道你的任务在哪个主机上。

于 2012-07-14T20:24:20.177 回答