streamlit 和mysql数据库的问题

caeser

1. import streamlit as st
   
2. import mysql.connector
   
3. from mysql.connector import Error
   
4. import streamlit_antd_components as sac
   
5. 
   
6. 
   
7. db_config = {
   
8.     'host': 'localhost',
   
9.     'user': 'root',
   
10.     'password': '791120',
    
11.     'database': 'familyfinance'
    
12.     #'raise_on_warnings': True
    
13. }
    
14. 
    
15. # CSS样式，用于隐藏Streamlit的默认导航菜单、页脚和页眉
    
16. hide_st_style = """
    
17.         <style>
    
18.         #MainMenu {visibility: hidden;}
    
19.         footer {visibility: hidden;}
    
20.         header {visibility: hidden;}
    
21.         #root > div:nth-child(1) > div > div > div > div > section > div {padding-top: 0rem;}
    
22.         </style>
    
23.         """
    
24. st.markdown(hide_st_style, unsafe_allow_html=True)# 应用CSS样式
    
25. 
    
26. def authenticate_user(username, password):
    
27.     try:
    
28.         connection = mysql.connector.connect(**db_config)
    
29.         if connection.is_connected():
    
30.             cursor = connection.cursor()
    
31.             # 编写SQL查询，根据实际情况调整
    
32.             query = "SELECT username, password,name FROM users WHERE username = %s AND password = %s"
    
33.             cursor.execute(query, (username,password))
    
34.             result = cursor.fetchone()
    
35.             # 如果用户名存在且密码匹配，则返回True
    
36.             if result :
    
37.                 return True, result[0],result[2]  # 返回True和用户名
    
38.             else:
    
39.                 return False, None,None
    
40.     except Error as e:
    
41.         st.write(f"Database error: {e}")
    
42.     finally:
    
43.         if connection.is_connected():
    
44.             connection.close()
    
45.     return False, None,None
    
46. 
    
47. # 初始化session状态，用于跟踪用户认证状态和当前登录用户
    
48. if "auth_status" not in st.session_state:
    
49.     st.session_state["auth_status"] = 1  # 默认为未认证状态
    
50. 
    
51. auth_status = st.session_state["auth_status"] # 获取当前认证状态
    
52. 
    
53. if "auth_user" not in st.session_state: # 如果没有设置当前用户，则默认为"系统"
    
54.     st.session_state["auth_user"] = "系统"
    
55. name = st.session_state["auth_user"]  # 获取当前登录用户名
    
56. 
    
57. # 用户登录逻辑
    
58. if auth_status == 1:
    
59.     st.sidebar.title("用户登录") # 设置侧边栏标题
    
60. 
    
61.     # 在侧边栏添加用户名和密码输入框
    
62.     username = st.sidebar.text_input("用户名")
    
63.     password = st.sidebar.text_input("密码", type="password")
    
64. 
    
65.     # 登录按钮的事件处理
    
66.     if st.sidebar.button("登录"):
    
67.         login_success, user_name,user_full_name = authenticate_user(username, password)
    
68.         if login_success:
    
69.             st.success("登录成功！") # 显示登录成功消息
    
70.             st.session_state["auth_status"] = 2 # 更新认证状态为已认证
    
71.             st.session_state["auth_user"] = user_name # 更新当前用户
    
72.             user_tag = sac.Tag(user_full_name)  # 使用用户姓名作为tag
    
73.             st.rerun() # 重新运行应用以刷新界面
    
74.         else:
    
75.             st.error("用户名或密码错误！") # 显示错误消息
    
76.             st.session_state["auth_status"] = 1 # 重置认证状态
    
77.                
    
78. # 用户已登录，显示侧边栏菜单
    
79. if auth_status == 2:
    
80.     name = st.session_state["auth_user"]
    
81.     with st.sidebar.container():
    
82.         # 创建标签
    
83.         user_tag = sac.Tag(name, color='orange', bordered=False)
    
84.         quit = sac.Tag('注销', color='red', bordered=False)
    
85.   
    
86.         # 目录
    
