Use try to avoid race condition on file glob (Chia-Network#140)

* Use try to avoid race condition on file glob * initialize used
nuyawktuah · Dec 10, 2020 · 08d153b · 08d153b
1 parent f63fdef
commit 08d153b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/plot-resources.py b/tests/plot-resources.py
@@ -43,8 +43,12 @@ def pollSpace():
 
     tempdirpath = os.getcwd() + "/" + tempdir
     print("Temporary directory path is " + tempdirpath)
+    used = 0
     while bPollSpace:
-        used = get_size(tempdirpath)
+        try:
+            used = get_size(tempdirpath)
+        except used.TempFileDeleted:
+            print("A temp file was deleted while polling. Skipping.")
         if used > pollDisk:
             pollDisk = used