Personalisieren Sie für jeden Benutzer mit Streamlit in Snowflake (SiS)

Einführung

Ende Juli 2024 wurden die Current_User- und Row Access Policy in Streamlit in Snowflake verfügbar.

Das Spannende an diesem Update ist, dass es jetzt einfach und sicher ist, den mit der Anwendung verbundenen Snowflake-Benutzer zu identifizieren und die Verarbeitung für jeden Benutzer anzupassen.

Ohne einen benutzerdefinierten Anmeldemechanismus oder eine Benutzerverwaltungstabelle einrichten zu müssen, können Sie eine einzelne Anwendung für jeden Benutzer auf folgende Weise personalisieren:

• Ändern der Anwendungsanzeige für jeden Benutzer
• Vorbereitung personalisierter Analyse-Dashboards für jeden Benutzer
• Verwenden von Zeilenzugriffsrichtlinien, um für jeden Benutzer unterschiedliche Abfrageausgabeergebnisse zu erhalten (Enterprise Edition und höher)

In diesem Beitrag erstellen wir eine einfache ToDo-Liste, die individuelle Benutzerinformationen anzeigt.

Hinweis: Dieser Beitrag stellt meine persönlichen Ansichten dar und nicht die von Snowflake.

Funktionsübersicht

Ziele

• Verwalten Sie persönliche ToDo-Listen mit einer einzigen gemeinsamen App
• Verwenden Sie die Zeilenzugriffsrichtlinie, um zu verhindern, dass die ToDos anderer Personen angezeigt werden

Fertiggestelltes Bild

Bildschirm von Benutzer TKANNO

Bildschirm des Benutzers TARO

Voraussetzungen

• Snowflake-Konto
  • Für die Verwendung der Zeilenzugriffsrichtlinie ist ein Enterprise Edition-Konto erforderlich

Notiz

• Streamlit in Snowflake wird mit Besitzerrechten ausgeführt, sodass Current_Role mit der Streamlit in Snowflake-Anwendungsrolle identisch ist. (Daher kann es nicht zur Personalisierung verwendet werden)

Verfahren

Erstellen Sie eine Tabelle zum Speichern der ToDo-Liste

Führen Sie den folgenden Befehl aus einem Arbeitsblatt aus:

-- Create ToDo list table
CREATE TABLE IF NOT EXISTS todo_list (
    id INT AUTOINCREMENT,
    task VARCHAR(255),
    status VARCHAR(20),
    due_date DATE,
    completed_date DATE,
    owner VARCHAR(50)
);

Erstellen Sie eine Zeilenzugriffsrichtlinie

Diese Richtlinie gibt Zeilen zurück, in denen der Besitzer in der todo_list-Tabelle mit dem aktuellen_Benutzer übereinstimmt, der mit der Streamlit in Snowflake-Anwendung verbunden ist.

Führen Sie den folgenden Befehl aus dem Arbeitsblatt aus:

-- Create row access policy
CREATE ROW ACCESS POLICY IF NOT EXISTS todo_row_access_policy
    AS (owner VARCHAR) RETURNS BOOLEAN ->
        owner = CURRENT_USER();

Wenden Sie die Zeilenzugriffsrichtlinie an

Führen Sie den folgenden Befehl aus dem Arbeitsblatt aus:

-- Apply row access policy
ALTER TABLE todo_list ADD ROW ACCESS POLICY todo_row_access_policy ON (owner);

Damit sind die Arbeitsblattoperationen abgeschlossen.

Führen Sie die Streamlit in Snowflake-App aus

Erstellen Sie eine neue Streamlit in Snowflake-App und kopieren Sie den folgenden Code und fügen Sie ihn ein:

In Zeile 14 wird der aktuell mit der App verbundene Benutzer als Zeichenfolge abgerufen.

import streamlit as st
from snowflake.snowpark.context import get_active_session
import pandas as pd

# Layout settings
st.set_page_config(
    layout="wide"
)

# Get Snowflake session
session = get_active_session()

# Get current user
current_user = session.sql("SELECT CURRENT_USER()").collect()[0][0]

# Get ToDo list
def get_todo_list():
    return session.table("todo_list").to_pandas()

