簡單的工資管理系統小程序，只是練手

需求如下：

工資管理系統
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