From 20fd5d19f663fa76bcc7a57ba7328c259f2561fb Mon Sep 17 00:00:00 2001 From: cheney Date: Sun, 24 May 2026 16:38:56 +0800 Subject: [PATCH] =?UTF-8?q?=E6=94=B9=E4=B8=BA=E4=BD=BF=E7=94=A8=20csv?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- 脚本/开票.py | 38 +++++++++++++------------------------- 1 file changed, 13 insertions(+), 25 deletions(-) diff --git a/脚本/开票.py b/脚本/开票.py index b7ac9e2..ba7d517 100644 --- a/脚本/开票.py +++ b/脚本/开票.py @@ -1,35 +1,23 @@ -from openpyxl import load_workbook -from openpyxl.styles.fills import PatternFill +import csv -EXCEL_PATH = r'd:\Cheney\geely-kaipiao\doc\2025-2026团队结算进度.xlsx' +EXCEL_PATH = r'd:\Cheney\geely-kaipiao\doc\2025-2026团队结算进度.csv' def read_team_column() -> list: + team_list = [] try: - wb = load_workbook(EXCEL_PATH, read_only=True) - ws = wb.active - team_list = [] - - header_row = 1 - team_col_index = None - - for col in range(1, ws.max_column + 1): - cell_value = ws.cell(row=header_row, column=col).value - if cell_value == '团号': - team_col_index = col - break - - if team_col_index is None: - return team_list - - for row in range(2, ws.max_row + 1): - cell_value = ws.cell(row=row, column=team_col_index).value - if cell_value and isinstance(cell_value, str) and cell_value.lower().startswith('team'): - team_list.append(cell_value) - + with open(EXCEL_PATH, 'r', encoding='utf-8-sig') as f: + reader = csv.DictReader(f) + for row in reader: + team_value = row.get('团号', '') + if team_value and isinstance(team_value, str) and team_value.lower().startswith('team'): + team_list.append(team_value) return team_list + except FileNotFoundError: + print(f"错误:文件 {EXCEL_PATH} 未找到") + return [] except Exception as e: - print(f"读取Excel时出错: {e}") + print(f"读取文件时出错: {e}") return []