Download deployment config after modification

[fuel.git] / deploy / reap.py

diff --git a/deploy/reap.py b/deploy/reap.py
old mode 100755 (executable)
new mode 100644 (file)
index bf64d40..1262d4c
--- a/deploy/reap.py
+++ b/deploy/reap.py
@@ -92,8 +92,10 @@ class Reap(object):
 
     def get_env(self):
         env_list = parse(exec_cmd('fuel env'))
-        if len(env_list) > 1:
-            err('Not exactly one environment')
+        if len(env_list) == 0:
+            err('No environment deployed')
+        elif len(env_list) > 1:
+            err('More than 1 environment deployed')
         self.env = env_list[0]
         self.env_id = self.env[E['id']]
 
@@ -135,9 +137,8 @@ class Reap(object):
 
     def reap_transformation(self, node_id, roles, transformations):
         main_role = 'controller' if 'controller' in roles else 'compute'
-        node_file = glob.glob('%s/deployment_%s/*%s_%s.yaml'
-                              % (self.temp_dir, self.env_id,
-                                 main_role, node_id))
+        node_file = glob.glob('%s/deployment_%s/%s.yaml'
+                              % (self.temp_dir, self.env_id, node_id))
         tr_name = None
         with open(node_file[0]) as f:
             node_config = yaml.load(f)