Marina Berth Booking

1"""Marina Berth Booking API built with Pixeltable.
2
3    pxt schema update app.py marina
4    pxt service run app.py marina
5"""
6import pixeltable as pxt
7import pixeltable.functions as pxtf
8from pixeltable.serving import FastAPIRouter
9
10from udfs import berth_fee, size_class, slip_label, stay_band
11
12# ---- tables ----
13TableModel = pxt.model_base()
14
15
16class Slips(TableModel, name='slips'):
17    slip_id = pxt.Column(type=pxt.String, primary_key=True)
18    dock: pxt.String
19    length_m: pxt.Float
20    power_amps: pxt.Int
21
22    label = slip_label(dock, slip_id, length_m)
23
24
25class Vessels(TableModel, name='vessels'):
26    id = pxt.Column(value=pxtf.uuid.uuid7(), primary_key=True)
27    name: pxt.String
28    loa_m: pxt.Float
29    owner: pxt.String
30    note_bin: pxt.Binary | None
31
32    size = size_class(loa_m)
33
34
35class Bookings(TableModel, name='bookings'):
36    id = pxt.Column(value=pxtf.uuid.uuid7(), primary_key=True)
37    slip_id: pxt.String
38    vessel_name: pxt.String
39    loa_m: pxt.Float
40    nights: pxt.Int
41
42    fee = berth_fee(loa_m, nights)
43    band = stay_band(nights)
44
45
46# ---- queries ----
47@pxt.query
48def slips_that_fit(loa_m: float):
49    """Slips long enough for a vessel, tightest fit first."""
50    return Slips.where(Slips.length_m >= loa_m).select(Slips.slip_id, Slips.label, Slips.power_amps).order_by(Slips.length_m)
51
52
53@pxt.query
54def slip_bookings(slip_id: str):
55    return Bookings.where(Bookings.slip_id == slip_id).select(
56        Bookings.id, Bookings.vessel_name, Bookings.nights, Bookings.fee, Bookings.band
57    ).order_by(Bookings.vessel_name)
58
59
60# ---- routes ----
61marina_api = FastAPIRouter(name='marina_api')
62marina_api.add_insert_route(Slips, path='/slips', inputs=[Slips.slip_id, Slips.dock, Slips.length_m, Slips.power_amps],
63                            outputs=[Slips.slip_id, Slips.label])
64marina_api.add_insert_route(Vessels, path='/vessels', inputs=[Vessels.name, Vessels.loa_m, Vessels.owner],
65                            outputs=[Vessels.id, Vessels.size])
66marina_api.add_insert_route(Bookings, path='/bookings',
67                            inputs=[Bookings.slip_id, Bookings.vessel_name, Bookings.loa_m, Bookings.nights],
68                            outputs=[Bookings.id, Bookings.fee, Bookings.band])
69marina_api.add_update_route(Bookings, path='/bookings/extend', inputs=[Bookings.nights],
70                            outputs=[Bookings.id, Bookings.nights, Bookings.fee, Bookings.band])
71marina_api.add_delete_route(Bookings, path='/bookings/cancel')
72marina_api.add_compute_route(Bookings, path='/band', inputs=[Bookings.nights], outputs=[Bookings.band])
73marina_api.add_compute_route(Bookings, path='/quote', inputs=[Bookings.loa_m, Bookings.nights],
74                             outputs=[Bookings.fee, Bookings.band])
75marina_api.add_query_route(path='/slips/fit', query=slips_that_fit, method='get')
76marina_api.add_query_route(path='/slips/bookings', query=slip_bookings, method='get')