storage: simplify coroutine handling

Suggested by @woju
QubesOS · Jun 23, 2017 · f976f7e · f976f7e
1 parent 9242202
commit f976f7e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/qubes/storage/__init__.py b/qubes/storage/__init__.py
@@ -456,9 +456,8 @@ def clone_volume(self, src_vm, name):
         # clone/import functions may be either synchronous or asynchronous
         # in the later case, we need to wait for them to finish
         if asyncio.iscoroutine(clone_op_ret):
-            self.vm.volumes[name] = yield from clone_op_ret
-        else:
-            self.vm.volumes[name] = clone_op_ret
+            clone_op_ret = yield from clone_op_ret
+        self.vm.volumes[name] = clone_op_ret
         return self.vm.volumes[name]
 
     @asyncio.coroutine