From 21f5f9667766a98c930154c0cd2a1f74330d167f Mon Sep 17 00:00:00 2001 From: cheney Date: Sun, 24 May 2026 16:43:25 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=89=93=E5=8D=B0=E8=BE=93?= =?UTF-8?q?=E5=87=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- 脚本/开票.py | 27 ++++++++++++++++++++++++--- 1 file changed, 24 insertions(+), 3 deletions(-) diff --git a/脚本/开票.py b/脚本/开票.py index ba7d517..cd1061c 100644 --- a/脚本/开票.py +++ b/脚本/开票.py @@ -1,9 +1,10 @@ import csv +import argparse EXCEL_PATH = r'd:\Cheney\geely-kaipiao\doc\2025-2026团队结算进度.csv' -def read_team_column() -> list: +def read_team_column(start_team: str = None) -> list: team_list = [] try: with open(EXCEL_PATH, 'r', encoding='utf-8-sig') as f: @@ -12,6 +13,14 @@ def read_team_column() -> list: team_value = row.get('团号', '') if team_value and isinstance(team_value, str) and team_value.lower().startswith('team'): team_list.append(team_value) + + if start_team and team_list: + lower_start = start_team.lower() + for i, team in enumerate(team_list): + if team.lower() == lower_start: + return team_list[i:] + print(f"警告:未找到团队 '{start_team}',返回全部团队") + return team_list except FileNotFoundError: print(f"错误:文件 {EXCEL_PATH} 未找到") @@ -21,6 +30,18 @@ def read_team_column() -> list: return [] +def print_list_summary(data: list): + total = len(data) + if total <= 6: + print(f"共 {total} 个团队: {data}") + else: + print(f"共 {total} 个团队: {data[:3]} ... {data[-3:]}") + + if __name__ == '__main__': - result = read_team_column() - print(result) + parser = argparse.ArgumentParser(description='读取CSV文件中的团号列') + parser.add_argument('-s', '--start', type=str, help='从指定的team开始返回(不包含之前的team)') + args = parser.parse_args() + + result = read_team_column(args.start) + print_list_summary(result)