练手小程序用了pandas模块和json模块以及time模块

6.16自我总结

功能介绍

程序功能介绍:
    商品信息再读取修改买卖均已xlsx格式
    且生成购物记录也按/用户名/购买时间.xlsx格式生成
    账号密码输入错误三次按照时间进行冻结
    用户信息已json格式保存
 程序写的过程
先生成功能模块和运行模块
再写功能模块中用到的固定的文件目录全放在setting.py文件中
与商品交互全放在shop.py中
与用户交互全放在user.py中
一些返回界面延迟动画全放在辅助模块中

1.程序设计目录

2.run.py

from core.src import run
if __name__ == '__main__':
    run()

3.src.py

from conf.setting import *
from lib.common import *
from api.user import *
from api.shop import *
import re

import numpy as np

goods_count = [1]
#注册
def register():
    print('\033[47;;m\t\t欢迎使用注册功能\t\t\033[0m')
    count = 0
    while count == 0:
        register_name = input('请输入名字按q退出程序\n'
                              '\033[31;;m账号由汉字,字母,数字,下划线组成\033[0m'
                              '\n请输入:')
        if register_name == 'q':
            go_to_run()
            return false
        if decide_user_name(register_name):
            print('\033[31;;m账号存在\033[0m')
            continue
        register_name_1 = re.findall('\w+',register_name)
        if len(register_name) == 0:
            print('\033[31;;m请好好输入\033[0m')
            continue
        elif register_name_1[0] != register_name:
            print('\033[31;;m账号由汉字,字母,数字,下划线组成\033[0m')
            continue
        else:
            count =1
            while true:
                register_pwd = input('请输入密码按q退出程序\n'
                                     '\033[31;;m密码由汉字,字母,数字,下划线组成\033[0m\n'
                                     '请输入')
                if register_pwd == 'q':
                    go_to_run()
                    return false
                register_pwd_1 = re.findall('\w+', register_pwd)
                if len(register_pwd) == 0:
                    print('\033[31;;m请好好输入\033[0m')
                    continue
                elif register_pwd_1[0] != register_pwd:
                    print('\033[31;;m密码由汉字,字母,数字,下划线组成\033[0m')
                    continue
                else:
                    save_info(register_name, register_pwd)
                    print('\033[32;;m注册成功\033[0m')
                    go_to_run()
                    return true

#登入
def login():
    print('\033[47;;m\t\t欢迎使用登入功能\t\t\033[0m')
    if not login_name[0]:
        pwd_count =0
        login_count = 0
        while login_count==0:
            login_name = input('请输入名字按q退出程序\n请输入:')
            if login_name == 'q':
                go_to_run()
                return false
            if not decide_user_name(login_name):
                print('\033[31;;m账号不存在\033[0m')
                continue
            if not load_freeze_user(login_name):
                go_to_run()
                return false
            else:
                login_count = 1
                while pwd_count<3:
                    login_pwd = input('\033[41;;m请输入密码输错三次会被冻结五分钟\033[0m\n请输入:')
                    if not load_info(login_name,login_pwd):
                        pwd_count +=1
                        print(f'\033[31;;m还有{3-pwd_count}次机会\033[0m')
                        continue
                    else:
                        print('\033[32;;m登入成功\033[0m')
                        login_name[0] = login_name
                        go_to_run()
                        return true
                if pwd_count == 3:
                    freeze_user(login_name)
                    print(f'\033[31;;m账号{login_name}由于密码输入太多次暂时被冻结\033[0m')
                    go_to_run()
                    return false
    else:
        print(f'\033[32;;m账号{login_name[0]}以及登入')
        login_chiose = input('输入0为退出当前程序\n'
                             '输入1为退出当前账号重新登入\n'
                             '输入2为退出当前账号并退出程序\n'
                             '请选择:')
        if login_chiose not in ['0','1','2']:
            print('\033[31;;m请好好输入\033[0m')
        elif login_chiose == '0':
            go_to_run()
            return false
        elif login_chiose == '1':
            df = load_goods_pach()
            dump_goods(df)
            login_name[0] = none
            login()
        elif login_chiose == '2':
            df = load_goods_pach()
            dump_goods(df)
            login_name[0] = none
            print('\033[41;;m账号以退出\033[0m')
            go_to_run()
            return false

#充值
@login_deco
def top_up():
    print('\033[47;;m\t\t欢迎使用充值功能\t\t\033[0m')
    while true:
        chiose = input('请输入充值金额\n输入q退出功能\n请输入')
        if chiose  == 'q':
            go_to_run()
            return false
        elif not chiose.isdigit():
            print('\033[31;;m请输入阿拉伯数字\033[0m')
            continue
        else:
            info = load_login_info(login_name[0])
            info["balance"] += int(chiose)
            dump_login_info(login_name[0],info)
            print(f'\033[42;;m账号{login_name[0]}充值成功\n当前余额{info["balance"]}元\033[0m')
            go_to_run()
            return true

