修改打印输出

脚本/开票.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)