Instead of keeping the transfer_service in the handle, extract the
transfer service object in close() using the transfer id.
This will make it easier to extract the code for creating a transfer out
of open().
---
v2v/rhv-upload-plugin.py | 9 ++++++---
1 file changed, 6 insertions(+), 3 deletions(-)
diff --git a/v2v/rhv-upload-plugin.py b/v2v/rhv-upload-plugin.py
index 4e48033e4..3f0ec7b2e 100644
--- a/v2v/rhv-upload-plugin.py
+++ b/v2v/rhv-upload-plugin.py
@@ -206,13 +206,12 @@ def open(readonly):
'can_zero': can_zero,
'connection': connection,
'disk_id': disk.id,
+ 'transfer': transfer,
'failed': False,
'highestwrite': 0,
'http': http,
'needs_auth': needs_auth,
'path': destination_url.path,
- 'transfer': transfer,
- 'transfer_service': transfer_service,
}
def can_trim(h):
@@ -405,7 +404,11 @@ def flush(h):
def close(h):
http = h['http']
connection = h['connection']
- transfer_service = h['transfer_service']
+ transfer = h['transfer']
+
+ transfer_service = (connection.system_service()
+ .image_transfers_service()
+ .image_transfer_service(transfer.id))
# This is sometimes necessary because python doesn't set up
# sys.stderr to be line buffered and so debug, errors or
--
2.21.0