timmy / 使用 Streamlit 建立使用者身份驗證系統

cookie: expiry_days: 30 key: some_signature_key name: my_auth_cookie credentials: usernames: jsmith: email: jsmith@gmail.com failed_login_attempts: 0 # Will be managed automatically logged_in: False # Will be managed automatically name: John Smith password: abc # Will be hashed automatically rbriggs: email: rbriggs@gmail.com failed_login_attempts: 0 # Will be managed automatically logged_in: False # Will be managed automatically name: Rebecca Briggs password: def # Will be hashed automatically pre-authorized: emails: - admin@example.com - manager@example.com

import pretty_errors import streamlit as st import streamlit_authenticator as stauth import yaml from yaml.loader import SafeLoader pretty_errors.configure( line_number_first=True, lines_before=5, lines_after=2, line_color=pretty_errors.RED + "> " + pretty_errors.default_config.line_color, display_locals=True, ) # 從 config.yaml 檔案中讀取設定 try: with open("./config.yaml") as file: config = yaml.load(file, Loader=SafeLoader) except Exception as e: st.error(f"讀取 config.yaml 檔案時發生錯誤: {e}") # 確保配置文件讀取成功並包含必要的鍵 if config and "credentials" in config and "cookie" in config: # 檢查 credentials 結構 try: credentials = config["credentials"] if "usernames" in credentials: for username, details in credentials["usernames"].items(): if not isinstance(details.get("password"), str): raise TypeError(f"使用者 {username} 的密碼不是字串: {details.get('password')}") except TypeError as e: st.error(f"配置文件中的錯誤: {e}") except Exception as e: st.error(f"驗證配置文件時發生未知錯誤: {e}") else: st.error("配置文件不完整或無法讀取") # User Authentication with Authenticator try: authenticator = stauth.Authenticate( config["credentials"], config["cookie"]["name"], config["cookie"]["key"], config["cookie"]["expiry_days"], config["pre-authorized"], ) except Exception as e: st.error(f"初始化 Authenticator 時發生錯誤: {e}") # Attempt User Login with Authenticator try: authenticator.login( fields={ "Form name": "登入", "Username": "使用者名稱", "Password": "密碼", "Login": "登入", }, location="main", ) except Exception as e: st.error(f"登入時發生錯誤: {e}") # 根據驗證的情況執行不同的操作 if st.session_state.get("authentication_status"): authenticator.logout("Logout", "main", key="unique_key") user_info = { "name": st.session_state.get("name"), "username": st.session_state.get("username"), } st.write(user_info) elif st.session_state.get("authentication_status") is False: st.error("使用者名稱/密碼不正確") elif st.session_state.get("authentication_status") is None: st.warning("請輸入你的使用者名稱和密碼") with open("./config.yaml", "w") as file: yaml.dump(config, file, default_flow_style=False)