Trail Permit

1"""Trail Permit API built with Pixeltable.
2
3    pxt schema update app.py trails
4    pxt service run app.py trails
5"""
6import pixeltable as pxt
7import pixeltable.functions as pxtf
8from pixeltable.serving import FastAPIRouter
9
10from udfs import checkin_status, permit_band
11
12# ---- tables ----
13TableModel = pxt.model_base()
14
15
16class Trails(TableModel, name='trails'):
17    code = pxt.Column(type=pxt.String, primary_key=True)
18    name: pxt.String
19    region: pxt.String
20    miles: pxt.Float
21    difficulty: pxt.String
22    open: pxt.Bool
23
24    region_upper = pxtf.string.upper(region)
25    name_lower = pxtf.string.lower(name)
26
27
28class Permits(TableModel, name='permits'):
29    id = pxt.Column(value=pxtf.uuid.uuid7(), primary_key=True)
30    trail_code: pxt.String
31    holder: pxt.String
32    days: pxt.Int
33    party_size: pxt.Int
34    start_date: pxt.String
35    notes: pxt.String | None
36
37    band = permit_band(days, party_size)
38    trail_upper = pxtf.string.upper(trail_code)
39
40
41class Checkins(TableModel, name='checkins'):
42    id = pxt.Column(value=pxtf.uuid.uuid7(), primary_key=True)
43    permit_holder: pxt.String
44    trail_code: pxt.String
45    status: pxt.String          # in / out
46    late_min: pxt.Int
47    gate: pxt.String
48    note: pxt.String | None
49
50    status_label = checkin_status(status, late_min)
51    gate_upper = pxtf.string.upper(gate)
52
53
54# ---- queries ----
55@pxt.query
56def open_trails(region: str):
57    return Trails.where((Trails.region == region) & (Trails.open == True)).select(  # noqa: E712
58        Trails.code, Trails.name, Trails.miles, Trails.difficulty
59    ).order_by(Trails.miles)
60
61
62@pxt.query
63def permits_for(trail_code: str):
64    return Permits.where(Permits.trail_code == trail_code).select(
65        Permits.holder, Permits.start_date, Permits.days, Permits.party_size, Permits.band
66    ).order_by(Permits.start_date)
67
68
69# ---- routes ----
70trails_api = FastAPIRouter(name='trails_api')
71trails_api.add_insert_route(Trails, path='/trails',
72                            inputs=[Trails.code, Trails.name, Trails.region, Trails.miles, Trails.difficulty, Trails.open],
73                            outputs=[Trails.code, Trails.region_upper, Trails.name_lower])
74trails_api.add_insert_route(Permits, path='/permits',
75                            inputs=[Permits.trail_code, Permits.holder, Permits.days, Permits.party_size,
76                                    Permits.start_date, Permits.notes],
77                            outputs=[Permits.id, Permits.band, Permits.trail_upper])
78trails_api.add_insert_route(Checkins, path='/checkins',
79                            inputs=[Checkins.permit_holder, Checkins.trail_code, Checkins.status, Checkins.late_min,
80                                    Checkins.gate, Checkins.note],
81                            outputs=[Checkins.id, Checkins.status_label, Checkins.gate_upper])
82trails_api.add_compute_route(Permits, path='/band', inputs=[Permits.days, Permits.party_size], outputs=[Permits.band])
83trails_api.add_query_route(path='/trails/open', query=open_trails, method='get')
84trails_api.add_query_route(path='/permits/by-trail', query=permits_for, method='get')