#余额查询
@login_deco
def balance():
    print('\033[47;;m\t\t欢迎使用余额查询功能\t\t\033[0m')
    info = load_login_info(login_name[0])
    print(f'\033[42;;m账号{login_name[0]}\n当前余额{info["balance"]}\033[0m')
    go_to_run()
    return true

#提现
@login_deco
def withdraw():
    print('\033[47;;m\t\t欢迎使用提现功能\t\t\033[0m')
    while true:
        chiose = input('\033[32;;m请输入提现金额\n输入q退出功能\n请输入')
        if chiose  == 'q':
            go_to_run()
            return false
        elif not chiose.isdigit():
            print('\033[31;;m请输入阿拉伯数字\033[0m')
            continue
        else:
            info = load_login_info(login_name[0])
            if info["balance"] < int(chiose):
                print('\033[31;;m余额不足\033[0m')
                continue
            info["balance"] -= int(chiose)
            dump_login_info(login_name[0],info)
            print(f'\033[41;;m账号{login_name[0]}提现成功\n当前余额{info["balance"]}元\033[0m')
            go_to_run()
            return true

#转账
@login_deco
def transfer():
    print('\033[47;;m\t\t欢迎使用转账功能\t\t\033[0m')
    count = 0
    while count ==0:
        chiose = input('\033[32;;m请输入转账金额\n输入q退出功能\n请输入')
        if chiose  == 'q':
            go_to_run()
            return false
        elif not chiose.isdigit():
            print('\033[31;;m[请输入阿拉伯数字\033[0m')
            continue
        else:
            info = load_login_info(login_name[0])
            if info["balance"] < int(chiose):
                print('\033[31;;m余额不足\033[0m')
                continue
            info["balance"] -= int(chiose)

            while true:
                count =1
                transfer_name = input('\033[32;;m转账的名字\n输入q退出')
                if transfer_name == 'q':
                    go_to_run()
                    return false
                if not decide_user_name(transfer_name):
                    print('\033[31;;m转账账号不存在\033[0m')
                    continue
                if login_name[0] == transfer_name:
                    print('\033[31;;m不能转给自己\033[0m')
                    continue
                transfer_name_info = load_login_info(transfer_name)
                transfer_name_info["balance"] += int(chiose)
                dump_login_info(transfer_name, transfer_name_info)
                dump_login_info(login_name[0], info)
                print(f'\033[42;;m账号{login_name[0]}转账成功\n当前余额{info["balance"]}元\033[0m')
                go_to_run()
                return true

#流水
@login_deco
def user_history():
    print('\033[47;;m\t\t欢迎使用流水功能\t\t\033[0m')
    if not history(login_name[0]):
        print('\033[31;;m没有购买记录\033[0m')
        go_to_run()
        return false
    while true:
        date_lis = show_history(login_name[0])
        chiose = input('\033[32;;m请输入你要查看的日期\n'
                       '输入q退出')
        if chiose not in date_lis:
            print('\033[31;;m没有日期\033[0m')
            continue
        print(f'{chiose}')
        print(load_goods_history(login_name[0],chiose))
        history_count = 0
        while history_count == 0:
            next_chiose = input('\033[32;;m请输入y继续查看的日期\n'
                                '输入q退出\n'
                                '请输入')
            if next_chiose not in ('y','q'):
                print('\033[31;;m请好好输入\033[0m')
            elif next_chiose == 'y':
                history_count =1
            elif next_chiose =='q':
                go_to_run()
                return false
#购物
@login_deco
def shopping():
    count = 0
    chiose_count =0
    print('\033[47;;m\t\t欢迎使用购物功能\t\t\033[0m')
    global goods_count
    while count == 0:
        if not goods_count[0]:
            df = load_goods()
            print('\t\t\t商品目录')
            print(f'\033[35;36;m{df}\033[0m')
        else:
            df = load_goods_pach()
            print('\t\t\t商品目录')
            print(f'\033[35;36;m{df}\033[0m')

        goods = input('\033[32;;m请选择你的商品\n'
                      '输入q退出\n'
                      '请选择')
        goods_list = df.columns
        if goods == 'q':
            go_to_run()
            return false
        elif goods not in goods_list:
            print('\033[31;;m无此商品\033[0m')
            continue
        else:
            chiose_count =0
            while chiose_count == 0:
                num = input('\033[32;;m请选择你的商品数量\n'
                              '输入q退出\n'
                            '请选择')
                goods_num = df[goods]['数量']
                if goods == 'q':
                    go_to_run()
                    return false
                elif not num.isdigit():
                    print('\033[31;;m请输入数字\033[0m')
                    continue
                elif int(goods_num)<int(num):
                    print('\033[31;;m库存不足\033[0m')
                else:
                    df[goods]['数量'] -= int(num)
                    goods_num = int(num)
                    dump_goods_pach(df)
                    print(f'\033[42;;m你把{goods}{num}个加入购物车\033[0m')
                    while chiose_count == 0:
                        chiose =input('\033[32;;m是否继续购物\n'
                                      'y是继续,n是退出\n'
                                      '请选择')
                        if chiose not in ['y','n']:
                            print('\033[31;;m请好好输入\033[0m')
                            continue
                        if chiose == 'n':

                            goods_count[0] = 1
                            go_to_run()
                            return true
                        if chiose == 'y':
                            chiose_count = 1
                            goods_count[0] = 1
                            pass




