欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

周末小作业

程序员文章站 2022-07-04 09:55:55
import os import time # 添加员工信息 def zengjia(): print('请输入员工的基本信息:') while 1: while 1: id = input("请输入编号").strip() if not id.isdigit(): print("... ......

周末小作业

import os
import time

# 添加员工信息
def zengjia():
    print('请输入员工的基本信息:')
    while 1:
        while 1:
            id = input("请输入编号").strip()
            if not id.isdigit():
                print("格式不对,重新输入")
                continue
            with open('emp.db', mode='r', encoding='utf-8') as f:
                lst = []
                for line in f:
                    lst.append(line.split('_')[0]) # 将所有的编号取出来存放lst列表中
            for el in lst: # 遍历列表,查看输入编号是否重复
                if id == el:
                    print('该编号已存在,请重新输入')
                    break
            else:
                name = input("请输入名字").strip()
                birthday = input("请输入出生日期").strip()
                salary = input("请输入工资").strip()
                input_time = time.strftime("%y-%m-%d %h:%m:%s",time.localtime())
                with open('emp.db', mode='a+', encoding='utf-8') as f:
                    f.write(id + '_' + name + '_' + birthday + '_' + salary + '_' + input_time + '\n')
                print('增加成功')
                with open('emp.log', mode='a', encoding='utf-8') as f1:
                    d = time.strftime("%y-%m-%d %h:%m:%s",time.localtime())
                    f1.write('管理员在%s增加了添加员工信息操作。' % d + '\n')
                tc = input('输入q退出,继续按任意键').strip()
                if tc.upper() == 'q':
                    break
        break

#员工信息修改
def xiugai():
    with open('emp.db', mode='r', encoding='utf-8') as f, \
            open('emp_副本.db', mode='w', encoding='utf-8') as f1:
        for line in f:
            print(line.strip())
        f.seek(0) # 显示完所有员工信息,重新将光标放开头
        s = input("请输入要修改的id:").strip()
        money = input('请输入工资:').strip()
        for line in f:
            if line[0] == s:
                line = line.replace(line.split('_')[3], money) #  根据编号,取出来该员工的工资,进行修改
                # print(line.split('_'))
                f1.write(line)
            else:
                f1.write(line)
    os.remove('emp.db')
    os.rename('emp_副本.db','emp.db')

#删除员工信息
def shanchu():
    with open('emp.db', mode='r', encoding='utf-8') as f, \
            open('emp_副本.db', mode='w', encoding='utf-8') as f1:
        for line in f:
            print(line.strip())
        f.seek(0)
        s = input("请输入要删除的id:")
        for line in f:
            if line[0] < s:
                f1.write(line)
            elif line[0] > s:  # 将编号进行更改,比如删除编号3,后面的编号依次-1
                line = line.replace(line.split('_')[0],str(int(line.split('_')[0]) - 1))
                f1.write(line)
    os.remove('emp.db')
    os.rename('emp_副本.db','emp.db')

# 查看信息
def chakan():
    with open('emp.db', mode='r', encoding='utf-8') as f:
        for line in f:
            print(line.strip())

while 1:
    print('hr人力资源管理')
    print('1.查看员工信息   2.添加员工信息  3.修改员工信息  4.删除员工信息  5.退出')
    n = input('请输入选项:').strip()
    if n == '1':
        chakan()
    elif n == '2':
        zengjia()
    elif n == '3':
        xiugai()
    elif n == '4':
        shanchu()
    elif n == '5':
        exit()
    else:
        print("输入有误!请重新输入")
        continue