diff --git a/utils/createDf.py b/utils/createDf.py index 34e8d56..fa6d1b9 100644 --- a/utils/createDf.py +++ b/utils/createDf.py @@ -1,4 +1,4 @@ -from pandas import read_csv, read_excel +from pandas import read_csv, read_excel,read_json def createDf(requestFiles): if "file" not in requestFiles or not requestFiles["file"]: @@ -11,12 +11,14 @@ def createDf(requestFiles): df = read_excel(file,dtype=str) elif (file.filename.endswith(".ods")): df = read_excel(file,engine="odf",dtype=str) + elif file.filename.endswith(".json"): + df = read_json(file, dtype=str) else: return False,"Please upload valid file format" df = df.applymap(lambda x: x.strip() if isinstance(x,str) else "") df = df.loc[~(df == "").all(axis=1)] - + if not len(df): return False,"Empty file uploaded" return True,df