Skip to content
Snippets Groups Projects
Commit 1cb548a6 authored by Recolic Keghart's avatar Recolic Keghart
Browse files

reject main input

parent 9f029338
No related branches found
No related tags found
No related merge requests found
...@@ -18,6 +18,15 @@ import material ...@@ -18,6 +18,15 @@ import material
import csv_preprocess, xlsx_conv import csv_preprocess, xlsx_conv
import numpy as np import numpy as np
from io import StringIO 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(): def _main():
if len(sys.argv) < 2: if len(sys.argv) < 2:
...@@ -40,7 +49,7 @@ def _main(): ...@@ -40,7 +49,7 @@ def _main():
fcontent = xlsx_conv.read_as_csv(fname) fcontent = xlsx_conv.read_as_csv(fname)
fcontent = csv_preprocess.clean_csv(fcontent) fcontent = csv_preprocess.clean_csv(fcontent)
if fcontent.strip() == '': if fcontent.strip() == '':
raise RuntimeError('Empty input rejected.') _dirty_func_return_empty_sheet(fname)
fcontent = csv_preprocess.clean_csv_2(fcontent) fcontent = csv_preprocess.clean_csv_2(fcontent)
#contArr = np.loadtxt(StringIO(fcontent), delimiter=',', dtype=str) #contArr = np.loadtxt(StringIO(fcontent), delimiter=',', dtype=str)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment