跳至主要內容

简单的人事系统

知识库编程技巧PythonPython小于 1 分钟

脚本

import streamlit as st
import pandas as pd

# 创建员工类
class Employee:
    def __init__(self, name, age, position):
        self.name = name
        self.age = age
        self.position = position

# 创建员工列表
employee_list = []

# 添加员工函数
def add_employee(name, age, position):
    employee = Employee(name, age, position)
    employee_list.append(employee)

# 显示员工列表函数
def show_employee_list():
    if len(employee_list) == 0:
        st.write('员工列表为空!')
    else:
        df = pd.DataFrame([[e.name, e.age, e.position] for e in employee_list], columns=['姓名', '年龄', '职位'])
        st.dataframe(df)

# 添加员工界面
def add_employee_page():
    st.write('添加新员工')
    name = st.text_input('姓名')
    age = st.number_input('年龄', min_value=0, max_value=100)
    position = st.text_input('职位')
    if st.button('添加'):
        add_employee(name, age, position)
        st.success('添加成功!')

# 显示员工列表界面
def show_employee_list_page():
    st.write('员工列表')
    show_employee_list()

# 主程序
def main():
    st.title('人事系统')
    menu = ['添加员工', '员工列表']
    choice = st.sidebar.selectbox('选择菜单', menu)
    if choice == '添加员工':
        add_employee_page()
    elif choice == '员工列表':
        show_employee_list_page()

if __name__ == '__main__':
    main()