staff,kind,date,shift
看護師A,交代制,2026-11-01,日
看護師A,交代制,2026-11-02,準
看護師A,交代制,2026-11-03,休

import anthropic
from pathlib import Path
 
client = anthropic.Anthropic()
f = client.files.upload(file=Path("roster.csv"))
 
prompt = """添付の勤務表CSVを点検してください。計算は必ずコード実行で行います。
 
規則:
- 夜勤時間帯は16:30から翌8:30の16時間
- 日勤8:30-16:30、準夜16:30-翌1:00、深夜0:30-8:30
- 夜勤時間は、各勤務と夜勤時間帯が重なった時間
- 区分が夜勤専従の人と、月の夜勤時間が16時間未満の人は、
  人数にも延べ時間にも入れない
- 月平均の夜勤時間数 = 延べ夜勤時間 / 夜勤従事者数
- 夜勤が連続する暦日数と、勤務終了から次の勤務開始までの
  間隔が11時間未満の回数も、職員ごとに出す
 
職員ごとの表と月平均を示し、実行したコードを添えてください。"""
 
resp = client.messages.create(
    model="claude-opus-5-5",
    max_tokens=4096,
    messages=[{"role": "user", "content": [
        {"type": "text", "text": prompt},
        {"type": "container_upload", "file_id": f.id},
    ]}],
    tools=[{"type": "code_execution_20250825", "name": "code_execution"}],
)
print(resp.to_json())

import csv, sys
from collections import defaultdict
from datetime import datetime, timedelta
 
SHIFTS = {            # 勤務 -> (開始, 終了)  終了が開始より前なら翌日
    "日": ("08:30", "16:30"),
    "準": ("16:30", "01:00"),
    "深": ("00:30", "08:30"),
}
WIN_START, WIN_HOURS = "16:30", 16          # 夜勤時間帯 16:30-翌08:30(16時間)
MIN_HOURS, MAX_AVG = 16, 72                 # 月16時間未満は除外 / 月平均72時間以下
HANDOVER = {"準": 0.5}                      # 申し送り(時間)。除外する運用のときだけ使う
 
def span(day, key):
    s, e = SHIFTS[key]
    d = datetime.strptime(day, "%Y-%m-%d")
    st = datetime.combine(d, datetime.strptime(s, "%H:%M").time())
    en = datetime.combine(d, datetime.strptime(e, "%H:%M").time())
    return st, en + timedelta(days=1) if en <= st else en
 
def overlap(a, b, c, d):
    return max(timedelta(0), min(b, d) - max(a, c)).total_seconds() / 3600
 
def night_hours(day, key, exclude_handover):
    st, en = span(day, key)
    h = 0.0
    for off in (-1, 0, 1):   # 暦日をまたぐので前日・当日・翌日の窓を見る
        w0 = datetime.strptime(day, "%Y-%m-%d") + timedelta(days=off)
        w0 = w0.replace(hour=int(WIN_START[:2]), minute=int(WIN_START[3:]))
        h += overlap(st, en, w0, w0 + timedelta(hours=WIN_HOURS))
    if exclude_handover:
        h -= HANDOVER.get(key, 0)
    return h
 
def main(path, exclude_handover=False):
    staff = defaultdict(list); kind = {}
    for r in csv.DictReader(open(path, encoding="utf-8")):
        kind[r["staff"]] = r["kind"]
        if r["shift"] in SHIFTS:
            staff[r["staff"]].append((r["date"], r["shift"]))
    total = people = 0; naive_t = naive_p = 0
    print(f"{'職員':<7}{'区分':<8}{'夜勤時間':>8}  対象  メモ")
    for s in sorted(kind):
        recs = sorted(staff[s], key=lambda x: span(*x)[0])
        h = sum(night_hours(d, k, exclude_handover) for d, k in recs)
        # 平均の計算対象か
        counted = kind[s] != "夜勤専従" and h >= MIN_HOURS
        notes = []
        # 勤務間隔(前の勤務の終了から次の開始まで)
        short = []
        for (d0, k0), (d1, k1) in zip(recs, recs[1:]):
            gap = (span(d1, k1)[0] - span(d0, k0)[1]).total_seconds() / 3600
            if gap < 11:
                short.append(d1[5:])
        # 連続夜勤(夜勤を含む勤務が連続する暦日の数)
        nd = sorted({datetime.strptime(d, "%Y-%m-%d") for d, k in recs if night_hours(d, k, False) > 0})
        best = run = 0; last = None
        for d in nd:
            run = run + 1 if last and (d - last).days == 1 else 1
            best = max(best, run); last = d
        if short: notes.append(f"間隔11h未満{len(short)}回")
        if best >= 3: notes.append(f"夜勤{best}連続")
        if not counted: notes.append("平均から除外")
        print(f"{s:<7}{kind[s]:<8}{h:>8.1f}  {'○' if counted else '-':^4}  {' '.join(notes)}")
        if counted: total += h; people += 1
        if kind[s] != "夜勤専従": naive_t += h; naive_p += 1
    avg = total / people
    print(f"\n月延べ夜勤時間 {total:.1f} / 夜勤従事者 {people} 人 = 月平均 {avg:.1f} 時間 ({'基準内' if avg <= MAX_AVG else '72時間超'})")
    print(f"(参考)16時間未満の職員も含めた場合: {naive_t/naive_p:.1f} 時間")
 
main(sys.argv[1], "--handover" in sys.argv)