BiGuan commited on
Commit
7ffe265
·
verified ·
1 Parent(s): df4007a

Update tools/read_excel.py

Browse files
Files changed (1) hide show
  1. tools/read_excel.py +7 -5
tools/read_excel.py CHANGED
@@ -1,13 +1,15 @@
1
- import pandas as pd
2
  import requests
 
3
  from io import BytesIO
4
 
5
  def read_excel_sales(file_url: str) -> str:
6
  try:
7
- r = requests.get(file_url, timeout=10)
8
- # 显式指定 engine='openpyxl'
9
- df = pd.read_excel(BytesIO(r.content), engine='openpyxl')
10
- # 尝试查找销售额列
 
11
  sales_col = None
12
  for col in df.columns:
13
  if 'sales' in str(col).lower() or 'total' in str(col).lower() or 'price' in str(col).lower():
 
1
+ import os
2
  import requests
3
+ import pandas as pd
4
  from io import BytesIO
5
 
6
  def read_excel_sales(file_url: str) -> str:
7
  try:
8
+ headers = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
9
+ response = requests.get(file_url, headers=headers, timeout=10)
10
+ response.raise_for_status()
11
+ df = pd.read_excel(BytesIO(response.content), engine='openpyxl')
12
+ # 查找销售额列
13
  sales_col = None
14
  for col in df.columns:
15
  if 'sales' in str(col).lower() or 'total' in str(col).lower() or 'price' in str(col).lower():