Skip to content

Commit 3f0c463

Browse files
committed
refactor: read excel faster
1 parent a23f283 commit 3f0c463

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

backend/apps/datasource/api/datasource.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -273,9 +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_temp = pd.read_excel(save_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)
276+
# df_temp = pd.read_excel(save_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')
279279
insert_pg(df, tableName, engine)
280280

281281
# os.remove(save_path)

0 commit comments

Comments
 (0)