非法入侵 发表于 2018-8-11 11:05:41

Python第三周作业--工资管理系统

#!/use/bin/env python  # -*- coding:utf-8 -*-
  import os
  def select(): # 查询员工工资
  with open("info.txt", "r", encoding="utf-8") as f:
  fsb = f.readlines()
  user_choice = input("请输入你要查询的员工姓名:")
  for line in fsb:
  if user_choice.split() in line.split():
  print(user_choice, "的工资是:",line.split())
  def update(): # 修改员工工资
  user_choice_1 = input("请输入要修改的员工姓名和工资,用空格分隔(例如:Alex 10):").strip()
  with open("info.txt", "r", encoding="utf-8") as f,open("info.bak", "w", encoding="utf-8") as f1:
  for user_list in f:
  if user_list=="\n": # 判断文本中有空行的话直接跳过
  continue
  if user_choice_1.split() in user_list.split(): #判断输入的员工是否在文本中
  user_list = user_list.replace(user_list.split(), user_choice_1.split()) # 修改员工工资
  f1.write(user_list)# info.txt文本内容已经重新写入info.bak
  f.close()# 关闭f
  f1.close()# 关闭f1
  os.remove("info.txt")# 删除info.txt文件
  os.rename("info.bak","info.txt")# 把info.bak文本重新命名为info.txt
  print("修改成功")
  def add(): # 添加新员工
  user_add = input("请输入要增加的员工姓名和工资(例如:Eric 100000):")
  with open("info.txt", "a", encoding="utf-8") as f:
  # f.seek(0)
  f.write('\n' + user_add)
  print("添加成功")
  def delete(): # 删除员工
  user_choice = input("请输入要删除的员工姓名:").strip()
  with open("info.txt", "r", encoding="utf-8") as f, open("info.bak", "w", encoding="utf-8") as f1:
  for user_list in f:
  if user_list =="\n": # 判断文本中有空行的话直接跳过
  continue
  if user_choice in user_list.split():# 判断输入的员工是否在文本中
  continue # 直接跳过输入的用户,这样也相当于删除了
  f1.write(user_list) # 输入的用户直接跳过了,所以不会加载到文本中
  f.close() # 关闭f
  f1.close() # 关闭f1
  os.remove("info.txt") # 删除info.txt文件
  os.rename("info.bak", "info.txt") # 把info.bak文本重新命名为info.txt
  print("删除成功")
  while True:
  user_chioce = '''
  1.查询员工工资
  2.修改员工工资
  3.增加新员工记录
  4.删除员工
  5.退出
  '''
  print(user_chioce)
  user_choice = input("请输入你的选择:")
  if user_choice.isdigit(): # 判断输入的是否数字类型
  if user_choice == '1':
  select() # 查询
  elif user_choice == '2':
  update() # 修改
  elif user_choice == '3':
  add() # 添加
  elif user_choice == '4':
  delete() # 删除
  elif user_choice == '5':
  exit() # 退出
  else:
  print("请输入正确的选项")
  else:
  print("请输入正确的选项")
页: [1]
查看完整版本: Python第三周作业--工资管理系统