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)