Files
dingtalk-workspace-cli/skills/mono/scripts/attendance_team_shift.py
修雨 5ce7cb61b0 feat(skill): dws skill setup with mono/multi modes (#332)
新增 `dws skill setup` 交互式安装命令,支持单产品(mono)和多产品(multi,标记为试验版)两种 skill 布局,并与开源 dws v1.0.30 命令对齐。

主要变更:
- 骨架:skills/ 拆为 skills/mono/ 和 skills/multi/(29 个产品 skill),git rename 保留历史
- 命令:新增 `skill setup` cobra 子命令,支持 --mode、--target、--source、--yes,huh 交互
  - multi 模式支持 -s/--skill、-x/--exclude 按产品选择
  - 覆盖全部 14 个 agent home 目录(与 install.sh 一致)
- 发布:install.sh / install.ps1 / post-goreleaser.sh 改读 skills/mono;dws-skills.zip 同时包含 mono 和 multi 内容;~/.dws 本地缓存
- 内容对齐:删除 8 个仅内部使用的产品 skill、dingtalk-a2a;修正 mono/multi 共 15+ 处命令引用以匹配 OSS dws v1.0.30;chat skill 从 dws-wukong 回移更新
- 漂移清理:物理删除 skill 文档中 OSS 未暴露的命令,true-fail 命令转为 caveat;--help diff 100% 覆盖审计
- 测试:新增 Layer A 静态命令调度验证 + Layer B 端到端 mock 调度验证
- 同步 main:drive 9 个命令同步到 mono/multi skill
2026-05-26 09:17:37 +08:00

90 lines
2.4 KiB
Python

#!/usr/bin/env python3
"""
查询团队成员本周排班和出勤统计
用法:
python attendance_team_shift.py --users userId1,userId2,userId3
python attendance_team_shift.py --users userId1,userId2 \
--from 2026-03-10 --to 2026-03-14
python attendance_team_shift.py --users userId1 --dry-run
"""
import sys
import json
import subprocess
import argparse
from datetime import datetime, timedelta
from typing import List, Any, Optional
def run_dws(
args: List[str], dry_run: bool = False,
) -> Optional[Any]:
cmd = ['dws'] + args
if dry_run:
print(f"[dry-run] {' '.join(cmd)}")
return None
try:
result = subprocess.run(
cmd, capture_output=True, text=True, timeout=60
)
if result.returncode != 0:
print(f"错误:{result.stderr.strip()}", file=sys.stderr)
return None
return json.loads(result.stdout)
except (subprocess.TimeoutExpired, json.JSONDecodeError,
FileNotFoundError) as e:
print(f"错误:{e}", file=sys.stderr)
return None
def get_week_range():
today = datetime.now()
monday = today - timedelta(days=today.weekday())
friday = monday + timedelta(days=4)
return monday.strftime('%Y-%m-%d'), friday.strftime('%Y-%m-%d')
def main():
parser = argparse.ArgumentParser(
description='查询团队成员排班和出勤统计'
)
parser.add_argument(
'--users', required=True, help='用户 ID 列表,逗号分隔'
)
mon, fri = get_week_range()
parser.add_argument('--from', dest='from_date', default=mon)
parser.add_argument('--to', dest='to_date', default=fri)
parser.add_argument('--dry-run', action='store_true')
args = parser.parse_args()
user_count = len(args.users.split(','))
if user_count > 50:
print('错误:最多查询 50 人')
sys.exit(1)
print(f"📊 团队排班查询 ({args.from_date} ~ {args.to_date})")
print(f" 人数: {user_count}")
print('=' * 50)
print('\n🔍 查询排班信息...')
data = run_dws([
'attendance', 'shift', 'list',
'--users', args.users,
'--start', args.from_date,
'--end', args.to_date,
'--format', 'json',
], dry_run=args.dry_run)
if args.dry_run:
return
if not data:
print('未查到排班信息')
return
print(json.dumps(data, ensure_ascii=False, indent=2))
if __name__ == '__main__':
main()