1. Mikhail Korobov
  2. fabtest

Commits

Adolfo Fitoria  committed c681f3e

Fixed import errrors

  • Participants
  • Parent commits 73b0900
  • Branches default

Comments (0)

Files changed (1)

File fabtest/utils.py

View file
 from fabric import state
-from fabric.main import get_hosts
-from fabric.network import interpret_host_string
+from fabric.tasks import WrappedCallableTask
 
 class FabricAbortException(Exception):
     pass
 
     # partially implement the logic from fabric.main.main
     state.env.command = command.__name__
-    state.env.all_hosts = hosts = get_hosts(command, None, None)
+    state.env.all_hosts = hosts = WrappedCallableTask(command).get_hosts(*args)
 
     try: # convert fabric.abort() calls to real exceptions
         for host in hosts: