Python作业之工资管理

                                             作业之工资管理

工资管理实现要求:

工资管理系统

Alex 100000

Rain 80000

Egon 50000

Yuan 30000

-----以上是info.txt文件-----

实现效果:

从info.txt文件中读取员工及其工资信息,最后将修改或增加的员工工资信息也写入原info.txt文件。

效果演示:

1. 查询员工工资

2. 修改员工工资

3. 增加新员工记录

4. 退出

>>:1

请输入要查询的员工姓名(例如:Alex):Alex

Alex的工资是:100000。

1. 查询员工工资

2. 修改员工工资

3. 增加新员工记录

4. 退出

>>:2

请输入要修改的员工姓名和工资,用空格分隔(例如:Alex 10):Alex 10

修改成功!

1. 查询员工工资

2. 修改员工工资

3. 增加新员工记录

4. 退出

>>:3

请输入要增加的员工姓名和工资,共空格分割(例如:Eric 100000):Eric 100000

增加成功!

1. 查询员工工资

2. 修改员工工资

3. 增加新员工记录

4. 退出

>>:4

再见!

具体代码如下:

方法1:

import os
if not os.path.exists('d:\\users\\info.txt'):
    f = open('d:\\users\\info.txt','w')
    f.write('工资管理系统\nAlex 100000\nRain 80000\nEgon 50000\nYuan 30000\n')
    f.close()
menu = ['查询员工工资','修改员工工资','增加新员工记录','退出']
while True:
    print('欢迎来到工资管理系统!'.center(30,'*'))
    for index,option in enumerate(menu,1):#枚举并遍历菜单
        print(index,option)
    choice = input('请选择:').strip()
    if choice == str(1):
        name_c = input('请输入要查询的员工姓名(例如:Alex):').strip()
        f = open('d:\\users\\info.txt')
        f.readline()#先读取第一行内容(标题中间无空格,因此先读取标题)
        names = []#声明一个存储用户名的空列表
        salary = []#声明一个存储工资的空列表
        for line in f:#遍历文件内容
            names.append(line.strip().split()[0])#把文件中每行内容空格前面的内容添加到列表中
            salary.append(line.strip().split()[1])#把文件中每行内容空格后面的内容添加到列表中
        if name_c in names:
            print('%s的工资为:%s'%(name_c,salary[names.index(name_c)]))
        f.close()
    if choice == str(2):
        name_m = input('请输入要修改的员工姓名和工资,用空格分隔(例如:Alex 10):')
        f = open('d:\\users\\info.txt')
        f.readline()
        if name_m.strip().split()[0] in [name.strip().split()[0] for name in f]:#判断输入的用户名是否存在于文件中
            modify = []#声明一个列表存储文件每行内容
            f.seek(0)#由于上述判断读取了文件,因此需要把指针跳指开始
            for line in f:
                if name_m.strip().split()[0] == line.strip().split()[0]:#判断输入的用户名是否存在该行内容中
                    line = name_m.strip()+'\n'
                modify.append(line)
            f.close()#关闭只读模式的文件
            print(modify)
            f = open('d:\\users\\info.txt','w')#再次打开文件,只写模式
            for m_line in modify:#编历列表,把每行内容写入到文件中
                f.write(m_line)
            f.close()
    if choice == str(3):
        name_a = input('请输入要增加的员工姓名和工资,用空格分隔(例如:Mary 10):')
        f = open('d:\\users\\info.txt','a')
        f.write(name_a.strip()+'\n')
        f.close()
        print('增加成功!')
    if choice == str(4):
        print('再见!')
        break

注:该代码在修改用户工资时只使用原文件,需要把文件内容先保存到内存中,如果文件比较大,速度会比较慢,不推荐使用!

方法2:

import os
if not os.path.exists('d:\\users\\info.txt'):
    f = open('d:\\users\\info.txt','w')
    f.write('工资管理系统\nAlex 100000\nRain 80000\nEgon 50000\nYuan 30000\n')
    f.close()
menu = ['查询员工工资','修改员工工资','增加新员工记录','退出']
while True:
    print('欢迎来到工资管理系统!'.center(30,'*'))
    for index,option in enumerate(menu,1):#枚举并遍历菜单
        print(index,option)
    choice = input('请选择:').strip()
    if choice == str(1):
        name_c = input('请输入要查询的员工姓名(例如:Alex):').strip()
        f = open('d:\\users\\info.txt')
        f.readline()#先读取第一行内容(标题中间无空格,因此先读取标题)
        names = []#声明一个存储用户名的空列表
        salary = []#声明一个存储工资的空列表
        for line in f:#遍历文件内容
            names.append(line.strip().split()[0])#把文件中每行内容空格前面的内容添加到列表中
            salary.append(line.strip().split()[1])#把文件中每行内容空格后面的内容添加到列表中
        if name_c in names:
            print('%s的工资为:%s'%(name_c,salary[names.index(name_c)]))
        f.close()
    if choice == str(2):
        name_m = input('请输入要修改的员工姓名和工资,用空格分隔(例如:Alex 10):')
        f1 = open('d:\\users\\info.txt')
        f2 = open('d:\\users\\info2.txt','w')
        if name_m.strip().split()[0] in [i.strip().split()[0] for i in f1]:#判断输入的用户名是否存在于文件中
            f1.seek(0)#由于上述判断读取了文件,因此需要把指针跳指开始
            for line in f1:
                if name_m.strip().split()[0] == line.strip().split()[0]:#判断输入的用户名是否存在该行内容中
                    line = name_m.strip()
                f2.write(line+'\n')
            print('修改成功!')
        f1.close()
        f2.close()
        f1 = open('d:\\users\\info.txt','w')
        f2 = open('d:\\users\\info2.txt')
        for line in f2:#遍历文件把f2文件中的内容覆盖性地写入到f1文件中
            f1.write(line)
        f1.close()
        f2.close()
    if choice == str(3):
        name_a = input('请输入要增加的员工姓名和工资,用空格分隔(例如:Mary 10):')
        f = open('d:\\users\\info.txt','a')
        f.write(name_a.strip()+'\n')
        f.close()
        print('增加成功!')
    if choice == str(4):
        print('再见!')
        break

注:该代码在修改用户工资时使用到两个文件,然后把新创建的文件内容覆盖性写入到原文件,当文件比较大时,该方法比方法1速度要快。

转载于:https://www.cnblogs.com/single-boy/p/7536482.html

你可能感兴趣的:(Python作业之工资管理)