Archery Range

1"""Archery Lanes scheduled pipeline: migrate legacy JSON sessions, date helpers, alter_computed_column scoring. Tables and @pxt.query live here; UDFs in udfs.py.
2"""
3from __future__ import annotations
4
5
6import pixeltable as pxt
7import pixeltable.functions as pxtf
8from pixeltable.functions import date as pxtd
9
10from udfs import archer_tag, boto3_core_probe, score_band
11
12TableModel = pxt.model_base()
13
14
15class Sessions(TableModel, name='sessions', has_default_idxs=False):
16    """One archery-range scoring session migrated from legacy JSON."""
17
18    session_id = pxt.Column(value=pxtf.uuid.uuid7(), primary_key=True)
19    session_code: pxt.String
20    archer: pxt.String
21    lane: pxt.Int
22    session_date: pxt.Date
23    arrows: pxt.Int
24    hits: pxt.Int
25    x_rings: pxt.Int
26    notes: pxt.String
27
28    # --- fn.date secondary ---
29    session_year = pxtd.year(session_date)
30    session_month = pxtd.month(session_date)
31    session_day = pxtd.day(session_date)
32    session_weekday = pxtd.weekday(session_date)  # Mon=0 .. Sun=6
33    session_iso_weekday = pxtd.isoweekday(session_date)
34    session_iso = pxtd.isoformat(session_date)
35    next_practice = pxtd.add_days(session_date, 7)
36    calendar_label = pxtd.strftime(session_date, '%a %b %d, %Y')
37
38    # --- scoring (v1: hit%; later altered to include X-ring bonus) ---
39    hit_pct = (hits * 100.0) / arrows
40    # Club formula (v2): hit% + 2 per X-ring. pipeline.py alters from v1 hit_pct -> this.
41    score = (hits * 100.0) / arrows + (x_rings * 2.0)
42    band = score_band(hit_pct)
43    tag = archer_tag(archer, lane)
44    boto3_ok = boto3_core_probe()
45
46    __indexes__ = [
47        pxt.BtreeIndex(archer),
48        pxt.BtreeIndex(lane),
49        pxt.BtreeIndex(session_date),
50        pxt.BtreeIndex(band),
51        pxt.BtreeIndex(session_code),
52    ]
53
54
55@pxt.query
56def by_archer(archer: str):
57    """Sessions for one archer, newest first."""
58    return (
59        Sessions.where(Sessions.archer == archer)
60        .select(
61            Sessions.session_code,
62            Sessions.session_date,
63            Sessions.lane,
64            Sessions.hit_pct,
65            Sessions.score,
66            Sessions.band,
67            Sessions.calendar_label,
68            Sessions.next_practice,
69        )
70        .order_by(Sessions.session_date, asc=False)
71    )
72
73
74@pxt.query
75def weekend_club(min_pct: float = 70.0):
76    """Weekend sessions (Sat/Sun) at or above a hit percentage."""
77    return (
78        Sessions.where(
79            (Sessions.session_weekday >= 5) & (Sessions.hit_pct >= min_pct)
80        )
81        .select(
82            Sessions.session_code,
83            Sessions.archer,
84            Sessions.calendar_label,
85            Sessions.hit_pct,
86            Sessions.score,
87            Sessions.band,
88            Sessions.tag,
89        )
90        .order_by(Sessions.hit_pct, asc=False)
91    )
92
93
94@pxt.query
95def lane_day(lane: int, day_iso: str):
96    """Sessions on one lane for a given ISO date string (YYYY-MM-DD)."""
97    return (
98        Sessions.where((Sessions.lane == lane) & (Sessions.session_iso == day_iso))
99        .select(
100            Sessions.session_code,
101            Sessions.archer,
102            Sessions.arrows,
103            Sessions.hits,
104            Sessions.score,
105            Sessions.boto3_ok,
106        )
107        .order_by(Sessions.session_code)
108    )