【Streamlit v1.46.1】認証モジュール + SQLite3を使ってログイン機能を実装

"""
Streamlit + SQlite3 + streamlit_authenticator
によるログイン/サインアップ機能
"""
#  v1.46.1
import streamlit as st
# ↓画像を扱いたい方は以下もimportすると便利です。
from PIL import Image
import sqlite3
import streamlit_authenticator as stauth # 追加

# 状態管理
if "page" not in st.session_state:
    st.session_state["page"] = "login"

if "mypage" not in st.session_state:
    st.session_state["mypage"] = "Home"    

# 画面遷移先
def Home():
    st.header('ホーム画面')
    img1="./images/公園の河.jpg"
    img2="./images/palazzina寺院.jpg"
    col1,col2 = st.columns(2)
    with col1:
        image1= Image.open(img1)
        st.image(image1)
    with col2:
        image2 = Image.open(img2)
        st.image(image2)

def info():
    st.header('お知らせ')
    st.write("〇〇公園のひまわりが見頃です。")
    st.write("〇〇町の祇園祭が無事終了しました。")

def inquiry():
    st.header('お問い合わせ')


# credentials作成
def fetch_users(conn):
    try:
        cur = conn.cursor()
        cur.execute("select username ,password from users")
        rows = cur.fetchall()

        credentials = {"usernames":{}}
        for row in rows:
            username, hashed_password = row
            tmp_dict = {
                'name': username,   # ここでusernameをnameとして使用
                'password': hashed_password,
            }
            credentials["usernames"][username] =tmp_dict
        conn.close()
        return credentials
    
    except Exception as e:
        st.error(f"ユーザー情報取得エラー: {e}")
        return {"usernames": {}}

# -新規登録
def regist_user(conn, username, hashed_password):
    conn = sqlite3.connect("st-menber.db") # 一旦conn.close()しているので再度接続する!
    try:
        cur = conn.cursor()
        cur.execute("insert into users (username, password) values (?, ?)",(username, hashed_password))
        conn.commit()
        conn.close()
        return True
    except sqlite3.IntegrityError :
        st.warning("このユーザー名は既に登録されています。別のユーザー名をお試しください。")
        return False
    except Exception as e:
        st.error(f"ユーザー登録エラー: {e}")
        return False

 
# --メイン関数--
def main():
    conn = sqlite3.connect("st-menber.db")
    if conn is None:
        st.error("データベースへの接続に失敗しました。アプリケーションを終了します。")
        st.stop() # データベース接続ができない場合はここで停止

    mycredentials = fetch_users(conn)

    # ↓ローカル環境なので直書き
    mysignature_key = 'super_secret123456!#$_verylonglong-cool'
  
    authenticator = stauth.Authenticate(
        mycredentials,
        'streamlit_v1.46_login_cookie',  # 任意のクッキー名
        mysignature_key,                 # 任意の署名キー
        cookie_expiry_days=30,          # クッキーの有効期限（日数）
    )

    # サイドバーの設計
    with st.sidebar:
        if st.session_state["page"]== "login":
            if st.button("新規登録はこちらから"):
                st.session_state["page"] = "signup"
                st.rerun()
            st.divider()
            
            authenticator.login(location="sidebar",fields={
                "Form name": "ログイン",
                "Username": "ユーザー名",
                "Password": "パスワード",
                "Login": "ログイン"
            })
      
        
            if st.session_state["authentication_status"]:
                st.session_state["page"] = "dashboard"
                # print(st.session_state["authentication_status"])
            elif st.session_state["authentication_status"] is False:
                st.error("ユーザー名またはパスワードが間違っています")
            elif st.session_state["authentication_status"] is None:
                st.warning("ユーザー名とパスワードを入力してください")

        elif st.session_state["page"]== "dashboard":
            # st.markdown("ログインに成功しました。  \nようこそ!!")  # OK ※半角スペース２個+\nで改行になる
            if st.session_state["authentication_status"]:
                st.write("ログインに成功しました。")
                st.success(f"ようこそ {st.session_state['name']} さん！")
                st.session_state["page"] = "dashboard"
            dash_page = st.selectbox('選択してください',['Home','お知らせ', 'お問い合わせ'])
            st.session_state["mypage"] = dash_page

            authenticator.logout("ログアウト", location="sidebar")
            if st.session_state["authentication_status"] is None:
                st.session_state["page"] = "login"
                st.warning("ログアウトしました。再度ログインしてください。")
                st.rerun()
        elif st.session_state["page"] == "signup":
            st.header('登録フォーム')
            st.write("アカウントをお持ちでない方はこちらから登録してください。")

            with st.form("register_user_form"):
                username = st.text_input("ユーザー名")
                new_password = st.text_input("パスワード", type="password")
                confirm_password = st.text_input("パスワードの確認", type="password")
                
                submit_registration = st.form_submit_button("登録する")
                if submit_registration:
                    if username and new_password and confirm_password:
                        if new_password == confirm_password :
                            try:
                                hashed_password = stauth.Hasher.hash(new_password)
                                if regist_user(conn, username, hashed_password):
                                    st.session_state["page"] = "login"
                                    st.markdown("<span style='color:blue;'>ご登録ありがとうございました。</span>", unsafe_allow_html=True) 
                                    st.rerun() 

                            except Exception as e:
                                st.error(f"パスワードのハッシュ化または登録中にエラーが発生しました: {e}")
                        else:
                            st.error("パスワードが一致しません。再度確認してください。")
                    else :
                        st.error("全ての登録項目を入力してください。")

            st.divider()
            if st.button("ログイン画面に戻る"):
                st.session_state["page"] = "login"
                st.rerun()


    # メイン画面の設計
    if st.session_state["page"] == "dashboard":
        # 画面遷移
        if st.session_state["mypage"] == "Home":
            Home()
        elif st.session_state["mypage"] == "お知らせ":
            info()
        elif st.session_state["mypage"] == "お問い合わせ":
            inquiry()

    
if __name__ == '__main__':
    main()