btt_plot.py: Use `with open() as ...` context manager

to automatically handle close()

Signed-off-by: Vincent Legoll <vincent.legoll@gmail.com>
Signed-off-by: Jens Axboe <axboe@kernel.dk>

diff --git a/btt/btt_plot.py b/btt/btt_plot.py
index b67caaa1d298cbdf4204fec8168dc95ef397b6ff..40bc71ffbe0264aba5932e4a67661fe5b561f695 100755 (executable)
--- a/btt/btt_plot.py
+++ b/btt/btt_plot.py
@@ -168,14 +168,15 @@ def get_data(files):
 
                xs = []
                ys = []
-               for line in open(file, 'r'):
-                       f = line.rstrip().split(None)
-                       if line.find('#') == 0 or len(f) < 2:
-                               continue
-                       (min_x, max_x, x) = check(min_x, max_x, f[0])
-                       (min_y, max_y, y) = check(min_y, max_y, f[1])
-                       xs.append(x)
-                       ys.append(y)
+               with open(file, 'r') as fi:
+                       for line in fi:
+                               f = line.rstrip().split(None)
+                               if line.find('#') == 0 or len(f) < 2:
+                                       continue
+                               (min_x, max_x, x) = check(min_x, max_x, f[0])
+                               (min_y, max_y, y) = check(min_y, max_y, f[1])
+                               xs.append(x)
+                               ys.append(y)
 
                db[file] = {'x':xs, 'y':ys}
                if len(xs) > 10:
@@ -385,11 +386,12 @@ def do_live(files):
        def get_live_data(fn):
                xs = []
                ys = []
-               for line in open(fn, 'r'):
-                       f = line.rstrip().split()
-                       if f[0] != '#' and len(f) == 2:
-                               xs.append(float(f[0]))
-                               ys.append(float(f[1]))
+               with open(fn, 'r') as fi:
+                       for line in fi:
+                               f = line.rstrip().split()
+                               if f[0] != '#' and len(f) == 2:
+                                       xs.append(float(f[0]))
+                                       ys.append(float(f[1]))
                return xs, ys
 
        #----------------------------------------------------------------------