ATM 程序

ATM 目前只有用户操作，数据操作暂时未做，算遗留吧，万变不理其中，都是这么操作的！

# -*- coding:utf-8 -*-
# Author:JinYu

dect_info = {
    'name_login':'',    #登录名
    'pwd_login':'',     #密码
    'account':'',       #账号
    'login_data':'',    #注册时间
    'billing_day':'15', #账单日
    'repayment_day':'1',#还款日
    'status':'0',       #账户状态(0活跃 1冻结)
    'cash_balance':'0',  #现金余额
    'facility_extent':'15000',#默认透支额度15000
    'amountin_arrears':'0',#欠款金额
    'shopping_history':{},#购物历史
}
import json,time,json,os,sys
#print(time.localtime(time.time()))
day = time.strftime('%Y-%m-%d',time.localtime(time.time()))
time = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time()))
print(day,time)
BASE_PATH = os.path.dirname(os.path.abspath(__file__))
FILE_NAME = BASE_PATH+os.path.sep
#print(BASE_PATH)

def check(a,b,c=None):
    if len(a) ==0 and len(b) ==0:
        print('你输入的用户名密码不可为空！')
        return True
    elif len(a) ==0 and len(b) ==0 and len(c) == 0:
        print('你输入的字段不可为空！')
        return True
    else:
        return False

#用户查询
def login_select():
    count = 0
    while True:
        username = input('输入用户名:')
        password = input('输入密码:')
        if check(username,password):
            continue
        else:
            with open(FILE_NAME+'creadit_history','r') as write_f:
                flag = False
                for line in  write_f:
                    data = json.loads(line)
                    if username in data['name_login']:
                        #print(data)
                        if  data['status'] == '1' : #data['name_login'] == username and
                            #print(data['status'],data['name_login'],count)
                            exit('\033[32;1m您的账号已被锁定，请到营业厅办理办理!!!\n\033[0m ')
                            #return  True
                        elif  count >=2:
                            #print(count+1)
                            exit('\033[32;1m您的账号多次登录，请稍后再试!!!\n\033[0m ')
                        elif username == data['name_login'] and password == data['pwd_login']:
                            print('\033[32;0m登录成功\033[0m'.center(50,'*'))
                            return True
                        else:
                            print('\033[32;1m账号密码错误，请重新输入！！\033[0m')
                            flag = True
                            continue
                else:
                    #print(flag)
                    if flag == True:
                        count +=1
                    if flag == False:
                            print('\033[32;1m没有这个用户！！\033[0m')

#login_select()

#新增用户
def add_login():
    #import  json
    while True:
        username = input('输入用户名:')
        password = input('输入密码:')
        account = input('输入账号:')
        atm = input('输入信用卡额度:')
        # if check(username,password,account):
        #     continue
        # else:
        with open(FILE_NAME+'creadit_history','r+') as write_f:
            for line in  write_f:
                data = json.loads(line)
                if username not in data['name_login']:
                    dect_info['name_login'] =username
                    dect_info['pwd_login'] = password
                    dect_info['account'] = account
                    if len(atm) and atm.isdecimal():
                        dect_info['facility_extent'] = atm
                    dect_info['login_data'] =day
                    #write_f.write(json.dumps(dect_info))
                    write_f.write(json.dumps(dect_info)+'\n')
                    write_f.flush()
                    exit('\033[32;1m新增用户成功！！\033[0m')

                else:
                    exit('\033[32;1m您输入的用户已存在！！\033[0m')

#add_login()

#删除用户记录
def del_login():
    while True:
        username = input('输入要删除的用户名:')
        with  open(FILE_NAME+'creadit_history','r') as read_f,open(FILE_NAME+'creadit_crad','w') as write_f:
            for line in read_f:
                data = json.loads(line)
                #print(data)
                if username in data['name_login']:
                    data.clear()
                    pass
                else:
                    write_f.write(json.dumps(data)+'\n')
                    write_f.flush()
        os.rename(FILE_NAME+'creadit_history',FILE_NAME+'creadit_history.bak')
        os.rename(FILE_NAME+'creadit_crad',FILE_NAME+'creadit_history')
        exit('\033[32;1m删除用户成功！！\033[0m')

