建立Python 编程的项目观和架构思维 - qidao123.com技术社区-IT企服评测·应用市场

import streamlit as st
import pandas as pd
import hashlib
import sqlite3
if 'user' not in st.session_state:
st.session_state.user = None
# Hashing password
def hash_password(password):
return hashlib.sha256(password.encode()).hexdigest()
# Database connection
conn = sqlite3.connect('grades.db', check_same_thread=False)
c = conn.cursor()
# Create tables if they don't exist
def init_db():
c.execute('''
CREATE TABLE IF NOT EXISTS users (
username TEXT PRIMARY KEY,
password_hash TEXT
)
''')
c.execute('''
CREATE TABLE IF NOT EXISTS grades (
id INTEGER PRIMARY KEY AUTOINCREMENT,
math REAL,
english REAL,
major REAL
)
''')
conn.commit()
init_db()
# Login function
def login():
st.subheader("登录")
username = st.text_input("用户名")
password = st.text_input("密码", type="password")
if st.button("登录"):
hashed_password = hash_password(password)
c.execute("SELECT * FROM users WHERE username=? AND password_hash=?", (username, hashed_password))
user = c.fetchone()
if user:
st.success(f"成功登录为 {username}")
st.session_state.user = username
else:
st.error("无效的用户名或密码")
# Register function
def register():
st.subheader("注册")
new_username = st.text_input("新用户名")
new_password = st.text_input("新密码", type="password")
confirm_password = st.text_input("确认密码", type="password")
if st.button("注册"):
if new_password == confirm_password:
hashed_password = hash_password(new_password)
try:
c.execute("INSERT INTO users (username, password_hash) VALUES (?, ?)", (new_username, hashed_password))
conn.commit()
st.success("用户注册成功！")
except sqlite3.IntegrityError:
st.error("用户名已存在")
else:
st.error("密码不匹配")
# Grades input function
def input_grades():
st.subheader("输入成绩")
math = st.number_input("数学成绩", min_value=0, max_value=100, step=1)
english = st.number_input("英语成绩", min_value=0, max_value=100, step=1)
major = st.number_input("专业课成绩", min_value=0, max_value=100, step=1)
if st.button("提交"):
c.execute("INSERT INTO grades (math, english, major) VALUES (?, ?, ?)", (math, english, major))
conn.commit()
st.success("成绩提交成功！")
# Grades visualization function
def visualize_grades():
st.subheader("成绩可视化")
c.execute("SELECT * FROM grades")
grades_data = c.fetchall()
if grades_data:
df = pd.DataFrame(grades_data, columns=["ID", "Math", "English", "Major"])
st.dataframe(df)
st.bar_chart(df.set_index("ID"))
else:
st.warning("没有可用的成绩数据")
# Main application logic
def main():
st.title("学生成绩管理系统")
menu = ["登录", "注册", "输入成绩", "成绩可视化"]
choice = st.sidebar.selectbox("菜单", menu)
if choice == "登录":
login()
elif choice == "注册":
register()
elif choice == "输入成绩":
if st.session_state.user:
input_grades()
else:
st.warning("请先登录以访问此功能")
elif choice == "成绩可视化":
if st.session_state.user:
visualize_grades()
else:
st.warning("请先登录以访问此功能")
if __name__ == "__main__":
main()

复制代码