#购物车
@login_deco
def shopping_car():
    print('\033[47;;m\t\t欢迎使用购物车功能\t\t\033[0m')
    global goods_count
    if not goods_count[0]:
        print('\033[31;;m购物车无商品\033[0m')
    else:
        df_1 = load_goods()
        df_2 = load_goods_pach()
        df = df_1 - df_2
        print('-'*50)
        print('\t\t\t购物车目录')
        print(f'\033[35;36;m{df}\033[0m')
        df_mun = df.values[1,:]
        df_pice = df_1.values[0,:]
        df_add = df_mun*df_pice
        money = sum(df_add)
        print(f'合计{money}元')
        print('-' * 50)
        while true:
            chiose = input('\033[32;;m输入q退出程序\n'
                           '输入0清空购物车并退出程序\n'
                           '输入1结算\n'
                           '请选择')
            if chiose not in ['q','0','1']:
                print('\033[31;;m请好好输入\033[0m')
                continue
            elif chiose == 'q':
                go_to_run()
                return false
            elif chiose == '0':
                goods_count[0] = none
                df_new = load_goods()
                dump_goods_pach(df_new)
                go_to_run()
                return false
            elif chiose == '1':
                info = load_login_info(login_name[0])
                if int(money) > info['balance']:
                    print('\033[31;;m余额不足\033[0m')
                    continue
                else:
                    print('\033[41;;m支付成功\033[0m')
                    info['balance'] -= int(money)
                    dump_login_info(login_name[0],info)
                    df_new = load_goods_pach()
                    dump_goods(df_new)
                    df_mun = df[1:2]
                    df_pice = df_1[0:1]
                    df = df.columns
                    new_df = df_pice.append(df_mun)
                    print(new_df)
                    time =time_strftime()
                    dump_goods_history(login_name[0],time,new_df)
                    go_to_run()
                    return true











#运行模块
def run():
    action_dict={
        '0':register,
        '1':login,
        '2':top_up,
        '3':balance,
        '4':withdraw,
        '5':transfer,
        '6':user_history,
        '7':shopping,
        '8':shopping_car,
    }

    while true:
        if login_name[0]:
            print(f'你好{login_name[0]}')
        for action_num,action in action_info.items():
            print(f'\033[35;;m\t\t输入{action_num}功能为{action}\033[0m')
        action_chiose = input('\033[32;;m请输入你要选择的功能:')
        if action_chiose == 'q':
            print('退出程序')
            return
        if action_chiose not in action_dict:
            print('\033[31;;m输入错误\033[0m')
            continue
        action_dict[action_chiose]()

if __name__ == '__main__':
    run()

3.common.py

from conf.setting import *
import time

def login_deco(func):
    def wrapper(*args,**kwargs):
        if not login_name[0]:
            print('请先登入')
            go_to_run()
            return false
        func()
        return true
    return wrapper

def go_to_run():
    for a in range(20):
        time.sleep(0.1)
        txt = '\t\t返回主界面中'
        txt += '.'*int(a%4)
        print('\r',f'\033[32;;m{txt}\033[0m',end='')
    print('')

def time_now():
    return time.time()

def time_strftime():
    return str(time.strftime('%y-%m-%d-%h-%m-%s'))

4.setting.py

import os
import time
action_info={
        '0':'注册',
        '1':'登入',
        '2':'充值',
        '3':'余额查询',
        '4':'提现',
        '5':'转账',
        '6':'购物历史记录',
        '7':'购物',
        '8':'购物车',
        'q' :'退出'
    }

login_name = [none]


atm_path = os.path.dirname(os.path.dirname(__file__))
user_path = os.path.join(atm_path,'db')
goods_path = os.path.join(atm_path,'db','goods_info.xlsx')
goods_patch = os.path.join(atm_path,'db','goods_info_patch.xlsx')

5.shop.py

