144 lines
3.7 KiB
Python
Executable File
144 lines
3.7 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
from datetime import datetime
|
|
|
|
import arcticdb as adb
|
|
import pandas as pd
|
|
from nicegui import app, ui
|
|
|
|
from api import router as api_router
|
|
|
|
from auth.login import router as login_router
|
|
from auth.logout import logout
|
|
from auth.mw import AuthMiddleware
|
|
from auth.secret import get_secret
|
|
|
|
from aggridx import aggridx
|
|
from singletons import grids, lib, PGroup
|
|
|
|
|
|
app.include_router(login_router)
|
|
app.add_middleware(AuthMiddleware)
|
|
|
|
|
|
OPTIONS = {
|
|
":getRowId": "(params) => params.data.index", # set row ID to index column
|
|
"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"]
|
|
row_id = evt.args["rowId"]
|
|
col_id = evt.args["colId"]
|
|
new_val = evt.args["newValue"]
|
|
|
|
index = datetime.fromisoformat(row_id) # 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"]
|
|
row_index = evt.args["rowIndex"]
|
|
row_id = evt.args["rowId"]
|
|
col_id = evt.args["colId"]
|
|
new_val = evt.args["newValue"]
|
|
|
|
for grid in grids[pgroup]:
|
|
grid.set_cell_server(row_index, col_id, new_val)
|
|
if grid == sender:
|
|
# the sender is already up-to-date
|
|
continue
|
|
grid.set_cell_client(row_id, col_id, new_val)
|
|
|
|
|
|
@ui.page("/")
|
|
def main():
|
|
with ui.column().classes("absolute-center items-center"):
|
|
username = app.storage.user.get("username", "unknown user")
|
|
ui.label(f"Hello {username}!").classes("text-2xl")
|
|
ui.button("log out", icon="logout", on_click=logout)
|
|
|
|
pgroups = app.storage.user.get("pgroups", set())
|
|
ui.select(
|
|
label="pgroup",
|
|
options=sorted(pgroups),
|
|
with_input=True,
|
|
on_change=lambda e: ui.navigate.to(f"/tables/{e.value}")
|
|
)
|
|
|
|
|
|
@ui.page("/tables/{pgroup}")
|
|
def table(pgroup: PGroup):
|
|
pgroups = app.storage.user.get("pgroups", set())
|
|
if pgroup in pgroups:
|
|
table_show(pgroup)
|
|
else:
|
|
table_deny(pgroup)
|
|
|
|
|
|
def table_show(pgroup):
|
|
# with ui.left_drawer(value=False) as ld:
|
|
# dark = ui.dark_mode(value=True)
|
|
# ui.switch("dark mode").bind_value(dark)
|
|
|
|
# with ui.page_sticky(position="left", x_offset=-12.5):
|
|
# def cb():
|
|
# ld.toggle()
|
|
# btn.icon = "sym_o_left_panel_close" if ld.value else "sym_o_left_panel_open"
|
|
# btn = ui.button(icon="sym_o_left_panel_open", on_click=cb).props("flat dense")
|
|
|
|
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)
|
|
|
|
|
|
def table_deny(pgroup):
|
|
with ui.column().classes("absolute-center items-center gap-8"):
|
|
ui.icon("sym_o_front_hand", size="xl")
|
|
ui.label(f"access to {pgroup} denied").classes("text-2xl")
|
|
|
|
|
|
|
|
app.include_router(api_router)
|
|
|
|
ui.run(
|
|
title="stand",
|
|
favicon="assets/favicon.png",
|
|
storage_secret=get_secret(),
|
|
fastapi_docs=True,
|
|
dark=True,
|
|
show=False
|
|
)
|
|
|
|
|
|
|