File size: 2,116 Bytes
d6a789f d6ef789 d6a789f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
import streamlit as st
from streamlit_option_menu import option_menu
import json
from Home import dashboard
import pymongo
from pymongo.mongo_client import MongoClient
uri = "mongodb+srv://new-userr:[email protected]/?retryWrites=true&w=majority"
# Create a new client and connect to the server
client = MongoClient(uri)
# Send a ping to confirm a successful connection
try:
client.admin.command('ping')
print("Pinged your deployment. You successfully connected to MongoDB!")
except Exception as e:
print(e)
def loadfile():
with open("database/users.json") as file:
data = json.load(file)
return data
def savefile(data):
with open("database/users.json", "w") as file:
json.dump(data, file, indent=4)
def login():
st.write("Login")
username = st.text_input("Username")
password = st.text_input("Password", type="password")
if st.button("Login"):
data = loadfile()
if username in data:
if data[username]["password"] == password:
st.success("Logged In as {}".format(username))
st.session_state.user = username
else:
st.error("Wrong Password")
else:
st.error("User not found")
def register():
st.write("Register")
username = st.text_input("Username")
password = st.text_input("Password", type="password")
if st.button("Register"):
data = loadfile()
if username in data:
st.error("User already exists")
else:
data[username] = {}
data[username]["password"] = password
savefile(data)
st.success("User created")
def main():
if 'user' not in st.session_state:
st.session_state.user = None
if st.session_state.user is None:
with st.sidebar:
selected = option_menu(None, ['Login', 'Register'])
if selected == 'Login':
login()
elif selected == 'Register':
register()
else:
dashboard() |