90 lines
3.6 KiB
Python
90 lines
3.6 KiB
Python
import enum
|
|
import logging
|
|
import streamlit as st
|
|
import json
|
|
import pandas as pd
|
|
|
|
from enums import CounterType
|
|
from enum import StrEnum
|
|
from queries import crud, daily_stats, weekly_stats, monthly_stats, yearly_stats
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
counter_type_names = ([e.name for e in CounterType])
|
|
options = counter_type_names
|
|
options.remove(CounterType.SIMPLE.name)
|
|
|
|
if "counter_id" in st.query_params.keys():
|
|
|
|
'''
|
|
Show specific Counter analytics where the counter id is passed as query parameter "counter_id".
|
|
'''
|
|
|
|
counter_id = int(st.query_params["counter_id"])
|
|
df = crud.get_counter(counter_id)
|
|
|
|
counter_type_id = df['type'] - 1
|
|
counter_type = [e for e in CounterType][counter_type_id]
|
|
counter_color ='#' + df['color']
|
|
|
|
with st.container(horizontal_alignment="right", vertical_alignment="bottom", horizontal=True):
|
|
st.header('Counter: ' + df['name'])
|
|
selection = st.segmented_control("Time Range", options, selection_mode="single", required=True, default=counter_type.name, label_visibility="hidden")
|
|
|
|
match getattr(CounterType, selection):
|
|
case CounterType.DAILY:
|
|
st.bar_chart(daily_stats.get_daily_analytics(counter_id), x="date", y="count", color=counter_color)
|
|
case CounterType.WEEKLY:
|
|
st.bar_chart(weekly_stats.get_weekly_analytics(counter_id), x="week", y="count", color=counter_color)
|
|
case CounterType.MONTHLY:
|
|
st.bar_chart(monthly_stats.get_monthly_analytics(counter_id), x="month", y="count", color=counter_color)
|
|
case CounterType.YEARLY:
|
|
st.bar_chart(yearly_stats.get_yearly_analytics(counter_id), x="year", y="count", color=counter_color)
|
|
case _:
|
|
logger.error(f"Unknown selection: {selection}")
|
|
|
|
|
|
else:
|
|
|
|
'''
|
|
By default, if no counter id is passed then show all counters in a a stacked graph
|
|
'''
|
|
with st.container(horizontal_alignment="right", vertical_alignment="bottom", horizontal=True):
|
|
st.header("Statistics")
|
|
selection = st.segmented_control("Time range", options, selection_mode="single", default=f"{CounterType.DAILY.name}", required=True, label_visibility="hidden")
|
|
|
|
selectedRange = getattr(CounterType, selection)
|
|
match getattr(CounterType, selection):
|
|
case CounterType.DAILY:
|
|
unit = 'date'
|
|
unit_label = 'Date'
|
|
entries = daily_stats.get_all_daily_analytics()
|
|
case CounterType.WEEKLY:
|
|
unit = 'week'
|
|
unit_label ='Week'
|
|
entries = weekly_stats.get_all_weekly_analytics()
|
|
case CounterType.MONTHLY:
|
|
unit = 'month'
|
|
unit_label = 'Month'
|
|
entries = monthly_stats.get_all_monthly_analytics()
|
|
case CounterType.YEARLY:
|
|
unit = 'year'
|
|
unit_label = 'Year'
|
|
entries = yearly_stats.get_all_yearly_analytics()
|
|
case _:
|
|
logger.error(f"Unknown selection: {selection}")
|
|
|
|
entries_norm = pd.json_normalize(entries.counters.apply(json.loads)).fillna(0)
|
|
entries_full = pd.concat([entries, entries_norm], axis=1).drop(['counters'], axis=1)
|
|
|
|
selected_counters = [c for c in entries_full.columns if c != unit]
|
|
all_counters = crud.get_counters()
|
|
|
|
colors = all_counters.loc[all_counters['name'].isin(selected_counters), ["name", "color"]]
|
|
colors.name = colors.name.astype("category")
|
|
colors.name = colors.name.cat.set_categories(selected_counters)
|
|
colors = colors.sort_values(["name"])
|
|
colors = colors.color.apply(lambda c: "#" + c).tolist()
|
|
|
|
st.bar_chart(entries_full, x=unit, x_label=unit_label, y_label="Count", color=colors)
|