简单的工资系统，借鉴了别人的思路，自己补充了下，还有待提高

#!/usr/bin/env python
#-*- coding:utf-8 -*-
#3.0python版本以上运行
import sys
import os

#将文件变为列表
info_dict = {}
#员工工资
user_dict = {}
#文件修改
def file():
    os.remove('user.txt')
    os.rename('simple.txt','user.txt')



#所有员工工资
def info():
    with open('user.txt','r+') as f:
        for i in f:
            i = i.strip()
            info_dict[i.split(' ')[0]] = i.split(' ')[1]

#查询员工是否存在
def select():
    count = 0
    while True:
        info()
        user_info = input("请输入你的名字,按q则退出---》")
        if user_info in info_dict.keys():
            print('%s你的工资是：%s' %(user_info,info_dict[user_info]))
        elif user_info == 'q':
            break
        else:
            count+=1
            print("对不起你输入的名字不存在,你已经尝试%s次" % count)
            if count > 3:
                print("对不起你尝试%s次无效用户【%s】，退出！！" %(count,user_info))
                break
#新增员工
def insert():
    while True:
        info()
        user_info = input("请输入需要增加的员工和工资-->").strip()
        user_dict[user_info.split(' ')[0]] = user_info.split(' ')[1]
        user_list = user_info.split(' ')
        if user_info not in info_dict.keys():
            user_info = '
'+user_info
            with open('user.txt','a+') as f:
                f.write(user_info)
                print("%s已经添加成功" %user_info[0])
                break
        else:
            print("%s用户已经存在" %user_info)


#修改员工工资
def update():
        while True:
            info()
            user_info = input("请输入你的修改名字---》").strip()
            user_new_money = input("请输入你要修改的工资是多少--》")
            if user_info in info_dict.keys():
                 if user_new_money.isdigit():
                #     user_new_money = int(user_new_money)
                    #info_dict[user_info]=user_new_money
                    with open('user.txt','r') as f2,open('simple.txt','w') as f3:
                        for i in f2:
                            i = i.replace(info_dict[user_info],user_new_money)
                            f3.write(i)
                    print("你的%s员工工资修改为%s" %(user_info,user_new_money))
                    file()
                    break
                 else:
                     print("您输入的不是阿拉伯数字，请输入数字")
            else:
                print("用户不存在")



a = """
1.查询
2.增加
3.修改
4.退出
"""

all ={
    '1':select,
    '2':insert,
    '3':update,
    '4':sys.exit
}
while True:
    print(a)
    choose = input("请输入你需要的编号--》")
    if choose not in all.keys():
        continue
    all[choose]()