【Python技术】streamlit 量化分析应用增加鉴权模块例子

import streamlit as st
import home
from zhangting import zhangting_lianban
from zhangting import  zhangting_gainian
from zuigaoban import  zuigaoban_zhexian
from jingjia import jingjiafenxi
from gegu import gegu
from jishuzhibiao import dapan
from jishuzhibiao import shenqijiuzhuan
#from jishuzhibiao import jishuzhibiao
from bankuai import bankuai_tonghuashun
from bankuai import bankuai_dongfangcaifu
from  qingxu import  qingxu
from huice import  huice
from zhouqi import all_month
from kaipanla import kaipanla_ticai
from xuanxue import meiriyiji
from news import news
import streamlit as st
import bcrypt
# 添加在原有import之后
# 预存储用户信息（正式环境建议存储在数据库）
ADMIN_USER = {
    "username": "admin",
    "password_hash": bcrypt.hashpw("你的密码".encode(), bcrypt.gensalt())
    # 生成方式：bcrypt.hashpw(b"你的密码", bcrypt.gensalt())
}
# 登录验证函数
def authenticate(username, password):
    if username == ADMIN_USER["username"]:
        return bcrypt.checkpw(password.encode(), ADMIN_USER["password_hash"])
    return False
# 在main函数最前面添加鉴权逻辑
def main():
    # 初始化会话状态
    if 'logged_in' not in st.session_state:
        st.session_state.logged_in = False
    # 未登录时显示登录表单
    if not st.session_state.logged_in:
        st.title("用户登录")
        username = st.text_input("用户名")
        password = st.text_input("密码", type="password")
        if st.button("登录"):
            if authenticate(username, password):
                st.session_state.logged_in = True
                st.rerun()
            else:
                st.error("用户名或密码错误")
        return  # 阻止后续代码执行
    # 原有main函数内容保持不变
    st.sidebar.title("股票分析导航")
    selection = st.sidebar.radio("跳转到", list(PAGES.keys()))
    page = PAGES[selection]
    page.app()
#st.set_page_config(page_title="股票分析应用")
PAGES = {
    "主页": home,
    "涨停分析": zhangting_lianban,
    '涨停概念分析': zhangting_gainian,
    "最高板分析":zuigaoban_zhexian,
    "竞价分析": jingjiafenxi,
    "个股分析": gegu,
    "大盘分析": dapan,
    '大盘情绪': qingxu,
    '同花顺概念板块分析': bankuai_tonghuashun,
    '东方财富概念板块分析': bankuai_dongfangcaifu,
    '开盘啦概念板块分析': kaipanla_ticai,
    '回测': huice,
    #'技术指标': jishuzhibiao,
    '神奇九转': shenqijiuzhuan,
    '新闻联播': news,
    '历年月度分析': all_month,
    '每日宜忌': meiriyiji
}
if __name__ == "__main__":
    main()