main.py

import logging
import sys
import uuid

import streamlit as st

from todo_organizer.db.db_handler import DBHandler
from todo_organizer.db.schema.todo_item import TodoItem
from todo_organizer.db.schema.todo_list import TodoList
from todo_organizer.logic.logic import Logic
from todo_organizer.logic.priority_updater import PriorityUpdater
from todo_organizer.logic.update_algorithms import UpdateAlgorithms
from todo_organizer.settings.settings import Settings

logger = logging.getLogger(__name__)
settings = Settings()
db_handler = DBHandler(connection_string=settings.db_connection_string)
priority_updater = PriorityUpdater()
logic = Logic(db_handler=db_handler, priority_updater=priority_updater)

selected_list = None
task_items = []


def _save_changes(key):
    priority = st.session_state[f"{key}priority"]
    description = st.session_state[f"{key}description"]
    priority_update_deadline_max_priority = None
    priority_update_deadline = None
    if task.priority_update_algorithm == UpdateAlgorithms.Date:
        priority_update_deadline_max_priority = st.session_state[f"{key}priority_update_deadline_max_priority"]
        priority_update_deadline = st.session_state[f"{key}priority_update_deadline"]

    relevant_item = [x for x in task_items if x.id == key][0]
    relevant_item.priority = priority
    relevant_item.description = description
    relevant_item.priority_update_deadline = priority_update_deadline
    relevant_item.priority_update_deadline_max_priority = priority_update_deadline_max_priority

    logic.upsert_list(selected_list, task_items)


def _delete_list_item(key):
    relevant_item = [x for x in task_items if x.id == key][0]
    logic.delete_item(relevant_item)


def _finish_list_item(key):
    relevant_item = [x for x in task_items if x.id == key][0]
    logic.finish_item(selected_list, relevant_item)


def _add_new_list_item(priority, title, description, priority_update_increment_weight, priority_update_algorithm,
                       priority_target, priority_target_date):
    new_item = TodoItem(
        list_id=selected_list.id,
        priority=priority,
        title=title,
        description=description,
        priority_update_algorithm=priority_update_algorithm.value,
        priority_update_increment_weight=priority_update_increment_weight,
        priority_update_deadline=priority_target_date,
        priority_update_deadline_max_priority=priority_target,
    )
    logic.upsert_list(selected_list, task_items + [new_item])


def _add_new_list(name, max_priority):
    new_list = TodoList(id=str(uuid.uuid4()), name=name, priority_max=max_priority)
    logic.upsert_list(new_list, [])


with st.sidebar:
    selected_list = st.selectbox("TODO Lists", logic.get_all_lists())

    with st.expander("New List", expanded=False):
        with st.form(key="new_list_form"):
            name = st.text_input(label="Name")
            max_priority = st.number_input(
                label="Priority", min_value=0.0, max_value=sys.float_info.max, step=1.0, value=5.0
            )
            submitted = st.form_submit_button("Add")
            if submitted:
                _add_new_list(name, max_priority)
                st.rerun()

    if selected_list:
        st.markdown("""---""")
        with st.expander("New Item", expanded=False):
            priority = st.number_input(
                label="Priority", min_value=0.0, max_value=selected_list.priority_max, step=0.1, value=0.0
            )
            priority_update_algorithm = st.selectbox("Priority update Algorithm", UpdateAlgorithms)
            priority_update_increment_weight = 0
            priority_target = None
            priority_target_date = None

            if priority_update_algorithm == UpdateAlgorithms.Increment:
                priority_update_increment_weight = st.number_input(
                    label="Priority increment value", min_value=0.0, max_value=5.0, step=0.01
                )
            elif priority_update_algorithm == UpdateAlgorithms.Date:
                priority_target = st.number_input(
                    label="Priority target value", min_value=0.0, max_value=5.0, step=0.01, value=5.0
                )
                priority_target_date = st.date_input("Priority target date")

            title = st.text_input(label="Title")
            description = st.text_area(label="Description")
            submitted = st.button(label="Add")
            if submitted:
                _add_new_list_item(
                    priority, title, description, priority_update_increment_weight, priority_update_algorithm,
                    priority_target, priority_target_date
                )

        st.markdown("""---""")
        if st.button(key="delete_list", label="Delete list"):
            logic.delete_item(selected_list)

if selected_list:
    task_items = logic.get_list_items(selected_list)
    st.title(selected_list.name)

# for task in db.items:
for task in task_items:
    # print(task.id)
    st.session_state[f"{task.id}_priority"] = task.priority
    st.session_state[f"{task.id}_description"] = task.description
    with st.expander(task.title, expanded=False):
        new_priority = st.number_input(
            label="Priority",
            min_value=0.0,
            max_value=selected_list.priority_max,
            step=0.1,
            value=st.session_state[f"{task.id}_priority"],
            key=f"{task.id}priority",
            on_change=_save_changes,
            args=(task.id,),
        )
        if task.priority_update_algorithm == UpdateAlgorithms.Date.value:
            priority_target = st.number_input(
                label="Priority target value", min_value=0.0, max_value=selected_list.priority_max, step=0.01,
                value=task.priority_update_deadline_max_priority,
                key=f"{task.id}priority_update_deadline_max_priority", on_change=_save_changes, args=(task.id,)
            )
            priority_target_date = st.date_input("Priority target date", value=task.priority_update_deadline,
                                                 on_change=_save_changes, args=(task.id,),
                                                 key=f"{task.id}priority_update_deadline"
                                                 )
        new_description = st.text_area(
            label="Description",
            value=st.session_state[f"{task.id}_description"],
            key=f"{task.id}description",
            on_change=_save_changes,
            args=(task.id,),
        )
        col1, col2 = st.columns(2)
        with col1:
            st.button(label="Delete", key=f"{task.id}delete", on_click=_delete_list_item, args=(task.id,))
        with col2:
            st.button(label="Finish", key=f"{task.id}finish", on_click=_finish_list_item, args=(task.id,))