(商品信息存储文件与功能模块的交互)

import pandas as pd
from conf.setting import goods_path,goods_patch
import os

def dump_goods(df):
    df.to_excel(goods_path)

def dump_goods_pach(df):
    df.to_excel(goods_patch)

def load_goods():
    df = pd.read_excel(goods_path,index_col=0,header=0)
    return df
def load_goods_pach():
    df = pd.read_excel(goods_patch,index_col=0,header=0)
    return df

def dump_goods_history(name,time,df):
    path_1 = os.path.join(r'e:\atm\db', name)
    path = os.path.join(r'e:\atm\db',name,f'{time}.xlsx')
    if not os.path.exists(path_1):
        os.mkdir(path_1)
    df.to_excel(path)

def load_goods_history(name,time):
    path = os.path.join(r'e:\atm\db', name, f'{time}.xlsx')
    df = pd.read_excel(path,index_col=0,header=0)
    return df

def show_history(name):
    new_list =[]
    path_1 = os.path.join(r'e:\atm\db', name)
    lis = os.listdir(path_1)
    print('\033[46;;m提示:年-月-日-时-分-秒\033[0m')
    for info in lis:
        info = info[0:-5]
        print(info)
        new_list.append(info)
    return new_list

def history(name):
    path_1 = os.path.join(r'e:\atm\db', name)
    if os.path.exists(path_1):
        return true
    return false
if __name__ == '__main__':
    df = load_goods_pach()
    print(df)

6.user.py

(用户信息存储文件与功能模块的交互)

from conf.setting import *
import json
import hashlib
from lib.common import time_now

def save_info(user_name,pwd):
    m = hashlib.md5()
    m.update(pwd.encode('utf8'))
    pwd =m.hexdigest()
    user_path = os.path.join(user_path, f'{user_name}.json')
    with open(user_path,'w',encoding='utf8') as fw:
        info_dict = {'name':user_name,'pwd':pwd,'freeze':time_now(),'balance':0,'freeze_count':0}
        json.dump(info_dict,fw)

def load_info(user_name,pwd):
    user_path = os.path.join(user_path, f'{user_name}.json')
    if not login_name[0]:
        if not os.path.exists(user_path):
            print('\033[31;;m用户不存在\033[0m')
            return false
        m = hashlib.md5()
        m.update(pwd.encode('utf8'))
        pwd =m.hexdigest()
        with open(user_path, 'r', encoding='utf8') as fr:
            info_dict = json.load(fr)
        if info_dict.get('pwd') != pwd:
            print('\033[31;;m密码错误\033[0m')
            return false
    with open(user_path, 'r', encoding='utf8') as fr:
        info_dict = json.load(fr)
    info_dict['freeze_count'] = 0
    with open(user_path, 'w', encoding='utf8') as fw:
        json.dump(info_dict,fw)
    return info_dict

def decide_user_name(user_name):
    user_path = os.path.join(user_path, f'{user_name}.json')
    if  os.path.exists(user_path):
        return true
    return false

def freeze_user(user_name):
    user_path = os.path.join(user_path, f'{user_name}.json')
    with open(user_path, 'r', encoding='utf8') as fr:
        info_dict = json.load(fr)
    info_dict['freeze_count'] += 1
    info_dict['freeze'] = time_now() + 300*info_dict['freeze_count']
    with open(user_path,'w',encoding='utf8') as fw:
        json.dump(info_dict,fw)

def load_freeze_user(user_name):
    user_path = os.path.join(user_path, f'{user_name}.json')
    with open(user_path, 'r', encoding='utf8') as fr:
        dict = json.load(fr)
    if time_now() >= dict['freeze'] :
        print('\033[32;;m账号登入成功\033[0m')
        return true
    else:
        min = int(divmod(-time_now() + dict['freeze'], 60)[0])
        s = int(divmod(-time_now() + dict['freeze'], 60)[1])
        print(f'\033[31;;m账号{user_name}已被冻结,还需要{min}分{s}秒\033[0m')
        return false

def load_login_info(name):
    login_user_path = os.path.join(user_path, f'{name}.json')
    with open(login_user_path, 'r', encoding='utf8') as fr:
        info_dict = json.load(fr)
    return info_dict

def dump_login_info(name,dict):
    login_user_path = os.path.join(user_path, f'{name}.json')
    with open(login_user_path, 'w', encoding='utf8') as fw:
        json.dump(dict,fw)
        return true

if __name__ == '__main__':
    save_info('杨文益','12312')
    dict = load_info('杨文益','12312')
    print(dict['name'])

7.goods_info.xlsx/goods_info_patch.xlsx

	苹果	香蕉	西瓜	荔枝	梨
价格	2	3	10	5	5
数量	200	215	34	32	523