使用python和streamlit编写一个包含sqlite数据库的网站，包含...

37ck

1. import streamlit as st
   
2. import sqlite3
   
3. 
   
4. conn = sqlite3.connect('database.db')
   
5. cursor = conn.cursor()
   
6. 
   
7. operation = st.sidebar.selectbox("操作", ("创建表格", "插入数据", "查询数据", "删除数据"))
   
8. 
   
9. if operation == "创建表格":
   
10.     cursor.execute('''
    
11.     CREATE TABLE IF NOT EXISTS users (
    
12.         id INTEGER PRIMARY KEY AUTOINCREMENT,
    
13.         name TEXT,
    
14.         age INTEGER
    
15.     )
    
16.     ''')
    
17.     conn.commit()
    
18.     st.write("创建表格成功")
    
19. 
    
20. elif operation == "插入数据":
    
21.     name = st.text_input("请输入姓名")
    
22.     age = st.number_input("请输入年龄")
    
23.     if st.button("插入"):
    
24.         cursor.execute('''
    
25.         INSERT INTO users (name, age)
    
26.         VALUES (?, ?)
    
27.         ''', (name, age))
    
28.         conn.commit()
    
29.         st.write("插入数据成功")
    
30. 
    
31. elif operation == "查询数据":
    
32.     cursor.execute('SELECT * FROM users')
    
33.     data = cursor.fetchall()
    
34.     for row in data:
    
35.         st.write(f"ID: {row[0]}, 姓名: {row[1]}, 年龄: {row[2]}")
    
36. 
    
37. elif operation == "删除数据":
    
38.     cursor.execute('SELECT * FROM users')
    
39.     data = cursor.fetchall()
    
40.     for row in data:
    
41.         st.write(f"ID: {row[0]}, 姓名: {row[1]}, 年龄: {row[2]}")
    
42.     id = st.number_input("请输入要删除的ID")
    
43.     if st.button("删除"):
    
44.         cursor.execute('DELETE FROM users WHERE id = ?', (id,))
    
45.         conn.commit()
    
46.         st.write("删除数据成功")
    
47. 
    
48. if __name__ == "__main__":
    
49.     st.title("SQLite数据库网站")
    
50.     st.write("请在侧边栏选择操作")
    

复制代码