第三周作业part2

管理部分：

#!/usr/bin/env python  
#-*- coding:utf-8 -*-

'''管理员功能区域'''

import os
def acc_info():
    """
    :return: 返回account_info.txt中以字典形式{
    'msj': '123', 'scg': '123', 'lhy': '123', 'egon': '1234', 'flower': '123', 'fgh': '123', 'hua': '123'}
    """
    account_info = {}
    with open(r'account_info.txt',mode='r',encoding='utf-8') as f:
        for line in f:
            line = line.strip('
')
            if line != '':
                u,v = line.split('|')
                account_info[u] = v
    return account_info

def bk_info():
    '''
    钱和信用额度都是浮点数类型
    :return: {'msj': {'pwd': '123', 'money': 4600.0, 'credit': 15000.0}}
    '''
    bank_user ={}
    with open(r'bank_info.txt',mode='r',encoding='utf-8') as f:
        for line in f:
            line = line.strip('
')
            if line != '':
                u,p,m,c = line.split('|')
                bank_user[u]={'pwd':p,'money':float(m),'credit':float(c)}
        return bank_user

def select_type():
    while True:
        choice = input('请输入你想要输入的类型 1.shop 2.bank>>:').strip()
        if choice == '1':
            return 'shop'
        elif choice == '2':
            return  'bank'
        else:
            print('非法操作')
            continue

def is_name():
    type = select_type()
    if type == 'shop':
        user_info = acc_info()
    else:
        user_info = bk_info()
    name = input('please input your username(q to exit)>>:').strip()
    if name  in user_info:
        return name,type
    else:
        return False,name,type

def input_pwd():
    pwd = input('请输入密码>>:').strip()
    return pwd


def check_userinfo():
    type = select_type()
    if type == 'shop':
        user_info =acc_info()
    elif type =='bank':
        user_info = bk_info()
    for k,v in user_info.items():
        print('用户名%s'%k,'信息：%s'%v)



def add_user():
    """
    添加账户
    :param name:
    :param pwd:
    :param type: 账户种类 两种  account  和  bank
    :return:
    """
    info = is_name()
    if False in info:
        name = info[1]
        type = info[2]
    else:
        print('用户名已存在')
        return
    pwd = input_pwd()
    if type =='shop':
        file = 'account_info.txt'
        user_info ='%s|%s
'%(name,pwd)
    elif type =='bank':
        file = 'bank_info.txt'
        user_info ='%s|%s|0|15000
'%(name,pwd)
    else:
        print('非法操作2')
        return
    with open(r'%s'%file,mode='a',encoding='utf-8') as f:
        f.write(user_info)

def alter_credit():
    '''
    改变账号中的信用额
    :param name: 用户名
    :param money: int 钱
    :return:
    '''
    name =input('请输入用户名>>:').strip()
    user_info = bk_info()
    if name not in user_info:
        print('用户名不存在')
        return
    money = input('请输入金额>>:').strip()
    if not money.isdigit():
        print('输入的金额不对')
        return
    money = float(money)
    user_info[name]['credit']=user_info[name]['credit']+money
    # print(user_info[name]['money'])
    with open(r'bank_info.txt',mode='w',encoding='utf-8') as f:
        for k,v in user_info.items():
            msg = '{}|{}|{}|{}
'.format(k,v['pwd'],v['money'],v['credit'])
            f.write(msg)

def r_hmd(type):
    hmd =[]
    if type =='shop':
        file = 'shop_hmd.txt'
    elif type == 'bank':
        file ='bank_hmd.txt'
    with open(r'%s'%file,mode='r',encoding='utf-8') as f:
        for i in f:
            i=i.strip('
')
            if i != "":
                hmd.append(i)
    return hmd


def add_hmd():
    type =select_type()
    name = input('输入加入黑名单的用户名>>:').strip()
    if type =='shop':
        file = 'shop_hmd.txt'
        user_inf = acc_info()
    elif type == 'bank':
        file ='bank_hmd.txt'
        user_inf = bk_info()
    if name not in user_inf:
        print('用户不存在')
        return
    hmd = r_hmd(type)
    if name in hmd:
        print('用户名已存在黑名单')
        return
    with open(r'%s'%file,mode='a',encoding='utf-8') as f:
        f.write('%s
'%name)
    print('%s 已加入黑名单')



def read_hmd():
    type = select_type()
    res =r_hmd(type)
    print(res)

def del_hmd():
    type = select_type()
    name = input('想要删除的黑名单>>:').strip()
    hmd  = r_hmd(type)
    if name not in hmd:
        print('用户不在黑名单')
        return
    if type =='shop':
        file = 'shop_hmd.txt'
    elif type == 'bank':
        file ='bank_hmd.txt'
    with open(r'%s'%file, mode='r', encoding='utf-8') as rf,
            open(r'%s.swap'%file, mode='w', encoding='utf-8') as wf:
        for line in rf:
            if name == line.strip('
'):
                line = ''
            wf.write(line)
    os.remove('%s'%file)
    os.rename('%s.swap'%file,'%s'%file)



msg = """
    1、增加用户
    2、增加额度
    3、查看用户信息
    4、增加黑名单
    5、查看黑名单
    6、删除黑名单
    0、退出
"""
dic ={
    '1':add_user,
    '2':alter_credit,
    '3':check_userinfo,
    '4':add_hmd,
    '5':read_hmd,
    '6':del_hmd,
}

def admin():
    while True:
        print(msg)
        choice = input('请输入你的操作>>:')
        if choice == '0':
            print('exit')
            return
        elif choice in dic:
            dic[choice]()
        else:
            print('非法操作')
            continue

admin()