87.         menu = sac.menu(
    
88.             items=[
    
89.                 sac.MenuItem('当前登陆', icon="list",disabled=True, tag=user_tag),
    
90.                 sac.MenuItem('主页', icon="house-fill"),
    
91.                 sac.MenuItem('Dashboard', icon="speedometer"),  
    
92.                 sac.MenuItem('图例', icon='pie-chart', type='divider', children=[
    
93.                     sac.MenuItem('柱形图', icon='1-circle'),
    
94.                     sac.MenuItem('仪表盘', icon='2-circle'),
    
95.                 ]),
    
96.                 sac.MenuItem('退出登录', icon="x-circle-fill",tag=quit),        
    
97.                 #sac.MenuItem(type="divider"),            
    
98.                 sac.MenuItem('相关库',icon='megaphone',type='group', children=[
    
99.                     sac.MenuItem('StreamlitAntd', icon='github', disabled=False, href='https://nicedouble-streamlitantdcomponentsdemo-app-middmy.streamlit.app/'),
    
100.                     sac.MenuItem('Echarts', icon='globe-central-south-asia',disabled=False, href='https://echarts.apache.org/examples/zh/index.html#chart-type-line'),
     
101.                     sac.MenuItem('Pyecharts', icon='minecart-loaded', disabled=False, href='https://pyecharts.org/#/zh-cn/intro')                    
     
102.                 ])
     
103.             ],
     
104.             index=1,
     
105.             key='menu',
     
106.             open_all=False, indent=20,
     
107.             format_func='title'
     
108.         )
     
109. 
     
110.         #sac.divider('作者公众号',icon='envelope')
     
111.         #st.image('D:/houtu1.png',width=100)
     
112. 
     
113.     if menu == '主页':
     
114.         st.write("你自己的界面！！！")
     
115. 
     
116.     elif menu == '退出登录':
     
117.         with st.spinner("加载中……"):
     
118.             st.session_state["auth_status"] = 1
     
119.             st.rerun()
     
120.             

复制代码

图片红框中的admin 是数据库表中字段username数据，但我希望显示的是字段name的数据- 管理员，在代码中 我其实也定义了的 user_tag = sac.Tag(user_full_name)  # 指定name赋值给user_tag ，但程序运行后还是不行，不知道哪里出错了？求助各位老师和大侠,谢谢！附件为mysql数据库

阿奇_o

解决：

1.     # 登录按钮的事件处理
   
2.     if st.sidebar.button("登录"):
   
3.         login_success, user_name, user_fullname = authenticate_user(username, password)
   
4.         if login_success:
   
5.             st.success("登录成功！") # 显示登录成功消息
   
6.             st.session_state["auth_status"] = 2 # 更新认证状态为已认证
   
7.             st.session_state["auth_user"] = user_name # 更新当前用户
   
8.             # user_tag = sac.Tag(user_fullname)  # 使用用户姓名作为tag  (错误！！因为会被)
   
9.             # 正确做法：可以保存到会话状态里（以供后面“已登录”的状态下使用）
   
10.             st.session_state['user_fullname'] = user_fullname  
    
11.             st.rerun() # 重新运行应用以刷新界面（进入到“已登录”的状态）
    
12. 
    
13. # 用户已登录，显示侧边栏菜单
    
14. if auth_status == 2:
    
15.     name = st.session_state["auth_user"]
    
16.     user_fullname = st.session_state['user_fullname']
    
17.     with st.sidebar.container():
    
18.         # 创建标签
    
19.         # user_tag = sac.Tag(name, color='orange', bordered=False)
    
20.         user_tag = sac.Tag(user_fullname, color='orange', bordered=False)
    
21.         quit = sac.Tag('注销', color='red', bordered=False)
    

复制代码

ps: 小问题：connection的作用域，要改一下。

xutao1999

来了

caeser

阿奇_o 发表于 2024-6-27 18:33
解决：

ps: 小问题：connection的作用域，要改一下。

谢谢你的思路，把st.session_state["auth_user"] = user_name 改成st.session_state["auth_user"] = user_full_name就行了。谢谢