122 lines
2.7 KiB
Python
Executable File
122 lines
2.7 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
from collections import defaultdict
|
|
from datetime import datetime
|
|
from typing import Annotated, Any
|
|
|
|
import arcticdb as adb
|
|
import pandas as pd
|
|
from fastapi import APIRouter, Path
|
|
from nicegui import app, ui
|
|
|
|
from aggridx import aggridx
|
|
from registry import Registry
|
|
|
|
|
|
PGroup = Annotated[str, Path(pattern=r"^p\d{5}$")]
|
|
|
|
|
|
uri = "lmdb://adb"
|
|
ac = adb.Arctic(uri)
|
|
|
|
lib = ac.get_library(
|
|
"stand",
|
|
create_if_missing=True,
|
|
library_options=adb.LibraryOptions(dynamic_schema=True)
|
|
)
|
|
|
|
|
|
router = APIRouter()
|
|
grids = defaultdict(Registry)
|
|
|
|
|
|
OPTIONS = {
|
|
"context": {"pgroup": None},
|
|
"defaultColDef": {
|
|
"filter": True,
|
|
"editable": True,
|
|
"sortable": True,
|
|
"resizable": True
|
|
},
|
|
"pagination": True,
|
|
"paginationAutoPageSize": True,
|
|
"theme": "balham"
|
|
}
|
|
|
|
|
|
def update_adb(evt):
|
|
if evt.args.get("source") != "edit":
|
|
# ignore event if it was no direct edit
|
|
return
|
|
|
|
pgroup = evt.args["context"]["pgroup"]
|
|
index = evt.args["data"]["index"]
|
|
col_id = evt.args["colId"]
|
|
new_val = evt.args["newValue"]
|
|
|
|
index = datetime.fromisoformat(index) # nicegui converts datetime to str
|
|
df = lib.read(pgroup, date_range=[index]).data
|
|
df.at[index, col_id] = new_val
|
|
lib.update(pgroup, df)
|
|
|
|
|
|
def update_grids(evt):
|
|
sender = evt.sender
|
|
pgroup = evt.args["context"]["pgroup"]
|
|
for grid in grids[pgroup]:
|
|
grid.set_cell_server(evt)
|
|
if grid == sender:
|
|
# the sender is already up-to-date
|
|
continue
|
|
grid.set_cell_client(evt)
|
|
|
|
|
|
@ui.page("/{pgroup}")
|
|
def page(pgroup: PGroup):
|
|
# with ui.left_drawer() as ld:
|
|
# ld.hide()
|
|
# dark = ui.dark_mode()
|
|
# ui.switch("dark mode").bind_value(dark)
|
|
|
|
try:
|
|
df = lib.read(pgroup).data
|
|
except adb.exceptions.NoSuchVersionException:
|
|
df = pd.DataFrame()
|
|
|
|
df = df.reset_index()
|
|
|
|
options = OPTIONS.copy()
|
|
options["context"]["pgroup"] = pgroup
|
|
|
|
grid = aggridx.from_pandas(df, options=options)
|
|
grid.classes("h-[calc(100vh-2rem)]") # full height minus padding
|
|
grid.on("cellValueChanged", update_adb)
|
|
grid.on("cellValueChanged", update_grids)
|
|
grids[pgroup].add(grid)
|
|
|
|
|
|
@router.post("/{pgroup}/append")
|
|
def append(pgroup: PGroup, row: dict[str, Any]):
|
|
now = datetime.now()
|
|
|
|
df = pd.DataFrame(row, index=[now])
|
|
lib.append(pgroup, df)
|
|
|
|
now = str(now) # nicegui converts datetime to str
|
|
row = {"index": now, **row} # setdefault would not force index to be the first column
|
|
|
|
res = []
|
|
for grid in grids[pgroup]:
|
|
grid.append(row)
|
|
res.append(grid.options)
|
|
|
|
return res
|
|
|
|
|
|
app.include_router(router)
|
|
|
|
ui.run(title="stand", favicon="icon.png", fastapi_docs=True, dark=True, show=False)
|
|
|
|
|
|
|