Python+streamlit实现轻松创建人事系统

时间:2023-03-23 Python数据开发人气:0

系统简介

这个简单的人事系统包含了添加员工和显示员工列表两个功能。在 Streamlit 库中，可以使用 st.sidebar 创建侧边栏，其中可以包含各种菜单选项。在选择菜单选项后，可以使用 st.write 和 st.dataframe 显示文本和数据表格。此外，可以使用 st.text_input 和 st.number_input 创建输入框，让用户输入员工信息，并使用 st.button 创建按钮来触发添加员工操作。最后，在程序主函数中调用以上各个函数即可。

代码详情

我们先导入相关依赖，并创建员工类

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()

完整代码

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()

加载全部内容