Skip to content

Commit 504ce9a

Browse files
committed
refactor: ignore empty column
1 parent b7c2b0b commit 504ce9a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

backend/apps/datasource/api/datasource.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -273,7 +273,9 @@ def inner():
273273
for sheet_name in sheet_names:
274274
tableName = f"{sheet_name}_{hashlib.sha256(uuid.uuid4().bytes).hexdigest()[:10]}"
275275
sheets.append({"tableName": tableName, "tableComment": ""})
276-
df = pd.read_excel(save_path, sheet_name=sheet_name, engine='calamine')
276+
df_temp = pd.read_excel(path, nrows=5)
277+
non_empty_cols = df_temp.columns[df_temp.notna().any()].tolist()
278+
df = pd.read_excel(save_path, sheet_name=sheet_name, engine='calamine', usecols=non_empty_cols)
277279
insert_pg(df, tableName, engine)
278280

279281
# os.remove(save_path)

0 commit comments

Comments
 (0)