don't hardcode tmp dir location

cgoldberg · Jan 2, 2016 · c9264d6 · c9264d6
1 parent 15d458d
commit c9264d6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/xvfbwrapper.py b/xvfbwrapper.py
@@ -12,6 +12,7 @@
 import fnmatch
 import random
 import subprocess
+import tempfile
 import time
 
 
@@ -85,7 +86,7 @@ def search_for_free_display(self):
         return display_num
 
     def _lock_files(self):
-        tmpdir = '/tmp'
+        tmpdir = tempfile.gettempdir()
         pattern = '.X*-lock'
         names = fnmatch.filter(os.listdir(tmpdir), pattern)
         ls = [os.path.join(tmpdir, child) for child in names]