diff --git a/utils/createDf.py b/utils/createDf.py index 337c68b..e966a98 100644 --- a/utils/createDf.py +++ b/utils/createDf.py @@ -1,24 +1,25 @@ from pandas import read_csv, read_excel, read_json + def createDf(requestFiles): if "file" not in requestFiles or not requestFiles["file"]: - return False,"No file uploaded" - file=requestFiles["file"] + return False, "No file uploaded" + file = requestFiles["file"] - if (file.filename.endswith(".csv")): - df = read_csv(file,dtype=str) - elif (file.filename.endswith(".xlsx") or file.filename.endswith(".xls")): - 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) + if file.filename.endswith(".csv"): + df = read_csv(file, dtype=str) + elif file.filename.endswith(".xlsx") or file.filename.endswith(".xls"): + 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, encoding="utf-8-sig") else: - return False,"Please upload valid file format" + return False, "Please upload valid file format" - df = df.map(lambda x: x.strip() if isinstance(x,str) else "") + df = df.map(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 + return False, "Empty file uploaded" + return True, df