ATM

作业需求：

模拟实现一个ATM + 购物商城程序

1. 额度 15000或自定义
2. 实现购物商城，买东西加入 购物车，调用信用卡接口结账
3. 可以提现，手续费5%
4. 每月22号出账单，每月10号为还款日，过期未还，按欠款总额 万分之5 每日计息
5. 支持多账户登录
6. 支持账户间转账
7. 记录每月日常消费流水
8. 提供还款接口
9. ATM记录操作日志
10. 提供管理接口，包括添加账户、用户额度，冻结账户等。。。

import sys
sys.path.append("..")
from config import main

main.main()

import os
import pickle
import datetime
A = datetime.datetime.now()

def login():
    # 已有管理员 nikita; 密码 123
    '''
    管理员登录
    :return:
    '''
    while True:
        name = input('Please enter name:')
        password = input('Please enter password:')
        a = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
        line = pickle.load(open(os.path.join(a,'db','admin'), 'rb'))
        if line[name]== password:
            print ('登录成功！')
            with open('record', 'a') as f:
                f.write("%s %s 登陆
" % (name, A))
            break
        else:
            print('用户名或密码错误，请重新输入')


def ls():
    while True:
        choice = input('请选择 1 增加管理员； 2 查看管理员; 3 修改用户信息； 4 删除用户信息； q = 退出')
        if choice == '1':
            add_admin()
        elif choice == '2':
            check()
        elif choice == '3':
            ck_ls()
            amend_user()
        elif choice == '4':
            remove_user()
        elif choice == 'q':
            exit()
        else:
            print('输入有误，请重新输入')



def add_admin():
    '''
    添加管理员

    :return:
    '''
    while True:
        name = input('请输入姓名')
        a = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
        line = pickle.load(open(os.path.join(a,'db','admin'), 'rb'))
        if name in line:
            name1 = input('此用户已存在,y = 重新输入，其他键 = 退出')
            if name1 == 'y':
                continue
            else:
                break
        else:
            password = input('请输入密码')
            line[name]= password
            with open('record', 'a') as f:
                f.write("  %s 添加 %s
" % (A, name))
            pickle.dump(line, open(os.path.join(a, 'db', 'admin'), 'wb'))
            break


def remove_user():
    '''
    删除用户
    :return:
    '''
    while True:
        name = input('请输入需要删除的用户')
        a = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
        if os.path.exists(os.path.join(a,'db',name)) is True:
            rm = input('请确认删除，q = 退出， y = yes, n = 其他键')
            if rm == 'y':
                os.remove(os.path.join(a,'db',name))
                print('删除成功')
                with open('record', 'a') as f:
                    f.write("  %s 删除 %s
" % (A, name))
                break
            else:
                break
        else:
            print('用户不存在,请重新输入')


def amend_user():
    '''
    修改用户信息
    :return:
    '''

    while True:
         name = input('请输入要修改的用户, q = 退出')
         a = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
         if name == 'q':
             break
         elif os.path.exists(os.path.join(a,'db',name)) is True:
             line1 = pickle.load(open(os.path.join(a,'db',name), 'rb'))
             print (line1)
             key = input('请输入要修改的选项')
             item = input('请输入要修改的内容')
             line1[key] = item
             print(line1)
             ok = input('请确认是否正确，y = yes, 其他键 = no, q =退出')
             if ok =='q':
                 break
             elif ok == 'y':
                 with open('record', 'a') as f:
                    f.write("  %s 修改 %s %s 的 %s
" % (A, name, key, item))
                 pickle.dump(line1, open(os.path.join(a,'db',name), 'wb'))
                 break
             else:
                 continue
         else:
            print('用户不存在,请重新输入')

def check():
    '''
    查看管理员
    :return:
    '''

    a = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
    line = pickle.load(open(os.path.join(a,'db','admin'), 'rb'))
    for key in line:
        print (key, line[key])

def ck_ls():
    '''
    查看用户名单
    :return:
    '''
    a = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
    file = os.listdir(os.path.join(a,'db'))
    file.remove('admin')
    print ('用户名单：')
    for i in file:
        print (i)

import sys
sys.path.append("..")
from config import admin
from config import user
from config import shop
import datetime

B = datetime.datetime.now()
def main():
    T = '0'
    while True:
        login = input('请选择 1 普通用户；2 管理员; 3 注册; 4 退出')
        if login == '1':
            if T == '0':
                name = user.login()
            choice = input('请选择 1 购物； 2 进入ATM; 其他键 = 退出 ')
            if choice == '1':
                while True:
                    mk = shop.ls()
                    T = user.charge(name, mk[-1])
                    with open('record', 'a') as f:
                        f.write('%s %s 购物名单 %s' % (name, B, mk))
                    if T == '1':
                        continue
                    elif T == '2':
                        break
                    elif T == '3':
                        exit()


            elif choice == '2':
                user.ls(name)
                break
            else:
                exit()
        elif login == '2':
            admin.login()
            admin.ls()
            break
        elif login == '3':
            user.add_user()
        elif login == '4':
            exit()
        else:
            print('输入有误，请重新输入')

import datetime
B =datetime.datetime.now()

# 创造三级菜单
shoppingmall = {'sport': {'clothes':{'coat':{20: 80}, 'shirt':  {30: 100}, 'skirt': {20: 120}},
                            'tools': {'net': {20: 70}, 'bat': {40: 100}, 'fish pole': {20: 90}},
                            'ball': {'basketball': {90: 200}, 'pingpong': {10: 20}, 'football': {50: 170}}},
                'electrical equipment': {'phone': {'Apple': {10: 6000}, 'Xiaomi': {15: 2000}, 'Huawei': {5: 3000}},
                          'computer': {'HP': {2: 4000}, 'Lenovo': {6: 2200}, 'Dell': {10: 3000}},
                          'TV': {'Sony': {60: 5000}, 'LG': {30: 3000}, 'TCL': {40: 2200}}},
                'book': {'baby': {'Story': {50: 37},'Picture': {30: 64},'Music': {60: 93}},
                          'child': {'Exercise': {30: 36},'Earth': {20: 55},'Weather': {30: 64}},
                          'adult': {'Leaning': {40: 45},'Cookbook': {50: 74},'Tourism': {50: 43}}}}

def ls():
    list1=[]
    list2=[]
    list3=[]
    shopping_list=[]
    item_price={}
    while True:
        for key in shoppingmall:
            print(key)
            list1.append(key)
# 选择一项到下一级菜单或者退出
        choose = input('Please choose thing or q = quit:')
        if choose == 'q':
            exit('Bye!')
# 进入下一级菜单
        elif choose in list1:
            a = shoppingmall.get(choose)
            while True:
                for key1 in a:
                    print(key1)
                    list2.append(key1)
            # 选择一项到下一级菜单或者退出或返回
                choose1 = input('Please chose that you want or q = quit or b = back:')
                if choose1 == 'q':
                    exit('Bye!')
                elif choose1 == 'b':
                    break
                elif choose1 in list2:
                # 进入下一级菜单
                    b = a.get(choose1)
                    while True:
                        for key2 in b:
                            c = b.get(key2)
                            list3.append(key2)
                            for number in c:
                                price = c.get(number) #number剩余数量，price 商品价钱
                                item_price[key2] = price#商品和对应的价钱放入字典中
                                print ('%s is $%s left %s' % (key2, item_price[key2], number))
# 选择退出，返回上一级菜单 或者第一个菜单
                        choose2 = input('What do you want to buy? n = quit，b = back.')
                        if choose2 == 'n':
                            exit('Bye!')
                        elif choose2 == 'b':
                            break
                        elif choose2 in list3:
                            quantity = int(input('Please choose quantity'))
                            if quantity <=number: #检查购买数量是否大于存货量
                                if choose2 in shopping_list:
                                    local = shopping_list.index(choose2)
                                    quantity = int(quantity) + int(shopping_list[local+2])
                                    goods_price = (quantity * item_price[choose2])
                                    shopping_list.pop(local+1)
                                    shopping_list.pop(local+1)
                                    shopping_list.insert(local+1, goods_price)
                                    shopping_list.insert(local+2, quantity) #物品数量放入购物名单中
                                else:
                                    shopping_list.append(choose2)
                                    goods_price = (quantity * item_price[choose2])
                                    shopping_list.append(goods_price)
                                    shopping_list.append(quantity) # 物品数量放入购物名单中
                                shop_thing = shopping_list[::3]
                                shop_num = shopping_list[2::3]
                                print('Shopping list has:')
                                ii = len(shop_thing)

                                for item in range(ii): # 输出购买名单
                                    print ('%s has %s'% (shop_thing[item], shop_num[item]))
                                last = input('Goods have been put in shopping_list, what do you want to do? '
                                    'c = check out, q = quit,other keys = other thing I want to buy.')
                                if last == 'c':
                                    #计算总数
                                    total_price = 0
                                    for h in range(len(shop_thing)):
                                        total_price += goods_price
                                    print('Total price is %s' % (total_price))
                                    shopping_list.append(total_price)
                                    #shopping_list[购买物品，这个物品总价，购买数量，总价]
                                    return shopping_list
                                elif last == 'q':
                                    exit()
                                else:
                                    continue
                            else:
                                print ('Quantity is over, please choose again')
                        else:
                            print ('Yout choose is over the choose, please choose again')
                else:
                    print('Sorry, your choose not in list! Please choose again')
        else:
             print('Sorry, your choose not in list! Please choose again')

import os
import pickle
import datetime


A = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
B = datetime.datetime.now()



def login():
    '''
    普通用户登录
    :return:
    '''
    T = True
    while T:
        name = input('Please enter name:')
        password = input('Please enter password:')
        if os.path.exists(os.path.join(A, 'db', name)) is True:
            line = pickle.load(open(os.path.join(A,'db',name), 'rb'))
            if line['status'] == '0':
                if line['name'] == name and line['password'] == password:
                    print ('欢迎 %s!' % name)
                    with open('record', 'a') as f:
                        f.write("%s %s 成功登陆
" % (name, B))
                    T = False
                    return name
                else:
                    print('用户名或密码错误，请重新输入')
            else:
                print ('你的账号已被冻结，请联系管理员')
                break
        else:
            print ('没有该用户信息，请先注册')
            add_user()


def ls(name):
    '''

    :return:
    '''
    while True:
        choice = input('请输入需要的选项：1 取款；2 查看账单，还款还款；3 转账, 4 修改还款日；5 退出')
        if choice == '1':
            cash(name)
        elif choice == '2':
            pay (name)
        elif choice == '3':
            bring_fd(name)
        elif choice == '4':
            change_pay(name)
        elif choice == '5':
            exit()
        else:
            print('输入有误，请重新输入')


def cash(name):
    '''
    提取现金
    :param name:
    :return:
    '''
    while True:
        line = pickle.load(open(os.path.join(A,'db',name), 'rb'))
        mk = input('余额为 %.2f， 可以透支 %.2f， 请输入提取多少现金，或者 q = 退出' % (line['balance'], line['credit']))
        if mk == 'q':
             break
        elif float(mk) > 0:
            mk1 = float(line['balance'])
            mk2 = float(line['credit'])
            if mk1>float(mk):
                mk1 -= float(mk)
                line['balance'] = mk1
                print('取款成功')
                pickle.dump(line, open(os.path.join(A,'db',name), 'wb'))
                break
            elif mk1+mk2>=float(mk):
                mk1-=float(mk)
                mk2-=abs(float(mk1))
                line['balance'] = 0
                line['credit'] = mk2
                line['debt'].append(B)
                line['debt'].append(abs(float(mk1))*0.05+abs(float(mk1)))
                print('取款成功,手续为5%')
                with open('record', 'a') as f:
                    f.write("%s %s 提取 %s
" % (name, B, mk))
                pickle.dump(line, open(os.path.join(A,'db',name), 'wb'))
                break
            else:
                print('卡的余额不足，请重新输入')
        else:
            print('输入有误，请重新输入')




def add_user():
    '''
    添加新用户
    :return:
    '''

    while True:
        name = input('请输入姓名')
        if os.path.exists(os.path.join(A, 'db', name)) is True:
            name1 = input('此用户已存在,y = 重新输入，其他键 = 退出')
            if name1 == 'y':
                continue
            else:
                break
        else:
            password = input('请输入密码')
            print('注册成功')
            with open('record', 'a') as f:
                f.write("%s %s 成功注册
" % (name, B))
            info = {'name': name,
                'password': password,
                'credit': 15000,# 透支额度
                'balance': 0, # 余额
                'status': '0',  # 0 = 正常，1 = 冻结
                'debt': [],  #欠款
                'debt1': [],#上期欠款没有还玩
                'pay_date':[15] #还款日期，默认15号
                }
            pickle.dump(info, open(os.path.join(A,'db',name), 'wb'))
            break


def pay (name):
    '''
    查看账单,还款还款，22号出账单，指定日期还款，过期不还，收万分之5利息
    :param name:
    :return:
    '''
    total1 = 0
    total2 = 0
    ls=[] #不用计利息
    ls1=[] #计利息
    pay_ls=[]
    line = pickle.load(open(os.path.join(A,'db',name), 'rb'))
    credit = line['credit']
    for a in line['debt'][0::2]:
        #当月和上月10号之后消费的，不用计利息的放入ls
        if a.month - B.month == 0 or B.month - a.month == 1 and (a.day >=22 or B.day<=line['pay_date']):
            c = line['debt'].index(a)+1
            ls.append(line['debt'][c])
        #上个月22号前计利息的放入ls1
        else:
            c = line['debt'].index(a)+1
            day = (B - a).days
            ls1.append(day)
            ls1.append(line['debt'][c])
    if line['debt1']==[]:
        total3 = 0
    else:
        #欠钱的总和利息加欠款
        total3 = total2*(B-line['debt1'][0]).days * 0.0005+line['debt1'][1]
    #不计利息欠钱的总数
    for i in ls:
        total1 += i
    #把每个利息和本金的和放入pay_ls
    for i1 in ls1[::2]:
        d = ls1.index(i1)+1
        e = ls1[d]
        h = e * i1 * 0.0005 + e
        pay_ls.append(h)
    #计算这期账单
    for i3 in pay_ls:
        total2 += i3
    total2+=total3
    #一共欠款
    total = total1+total2
    print('你需要还%.2f, 还可以透支 %.2f,余额为%s' % (total2, credit, line['balance']))
    if total2 > 0 or credit < 15000:
        mk = input('请输入还款金额, q = 退出,')
        if mk == 'q':
            with open('record', 'a') as f:
                f.write("%s %s 查看账单
" % (name, B))
                exit()
        elif float(mk)>0 or float(mk)>0:
            #还款金额大于欠款
            if total2 <= float(mk):
                #还款金额大于欠款加透支
                if total <float(mk):
                    pay = total - float(mk)
                    line['balance']=abs(pay)+line['balance']
                    line['debt']=[]
                    line['debt1']=[]
                    line['credit']= 15000
                    print('款项已还清')
                else:
                    pay = total - float(mk)
                    if pay == 0:
                        line['credit']= 15000
                        line['debt']=[]
                        line['debt1'] =[]
                        print('款项已还清')
                    else:
                        credit = 15000-pay
                        line['credit']= credit
                        line['debt']=[]
                        line['debt'].append(B)
                        line['debt'].append(pay)
                        print('你需要还%.2f, 还可以透支 %.2f,余额为%s' % (total2, credit, line['balance']))
            else:
                pay = total2 - float(mk)
                print('还欠%.2f，请尽快归还' % pay)
                line['debt']=[]
                line['debt1']=[]
                line['debt1'].append(B)
                line['debt1'].append(pay)
                line['debt1'].append(total2)
                line['debt'].append(B)
                line['debt'].append(total1)

        else:
            print('输入有误')
        with open('record', 'a') as f:
            f.write("%s %s 还款 %s
" % (name, B, mk))
        pickle.dump(line, open(os.path.join(A,'db',name), 'wb'))

    else:
        chooic = input('存多少钱, q = 退出')
        if chooic == 'q':
            with open('record', 'a') as f:
                f.write("%s %s 查看账单
" % (name, B))
                exit()
        elif float(chooic)>0:
            balance = line['balance'] + float(chooic)
            line['balance'] = balance
            line['debt']=[]
            line['debt1']=[]
            print ('你的余额为%.2f' % balance)
        else:
            print('输入有误')
        with open('record', 'a') as f:
            f.write("%s %s 存款 %s
" % (name, B, chooic))
        pickle.dump(line, open(os.path.join(A,'db',name), 'wb'))


def bring_fd(name):
    '''
    用户之间转账
    :param name:
    :return:
    '''
    while True:
        line = pickle.load(open(os.path.join(A,'db',name), 'rb'))
        print('你账号余额为%s' % line['balance'])
        pay_name= input('请输入要转账的账号名, q = 退出')
        if pay_name == 'q':
            exit()
        if pay_name == name:
            print('不能转账给相同账户')
            continue
        if os.path.exists(os.path.join(A, 'db', pay_name)) is True:
            pay_line = pickle.load(open(os.path.join(A,'db',pay_name), 'rb'))
            mk = input('请输入转账数目')
            if line['balance'] > float(mk):
                line['balance'] = float(line['balance']) - float(mk)
                pay_line['balance'] = float(pay_line['balance']) + float(mk)
                pickle.dump(line, open(os.path.join(A,'db',name), 'wb'))
                pickle.dump(pay_line, open(os.path.join(A,'db',pay_name), 'wb'))
                print('转账成功')
                with open('record', 'a') as f:
                    f.write("%s %s 转了 %s元 给 %s 
" % (name, B, mk, pay_name))
                break
            else:
                print ('余额不足，请重新输入')
        else:
            print('该账号不存在，请重新输入')


def charge(name, mk):
    '''
    付款
    :param name:
    :param mk:
    :return:
    '''
    line = pickle.load(open(os.path.join(A,'db',name), 'rb'))
    a = float(line['credit'])
    b = float(line['balance'])
    c = float(mk)
    yes = input('请输入是否付款？y = yes,其他键退出')
    if yes == 'y':
        if b > c:
            line['balance']= a-c
            cont = input('付款成功, 1 继续购物 2 返回上级菜单 3 退出')
            pickle.dump(line, open(os.path.join(A,'db',name), 'wb'))
            with open('record', 'a') as f:
                f.write("%s %s 支付了 %s 
" % (name, B, mk))
            return cont
        elif a+b >c:
            line['0']=0
            line['balance']= 0
            line['credit']= a-c
            line['debt'].append(B)
            line['debt'].append(c)
            cont = input('付款成功, 1 继续购物 2 返回上级菜单; 其他键 = 退出')
            pickle.dump(line, open(os.path.join(A,'db',name), 'wb'))
            with open('record', 'a') as f:
                f.write("%s %s 支付了 %s 
" % (name, B, mk))
            return cont
        else:
            print('余额不足')
    else:
        exit()

def change_pay(name):
    '''
    设置还款日期
    :param name:
    :return:
    '''
    line = pickle.load(open(os.path.join(A,'db',name), 'rb'))
    while True:
        date = input('请设置还款日，q=退出')
        if date == 'q':
            break
        elif date.isdigit():
            line['pay_date']=int(date)
            print('你的还款日期改为%s' % (date))
            with open('record', 'a') as f:
                f.write("%s %s 还款日改成 %s 
" % (name, B, date))
            pickle.dump(line,open(os.path.join(A,'db',name), 'wb'))
            break
        else:
            print('输入有误，请重新输入')