Merge branch 'tests'

[nepi.git] / examples / linux / netcat_file_transfer.py

diff --git a/examples/linux/netcat_file_transfer.py b/examples/linux/netcat_file_transfer.py
index 0560337..6d43764 100644 (file)
--- a/examples/linux/netcat_file_transfer.py
+++ b/examples/linux/netcat_file_transfer.py
@@ -24,6 +24,8 @@
 # $ cd <path-to-nepi>
 # python examples/linux/netcat_file_transfer.py -a <hostname1> -b <hostname2> -u <username> -i <ssh-key>
 
+from __future__ import print_function
+
 from nepi.execution.ec import ExperimentController
 from nepi.execution.resource import ResourceAction, ResourceState
 
@@ -129,14 +131,12 @@ bw = ec.trace(server, "bw.txt")
 pcap = ec.trace(capture, "file_transfer.pcap")
 
 # Choose a directory to store the traces, example f = open("/home/<user>/bw.txt", "w")
-f = open("bw.txt", "w")
-f.write(bw)
-f.close()
-f = open("video_transfer.pcap", "w")
-f.write(pcap)
-f.close()
+with open("bw.txt", "w") as f:
+    f.write(bw)
+with open("video_transfer.pcap", "w") as f:
+    f.write(pcap)
 
 ec.shutdown()
 
-print "Total bytes transfered saved to bw.txt..."
+print("Total bytes transfered saved to bw.txt...")