app.py

from modules.config import retrieve_ui_config
from modules.authorisation import handle_login
import streamlit as st
from modules.client import ClientInterface
from modules.nav import SidebarNav
from oasis_data_manager.errors import OasisException
from modules.validation import NameValidation, ValidationError, ValidationGroup
import logging

logger = logging.getLogger(__name__)

st.set_page_config(
        page_title = "OasisLMF",
        layout="centered"
)

SidebarNav()

ui_config = retrieve_ui_config()

if ui_config.skip_login:
    with st.spinner("Loading platform..."):
        client_interface = ClientInterface(username=st.secrets["user"], password=st.secrets["password"])
    st.session_state["client"]  = client_interface.client

cols = st.columns([0.1, 0.8, 0.1])
with cols[1]:
    st.image(image="images/oasis_logo.png")

if "client" in st.session_state:
    st.write("Logged in")
    post_login_page = ui_config.post_login_page
    if post_login_page:
        st.switch_page(post_login_page)
else:
    with st.form("login_form"):
        user = st.text_input("Username", key="username")
        password = st.text_input("Password", key="password", type="password")
        submitted = st.form_submit_button("Login")

    if submitted:
        validations = ValidationGroup()
        validations.add_validation(NameValidation("Username"), user)
        validations.add_validation(NameValidation("Password"), password)

        valid = True
        msg = None
        try:
            validations.validate()
        except ValidationError as e:
            valid = False
            msg = e.message
            logger.error(e)

        if valid:
            try:
                client_interface = ClientInterface(username=user, password=password)
                st.session_state["client"] = client_interface.client
                st.session_state["client_interface"] = client_interface
                st.rerun()
            except OasisException as e:
                st.error("Authentication Failed")
                logger.error(e)
        else:
            st.error(msg)