#del_login()

#修改用户额度
def modify_login():
    while True:
        username = input('输入要修改额度的用户名:')
        with  open(FILE_NAME+'creadit_history','r') as read_f,open(FILE_NAME+'creadit_crad','w') as write_f:
            for line in read_f:
                data = json.loads(line)
                #print(data)
                if username in data['name_login']:
                    atm = input('输入信用卡额度:')
                    if len(atm) and atm.isdecimal():
                        data['facility_extent'] = atm
                        #print(data)
                    write_f.write(json.dumps(data)+'\n')
                    write_f.flush()
                else:
                    write_f.write(json.dumps(data)+'\n')
                    write_f.flush()
        os.rename(FILE_NAME+'creadit_history',FILE_NAME+'creadit_history.bak')
        os.rename(FILE_NAME+'creadit_crad',FILE_NAME+'creadit_history')
        exit('\033[32;1m修改用户额度成功！！\033[0m')

#modify_login()

#冻结用户
def freeze_login():
    while True:
        username = input('输入要冻结的用户名:')
        with  open(FILE_NAME+'creadit_history','r') as read_f,open(FILE_NAME+'creadit_crad','w') as write_f:
            for line in read_f:
                data = json.loads(line)
                #print(data)
                if username in data['name_login']:
                    data['status'] = '1'
                    #print(data)
                    write_f.write(json.dumps(data)+'\n')
                    write_f.flush()
                else:
                    write_f.write(json.dumps(data)+'\n')
                    write_f.flush()
        os.rename(FILE_NAME+'creadit_history',FILE_NAME+'creadit_history.bak')
        os.rename(FILE_NAME+'creadit_crad',FILE_NAME+'creadit_history')
        exit('\033[32;1m用户冻结成功！！\033[0m')
#freeze_login()

# -*- coding:utf-8 -*-
# Author:JinYu

import os,sys

DASE_DIR = os.path.dirname(os.path.abspath(__file__))
#DASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
#print(DASE_DIR)
sys.path.append(DASE_DIR)
from Login import *

#装饰器
def auth(auth_type):
    def outer_wrapper(func):
        def wrapper(*args,**kwargs):
            if auth_type =='local':
                login_select()
                ret = func(*args,**kwargs)
                return ret
            elif auth_type =='ldap':
                print('不需要登录，请直接操作！')
                ret = func(*args,**kwargs)
                return ret
        return wrapper
    return outer_wrapper

@auth(auth_type='local')
def index(name,age):
    print('welcome to index page! ',name,age)
index('jinyu',18)

@auth(auth_type='ldap')
def home():
    print('welcome to home page!')

home()

# -*- coding:utf-8 -*-
# Author:JinYu

import os,sys

BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
print(BASE_DIR)
sys.path.append(BASE_DIR)

from conf import Login


def login_action():
    Login.login_select()
    menu_dic ={
        1:Login.add_login,
        2:Login.del_login,
        3:Login.modify_login,
        4:Login.freeze_login,

    }
    # print(menu_dic['1']())
    # exit()
    exit_flag = False
    while not exit_flag:
        print('\n')
        print('\033[32;1m欢迎光临ATM电子银行用户管理系统\033[0m'.center(50,'*'))
        print('\033[32;1m\t\t用户操作请选择以下功能项\n\t\t1,信用卡用户新增\n\t\t2,信用卡用户删除\n\t\t3,信用卡用户额度修改\n\t\t4,信用卡用户冻结\033[0m')
        temp = int(input('请输入与功能菜单:'))
        if temp in  menu_dic:
            print(menu_dic[temp])
            menu_dic[temp]()
        else:
            #print(menu_dic[temp])
            print('\033[31;1m输入的功能编号有误，请重新输入要操作的功能:\033[0m')

login_action()