Avoid printing incomplete Gcode: write to tmp file first, then rename…

… atomically.
ebisso · Nov 29, 2014 · 9a0739d · 9a0739d
1 parent 74e0860
commit 9a0739d
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/clean_gcode.py b/clean_gcode.py
@@ -93,7 +93,9 @@ def rewrite(infile, outfile, verbose=False):
     if len(sys.argv) < 2:
         sys.exit('usage: clean_gcode.py <filename> [--verbose]')
     infilename = sys.argv[1]
+    tmpfilename = '%s.tmp%s' % os.path.splitext(infilename)
+    with open(infilename, 'r') as infile:
+        with open(tmpfilename, 'w') as tmpfile:
+            rewrite(infile, tmpfile, '--verbose' in sys.argv)
     outfilename = '%s.clean%s' % os.path.splitext(infilename)
-    with open(infilename) as infile:
-        with open(outfilename, 'w') as outfile:
-            rewrite(infile, outfile, '--verbose' in sys.argv)
+    os.rename(tmpfilename, outfilename)