48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
import csv
|
||
import argparse
|
||
|
||
EXCEL_PATH = r'd:\Cheney\geely-kaipiao\doc\2025-2026团队结算进度.csv'
|
||
|
||
|
||
def read_team_column(start_team: str = None) -> list:
|
||
team_list = []
|
||
try:
|
||
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)
|
||
|
||
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} 未找到")
|
||
return []
|
||
except Exception as e:
|
||
print(f"读取文件时出错: {e}")
|
||
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__':
|
||
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)
|