Pharmacy Refill

1"""Pharmacy Refill API built with Pixeltable.
2
3    export REFILL_SIGNING_KEY=<any-long-random-string>
4    pxt schema update app.py pharmacy
5    pxt service run app.py pharmacy
6"""
7import pixeltable as pxt
8import pixeltable.functions as pxtf
9from pixeltable.serving import FastAPIRouter
10
11from udfs import receipt_signature, refill_state, status_label
12
13# ---- tables ----
14TableModel = pxt.model_base()
15
16
17class Prescriptions(TableModel, name='prescriptions', has_default_idxs=False):
18    rx_number = pxt.Column(type=pxt.String, primary_key=True)
19    patient_ref: pxt.String          # an opaque reference, not a name
20    drug: pxt.String
21    refills_left: pxt.Int
22    prescriber: pxt.String
23
24    state = refill_state(refills_left)
25
26    __indexes__ = [pxt.BtreeIndex(patient_ref), pxt.BtreeIndex(drug)]
27
28
29class Refills(TableModel, name='refills', has_default_idxs=False):
30    id = pxt.Column(value=pxtf.uuid.uuid7(), primary_key=True)
31    rx_number: pxt.String
32    patient_ref: pxt.String
33    requested_at: pxt.String
34    channel: pxt.String              # counter / app / drive-thru
35    status: pxt.String               # requested / ready / picked-up
36
37    label = status_label(status, channel)
38    receipt_sig = receipt_signature(rx_number, requested_at, status)   # re-signed when status changes
39
40    __indexes__ = [pxt.BtreeIndex(rx_number), pxt.BtreeIndex(status)]
41
42
43# ---- queries ----
44@pxt.query
45def queue(status: str):
46    """Refills in one status, oldest request first (status index)."""
47    return Refills.where(Refills.status == status).select(
48        Refills.id, Refills.rx_number, Refills.requested_at, Refills.label
49    ).order_by(Refills.requested_at)
50
51
52@pxt.query
53def patient_prescriptions(patient_ref: str):
54    """A patient's prescriptions (patient_ref index)."""
55    return Prescriptions.where(Prescriptions.patient_ref == patient_ref).select(
56        Prescriptions.rx_number, Prescriptions.drug, Prescriptions.refills_left, Prescriptions.state
57    ).order_by(Prescriptions.rx_number)
58
59
60# ---- routes ----
61refill_api = FastAPIRouter(name='refill_api')
62refill_api.add_insert_route(
63    Prescriptions, path='/prescriptions',
64    inputs=[Prescriptions.rx_number, Prescriptions.patient_ref, Prescriptions.drug, Prescriptions.refills_left,
65            Prescriptions.prescriber],
66    outputs=[Prescriptions.rx_number, Prescriptions.state],
67)
68refill_api.add_insert_route(
69    Refills, path='/refills',
70    inputs=[Refills.rx_number, Refills.patient_ref, Refills.requested_at, Refills.channel, Refills.status],
71    outputs=[Refills.id, Refills.label, Refills.receipt_sig],
72)
73refill_api.add_update_route(Refills, path='/refills/status', inputs=[Refills.status],
74                            outputs=[Refills.id, Refills.label, Refills.receipt_sig])
75refill_api.add_compute_route(Refills, path='/sign',
76                             inputs=[Refills.rx_number, Refills.requested_at, Refills.status],
77                             outputs=[Refills.receipt_sig])
78refill_api.add_query_route(path='/refills/queue', query=queue, method='get')
79refill_api.add_query_route(path='/patients/prescriptions', query=patient_prescriptions, method='get')