From 799fcd51404490258a1b4a6436264efe4ec5ffb2 Mon Sep 17 00:00:00 2001
From: Carlo Tibaldi <tlbl@dtu.dk>
Date: Tue, 12 Apr 2016 14:17:43 +0200
Subject: [PATCH] adding parser to generator of dlcs
---
wetb/prepost/GenerateDLCs.py | 16 +++++++++++++---
1 file changed, 13 insertions(+), 3 deletions(-)
diff --git a/wetb/prepost/GenerateDLCs.py b/wetb/prepost/GenerateDLCs.py
index 3bcf9bd..7003a8f 100644
--- a/wetb/prepost/GenerateDLCs.py
+++ b/wetb/prepost/GenerateDLCs.py
@@ -14,6 +14,8 @@ from __future__ import absolute_import
from numpy import floor, arctan, pi
import pandas as pd
import xlrd
+from argparse import ArgumentParser
+import os
def multi_for(iterables):
@@ -230,7 +232,9 @@ class GenerateDLCCases(GeneralDLC):
self.add_formulas(dlc, general_functions)
self.eval_formulas(dlc)
df = pd.DataFrame(dlc)
- df.to_excel(folder+sheet.name+'.xls', index=False)
+ if not os.path.exists(folder):
+ os.makedirs(folder)
+ df.to_excel(os.path.join(folder, sheet.name+'.xls'), index=False)
class RunTest():
@@ -257,6 +261,12 @@ class RunTest():
assert_frame_equal(book1, book2, check_dtype=False)
if __name__ == '__main__':
+
+ parser = ArgumentParser(description = "generator of DLB spreadsheets")
+ parser.add_argument('--master', type=str, default='DLCs.xlsx', action='store',
+ dest='filename', help='Master spreadsheet file')
+ parser.add_argument('--folder', type=str, default='', action='store',
+ dest='folder', help='Destination folder name')
+ opt = parser.parse_args()
DLB = GenerateDLCCases()
- DLB.execute()
- pass
+ DLB.execute(filename=opt.filename, folder=opt.folder)
\ No newline at end of file
--
GitLab