X-Git-Url: https://git.auder.net/?p=talweg.git;a=blobdiff_plain;f=reports%2Fipynb_generator.py;h=456fc22f79775d0c9e56b32cd02532024586ab56;hp=ce546ad471c192f1f9934ced962ca4c3d5596d70;hb=ee8b1b4e3c13f8dcf13a2c8da6a3bef1520c8252;hpb=a866acb3c0ae138b22df9dae9ec576b866794417

diff --git a/reports/ipynb_generator.py b/reports/ipynb_generator.py
index ce546ad..456fc22 100755
--- a/reports/ipynb_generator.py
+++ b/reports/ipynb_generator.py
@@ -123,15 +123,15 @@ def driver():
         inputfile = sys.argv[1]
         with open(inputfile, 'r') as f:
             text = f.read()
-        outputfile = '-' if len(sys.argv) <= 2 else sys.argv[2]
+        # Assuming file extension .gj (generate Jupyter); TODO: less strict
+        outputfile = inputfile[:-3]+'.ipynb' if (len(sys.argv)<=2 or sys.argv[2]=='-') \
+            else sys.argv[2]
     except (IndexError, IOError) as e:
         print('Usage: %s inputfile [outputfile|- [Mako args]]' % (sys.argv[0]))
         print(e)
         sys.exit(1)
     cells = read(text, argv=sys.argv[3:])
     filestr = write(cells)
-    # Assuming file extension .gj (generate Jupyter); TODO: less strict
-    outputfile = inputfile[:-3]+'.ipynb' if outputfile == '-' else outputfile
     with open(outputfile, 'w') as f:
         f.write(filestr)