added login

0_Introduction.py

@@ -1,7 +1,9 @@
 import streamlit as st
 import os
 from utils.levels import complete_level, initialize_level, render_page, get_level
+from utils.login import initialize_login
 
+initialize_login()
 initialize_level()
 
 LEVEL=0

utils/levels.py

@@ -1,4 +1,5 @@
 import streamlit as st
+from utils.login import get_login
 
 def initialize_level():
     if 'level' not in st.session_state:
@@ -8,10 +9,13 @@ def get_level():
     return st.session_state['level']
 
 def render_page(page, level):
-    if st.session_state['level'] < level:
-        st.error(f"You need to complete Level {st.session_state['level']} first!")
+    if get_login()["status"]:
+        if st.session_state['level'] < level:
+            st.error(f"You need to complete Level {st.session_state['level']} first!")
+        else:
+            page()
     else:
-        page()
+        st.error("You need to login first!")
 
 def complete_level(level):
     if st.session_state['level'] > level:

utils/login.py

@@ -0,0 +1,26 @@
+import streamlit as st
+import os
+
+def initialize_login():
+    if 'login' not in st.session_state:
+        username = st.text_input('Username')
+        password = st.text_input('Password', type='password')
+        if st.button('Login'):
+            # TODO: replace with actual authorization check
+            authorized = {
+                "status": True,
+                "Name": "John Doe",
+                "username": "johndoe"
+            }
+            if authorized["status"]:
+                st.session_state['login'] = authorized
+                os.makedirs(os.path.join('.sessions', st.session_state["login"]["username"]), exist_ok=True)
+                st.success('Login Successful!')
+                st.experimental_rerun()
+            else:
+                st.error('Invalid username or password')
+    else:
+        st.sidebar.success(f'Hello, {st.session_state["login"]["Name"]}!')
+
+def get_login():
+    return st.session_state.get('login', {"status": False})