# Add or update task
def upsert_task(task_id, task, status, due_date, completed_date):
    due_date_sql = f"'{due_date}'" if due_date else "NULL"
    completed_date_sql = f"'{completed_date}'" if completed_date else "NULL"

    if task_id:
        session.sql(f"""
        UPDATE todo_list
        SET task = '{task}', status = '{status}', due_date = {due_date_sql}, completed_date = {completed_date_sql}
        WHERE id = {task_id}
        """).collect()
    else:
        session.sql(f"""
        INSERT INTO todo_list (task, status, owner, due_date, completed_date)
        VALUES ('{task}', '{status}', '{current_user}', {due_date_sql}, {completed_date_sql})
        """).collect()

# Delete task
def delete_task(task_id):
    session.sql(f"DELETE FROM todo_list WHERE id = {task_id}").collect()

# Main function
def main():
    st.title(f"{current_user}'s Personal Dashboard")

    # Task list
    st.subheader(f"{current_user}'s ToDo List")
    todo_df = get_todo_list()

    # Display header
    col1, col2, col3, col4, col5 = st.columns([3, 2, 2, 2, 2])
    col1.write("Task")
    col2.write("Status")
    col3.write("Due Date")
    col4.write("Completed Date")
    col5.write("Delete")

    # Display task list
    for _, row in todo_df.iterrows():
        col1, col2, col3, col4, col5 = st.columns([3, 2, 2, 2, 2])

        with col1:
            task = st.text_input("task", value=row['TASK'], key=f"task_{row['ID']}", label_visibility="collapsed")

        with col2:
            status = st.selectbox("status", ["Pending", "In Progress", "Completed"], index=["Pending", "In Progress", "Completed"].index(row['STATUS']), key=f"status_{row['ID']}", label_visibility="collapsed")

        with col3:
            due_date = st.date_input("due_date", value=pd.to_datetime(row['DUE_DATE']).date() if pd.notna(row['DUE_DATE']) else None, key=f"due_date_{row['ID']}", label_visibility="collapsed")

        with col4:
            completed_date = st.date_input("comp_date", value=pd.to_datetime(row['COMPLETED_DATE']).date() if pd.notna(row['COMPLETED_DATE']) else None, key=f"completed_date_{row['ID']}", label_visibility="collapsed")

        with col5:
            if st.button("Delete", key=f"delete_{row['ID']}"):
                delete_task(row['ID'])
                st.experimental_rerun()

        # Update database immediately if values change
        if task != row['TASK'] or status != row['STATUS'] or due_date != row['DUE_DATE'] or completed_date != row['COMPLETED_DATE']:
            upsert_task(row['ID'], task, status, due_date, completed_date)
            st.experimental_rerun()

    # Add new task
    st.subheader("Add New Task")
    new_task = st.text_input("New Task")
    new_status = st.selectbox("Status", ["Pending", "In Progress", "Completed"])
    new_due_date = st.date_input("Due Date")
    if st.button("Add"):
        upsert_task(None, new_task, new_status, new_due_date, None)
        st.success("New task added")
        st.experimental_rerun()

# Main process
if __name__ == "__main__":
    main()

Abschluss

Was denkst du? Durch die Kombination von Current_User und Zeilenzugriffsrichtlinie können Sie mit einfachen Schritten eine sichere, für jeden Benutzer personalisierte Anwendung erstellen. Dies eröffnet Möglichkeiten, noch benutzerfreundlichere Anwendungen basierend auf Ihren Ideen zu erstellen.

Einige fortgeschrittene Ideen umfassen das Hinzufügen von Current_User-Informationen als Signatur beim Schreiben in Tabellen über Streamlit in Snowflake oder die Verwendung personalisierter Informationen als Kontext für Cortex LLM, um einen persönlichen Assistenten zu erstellen.

Bitte versuchen Sie, sich selbst mit interessanten Einsatzmöglichkeiten von Current_User herauszufordern!

Ankündigungen

Was gibt es Neues bei Snowflake? Updates auf X

Ich teile Snowflakes Neuigkeiten-Updates auf X. Bitte folgen Sie uns gerne, wenn Sie interessiert sind!

Englische Version

Snowflake What's New Bot (englische Version)
https://x.com/snow_new_en

Japanische Version

Snowflake What's New Bot (japanische Version)
https://x.com/snow_new_jp

Änderungsverlauf

(20240914) Erster Beitrag

Originaler japanischer Artikel

https://zenn.dev/tsubasa_tech/articles/a23029dfe97c46

Das obige ist der detaillierte Inhalt vonPersonalisieren Sie für jeden Benutzer mit Streamlit in Snowflake (SiS). Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!