简单的工资管理系统小程序，只是练手

需求如下：

工资管理系统
aaa 100000
bbb 80000
ccc 50000
ddd 30000
-----以上是info.txt文件-----
实现效果：
从info.txt文件中读取员工及其工资信息，最后将修改或增加的员工工资信息
也写入原info.txt文件。
效果演示：
1. 查询员工工资
2. 修改员工工资
3. 增加新员工记录
4. 退出

#!/usr/bin/env python
#-*- coding:utf-8 -*-
# @Time    : 2017/9/21 15:19
# @Author  : lichuan
# @File    : salary.py

import  os


def select(read,write):
    i=input("please input the employee name:").strip()
    for line in read:
        if i == line.split()[0]:
            print("%s的工资是：%s" % (line.split()[0],line.split()[1]))
            return
    print("输入错误,请重试！")
    read.seek(0)
    select(read, write)

def modify(read,write):
    # import os
    FLAG_TAG=True
    m=input("请输入要修改的员工姓名和工资,以空格分隔").strip().split()
    if len(m) !=2:
        print("输入错误,请重试,以空格分隔！")
        modify(read,write)
        return
    for i in read:
        if i.split()[0] == m[0] and m[1].isdigit():
            FLAG_TAG=False
            i = "%s %s\n" % (m[0], m[1])
        write.write(i)
    if FLAG_TAG:
        print("输入错误,请重试！")
        read.seek(0)
        write.seek(0)
        modify(read,write)
    else:
        read.close()
        write.close()
        os.remove("info.txt")
        os.rename(".swap", "info.txt")



def add(read,write):
    # import os
    a=input("请输入员工姓名和工资，以空格分隔").strip().split()
    if len(a) !=2:
        print("输入错误,请重试,以空格分隔！")
        add(read,write)
        return

    for i in read:
        if i.split()[0] == a[0]:
            print("用户已存在，请重新添加新用户！")
            read.seek(0)
            write.seek(0)
            add(read,write)
            return
        write.write(i)
    write.write("\n%s %s" %(a[0],a[1]))
    read.close()
    write.close()
    os.remove("info.txt")
    os.rename(".swap","info.txt")



def out(read,write):

    print("再见！")
    # FLAG_TAG=False

item={
    1:"查询员工工资",
    2:"修改员工工资",
    3:"增加新员工记录",
    4:"退出"
      }

fun={
    1:select,
    2:modify,
    3:add,
    4:out
}

FLAG_TAG=True

while FLAG_TAG:
    for i in range(1,5):
        print("%d, %s" %(i,item[i]))
    choice=input("请选择：").strip()
    with open('info.txt','r',encoding="utf-8") as r_read,open('.swap','w',encoding="utf-8") as r_write:
        if choice.isdigit() and int(choice) >=1 and int(choice)<4:
            choice=int(choice)
            fun[choice](r_read,r_write)
            continue
        elif choice.isdigit() and int(choice) == 4:
            FLAG_TAG=False
            fun[int(choice)](r_read,r_write)
            break
        else:
            print("wrong input,try again!")
            continue
    FLAG_TAG=False