cherry picked cb5d027b about context managers

[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 4f8c74c..e632c44 100644 (file)
--- a/examples/linux/netcat_file_transfer.py
+++ b/examples/linux/netcat_file_transfer.py
@@ -131,12 +131,10 @@ 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()