let's make sure we have the update_bats directory

sdahlbac · Nov 1, 2016 · cdb5cf7 · cdb5cf7
1 parent a2514ca
commit cdb5cf7
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/update_data.py b/update_data.py
@@ -20,9 +20,14 @@ def __init__(self):
         self.files = []
 
         self.nfiles = 8
-
+        BATCH_FILE_DIR = 'update_bats'
+        try:
+            os.makedirs(BATCH_FILE_DIR)
+        except OSError:
+            if not os.path.isdir(BATCH_FILE_DIR):
+                raise
         for i in xrange(self.nfiles):
-            self.files.append(open('update_bats/loaddata_%d.bat' % i,'wb'))
+            self.files.append(open(BATCH_FILE_DIR + '/loaddata_%d.bat' % i,'wb'))
 
         self.gridnum = 0
         self.next = None