用python写一个查询员工信息，添加员工信息的功能--进阶版将添加的员工信息写入一个txt文件

最新推荐文章于 2022-12-31 13:28:12 发布

原创最新推荐文章于 2022-12-31 13:28:12 发布 · 2k 阅读

6 ·

CC 4.0 BY-SA版权

本文介绍了一个简单的员工信息管理系统，包括添加员工信息和查询员工信息的功能。系统通过文本文件存储员工数据，实现了基本的数据输入验证和重复数据检查。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

 # coding=utf8



# function 函数
def add_emp(e):
    """
    添加员工信息
    :return: 添加成功返回0,失败返回-1
    """
    name = raw_input('请输入员工名字:')
    #将输入的员工名字与文件中已经存在的员工姓名比较，如已存在则无法添加
    emps = open(e, 'r')
    for emp in emps:
        if name == emp.split(',')[0]:
            emps.close()
            return -1

    emps.close()
    age = input('请输入员工年龄:')
    salary = input('请输入员工工资:')

    f = open(e, 'a')
    line = '%s,%d,%d\n'%(name, age, salary)
    f.write(line)
    f.close()
    return 0


def query_emp(e):
    """
    查询员工信息
    :return: None
    """
    name = raw_input('请输入员工名字:')
    f = open(e, 'r')
    for line in f:
        line = line.strip('\n')
        line = line.split(',')
        if name == line[0]:
            print "名字:%s, 年龄:%s, 工资:%s\n"%(line[0], line[1], line[2])
            break
    else:
        print "没有此员工信息\n"
    f.close()


def main():
    # variable 变量
    emps = r'emp.txt'
    while True:
        print "1. 添加员工信息"
        print "2. 查询员工信息"
        print "3. 退出程序"
        option = raw_input('请选择[1-->3]: ')
        if option == '1':
            ret = add_emp(emps)
            if ret == 0:
                print "添加员工信息成功\n"
            else:
                print "名字已存在,添加员工信息失败\n"
        elif option == '2':
            query_emp(emps)
        elif option == '3':
            break
        else:
            print "错误选项"

    print "Bye."


main()