From 1cb548a6a08a2425d6b11f01527246cfb212efda Mon Sep 17 00:00:00 2001 From: Recolic Keghart <root@recolic.net> Date: Sat, 22 Feb 2020 05:30:52 -0800 Subject: [PATCH] reject main input --- main.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 7a6f4a8..9ae9582 100755 --- a/main.py +++ b/main.py @@ -18,6 +18,15 @@ import material import csv_preprocess, xlsx_conv import numpy as np from io import StringIO + +def _dirty_func_return_empty_sheet(fname): + materialFileName = fname[:-4] + '-(ææ–™è¡¨).csv' + partFileName = fname[:-4] + '-(下料表).csv' + with open(materialFileName, 'w+') as f: + f.write('空表') + with open(partFileName, 'w+') as f: + f.write('空表') + exit(0) def _main(): if len(sys.argv) < 2: @@ -40,7 +49,7 @@ def _main(): fcontent = xlsx_conv.read_as_csv(fname) fcontent = csv_preprocess.clean_csv(fcontent) if fcontent.strip() == '': - raise RuntimeError('Empty input rejected.') + _dirty_func_return_empty_sheet(fname) fcontent = csv_preprocess.clean_csv_2(fcontent) #contArr = np.loadtxt(StringIO(fcontent), delimiter=',', dtype=str) -- GitLab