Changing named temporary file to be compatable with windows

ODM2 · Jan 27, 2016 · 271e34d · 271e34d
1 parent b96a20e
commit 271e34d
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 6 deletions.
diff --git a/src/StreamingDataLoader.py b/src/StreamingDataLoader.py
@@ -48,6 +48,8 @@ def main(arguments):
         logger.setLevel(logging.INFO)
 
     # Set up logger to log to a file.
+    if not os.path.exists(LOG_FILENAME):
+        os.makedirs(LOG_FILENAME.strip('logfile.txt'))
     handler = logging.FileHandler(LOG_FILENAME)
 
     # Set up the formatting of the log strings.
@@ -107,10 +109,10 @@ def main(arguments):
                             logger.debug('Writing %s...' % table[0])
                             # If able to save table to database...
                             if dataMapModel.save(table[1]):
-                                print "COLUMNS"
-                                print table[1].columns
-                                print "VDT"
-                                print type(table[1]['ValueDateTime'])
+                                #print "COLUMNS"
+                                #print table[1].columns
+                                #print "VDT"
+                                #print type(table[1]['ValueDateTime'])
                                 # Update the latest date time for the table.
                                 dataMapModel.updateDateTime(table[1]['ResultID'][0],
                                     np.max(table[1]['ValueDateTime'])) 

diff --git a/src/controllers/Database.py b/src/controllers/Database.py
@@ -3,6 +3,7 @@
 
 import sys
 import logging
+from datetime import datetime
 
 
 

diff --git a/src/handlers/csvHandler.py b/src/handlers/csvHandler.py
@@ -79,13 +79,14 @@ def byteReader(self, filepath, start_byte, datecol, header=0, sep=None, dataBegi
                 data = response.read()
                 #filepath = StringIO(data)
 
-                temp = tempfile.NamedTemporaryFile()
+                temp = tempfile.NamedTemporaryFile(mode='w', delete=False)
+                tempName = temp.name
                 try:
                     temp.write(data)
                     temp.seek(0)
                 finally:
                     #print temp.name
-                    df = self.byteReader(temp.name, start_byte, datecol, header, sep, dataBegin)
+                    df = self.byteReader(tempName, start_byte, datecol, header, sep, dataBegin)
                     temp.close()
                     return df
                 #return self.byteReader(filepath, start_byte, datecol, header, sep, dataBegin)