97 lines
4.2 KiB
Python
97 lines
4.2 KiB
Python
import streamlit as st
|
|
|
|
import sql
|
|
from enums import CounterType
|
|
|
|
@st.dialog("Add New Counter", icon=":material/add_box:")
|
|
def _add_counter():
|
|
colors = sql.get_colors(1)
|
|
with st.form(key="add_counter", border=False, clear_on_submit=True):
|
|
title = st.text_input("Title:")
|
|
counter_type_name = st.selectbox("Type", options=[e.name for e in CounterType])
|
|
color = st.radio("Color",
|
|
key="color-selector",
|
|
width="stretch",
|
|
options=[colors[key][0] for key in colors],
|
|
format_func=lambda c: f"#{c}")
|
|
with st.container(horizontal=True, width="stretch", horizontal_alignment="center"):
|
|
if st.form_submit_button(label="Create", icon=":material/save:"):
|
|
sql.create_counter(title, CounterType[counter_type_name], color)
|
|
st.rerun()
|
|
|
|
|
|
@st.dialog("Remove Counter", icon=":material/delete:")
|
|
def _remove_counter(counter_id:int):
|
|
with st.form(key="remove_counter", border=False, clear_on_submit=True):
|
|
st.subheader("Are you sure?")
|
|
with st.container(horizontal=True, width="stretch", horizontal_alignment="center"):
|
|
if st.form_submit_button("Confirm", icon=":material/delete:"):
|
|
sql.remove_counter(counter_id)
|
|
st.rerun()
|
|
|
|
df = sql.get_counters()
|
|
|
|
with st.container(key="counter-table"):
|
|
for counter_id, name, counter_type_str, color in zip(df['id'], df['name'], df['type'], df['color']):
|
|
with st.container(width="stretch", key=f"counter_{counter_id}"):
|
|
with st.container(horizontal=True, width="stretch"):
|
|
st.header(f":material/calendar_clock: {name}", width="stretch")
|
|
|
|
if st.button("", icon=":material/exposure_plus_1:", key=f"increment_counter_{counter_id}"):
|
|
sql.increment_counter(counter_id)
|
|
st.rerun()
|
|
|
|
if st.button("", icon=":material/delete_forever:", key=f"remove_counter_{counter_id}"):
|
|
_remove_counter(counter_id)
|
|
|
|
with st.container(horizontal=True, width="stretch"):
|
|
|
|
counter_type = CounterType(counter_type_str)
|
|
stats_current_unit = counter_type.current_unit_text()
|
|
stats_prev_unit = counter_type.previous_unit_text()
|
|
match counter_type:
|
|
case CounterType.DAILY.value | CounterType.SIMPLE.value:
|
|
stats = sql.get_daily_analytics(counter_id)
|
|
stats_current = stats.iloc[0]["count"]
|
|
stats_prev = stats.iloc[1]["count"]
|
|
|
|
case CounterType.WEEKLY.value:
|
|
stats = sql.get_weekly_analytics(counter_id)
|
|
stats_current = stats.iloc[-1]["count"]
|
|
stats_prev = stats.iloc[-2]["count"]
|
|
|
|
case CounterType.MONTHLY.value:
|
|
stats = sql.get_monthly_analytics(counter_id)
|
|
stats_current = stats.iloc[-1]["count"]
|
|
stats_prev = stats.iloc[-2]["count"]
|
|
|
|
case CounterType.YEARLY.value:
|
|
stats = sql.get_yearly_analytics(counter_id)
|
|
stats_current = stats.iloc[-1]["count"]
|
|
stats_prev = stats.iloc[-2]["count"]
|
|
|
|
if counter_type is CounterType.SIMPLE.value:
|
|
st.markdown(f"**{stats_current} {stats_current_unit}**")
|
|
else:
|
|
st.markdown(f"""
|
|
**{stats_current} {stats_current_unit}**
|
|
*{stats_prev} {stats_prev_unit}*
|
|
""")
|
|
|
|
with st.container(horizontal=True, width="stretch", horizontal_alignment="right"):
|
|
st.page_link("pages/stats.py", icon=":material/bar_chart:", icon_position="right", label="", query_params={"counter_id": str(counter_id)})
|
|
|
|
st.html(f"""
|
|
<style>
|
|
div:has(> .st-key-counter_{counter_id}) {{
|
|
background-color: {color};
|
|
}}
|
|
</style>
|
|
""")
|
|
|
|
if st.button("Add Counter", width="stretch", icon=":material/add_box:"):
|
|
_add_counter()
|
|
|
|
|
|
|