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)