start
# 启动文件：在这个文件中右键运行代码 # import sys # # print(sys.path) from core import src if __name__ == ‘__main__‘: src.run()
src
import os import json user_data = { ‘username‘: None, ‘is_login‘: False } # 注册功能 def register(): print("注册功能：") if user_data[‘is_login‘]: print("已经登录") return while True: username = input(‘请输入用户名:>>‘).strip() # path = os.path.dirname(os.path.dirname(__file__)) path = settings.BASE_DIR # day19/db/egon.json path_file = os.path.join(path, ‘db‘, ‘%s.json‘ % username) # 判断是否存在, 其实就是判断文件是否存在 if os.path.exists(path_file): # 即可以判断文件也可以判断文件夹是否存在 # if os.path.isfile(path_file): # 可以判断文件 # if os.path.isdir(path_file): # 可以判断文件夹 dir => directory => 文件夹 print("用户名已经存在") continue pwd = input(‘请输入密码:>>‘).strip() # 注册 user_dic = {‘username‘: username, ‘password‘: pwd, ‘account‘: 10000, ‘locked‘: False, ‘flow‘: []} # 写文件 with open(path_file, ‘w‘, encoding=‘utf-8‘) as f: json.dump(user_dic, f) # 写日志 from lib import common common.write_log("%s注册成功" % username) print("恭喜哦~注册成功") break """ user_data = { ‘username‘: egon, ‘is_login‘: True } """ from conf import settings def login(): print("登录功能：") count = 0 if user_data[‘is_login‘]: print("已经登录") return while True: username = input(‘请输入用户名：>>‘).strip() path = settings.BASE_DIR path_file = os.path.join(path, ‘db‘, ‘%s.json‘ % username) if not os.path.isfile(path_file): print("用户名不存在") continue # 查询用户信息 with open(path_file, ‘r‘, encoding=‘utf-8‘) as f: data = json.load(f) # dict if data.get(‘locked‘): print("用户已经锁定，请联系我哦~") break pwd = input(‘请输入密码：>>‘).strip() # 判断密码是否正确 if pwd == data.get(‘password‘): print("登录成功") user_data[‘username‘] = username user_data[‘is_login‘] = True # 写日志 from lib import common common.write_log(‘%s登录了‘ % username) break else: print("密码错误") count += 1 if count == 3: data[‘locked‘] = True with open(path_file, ‘w‘, encoding=‘utf-8‘) as f: json.dump(data, f) """ user_data = { ‘username‘:None, ‘is_login‘:False } """ from lib import common from conf import settings import time # lib => library # 提现 @common.login_auth def withdraw(): """ 分析提现功能： 1. 输入提现金额 2. 扣除手续费， 5% 3. 判断金额是否足够， 4. 把金额减掉 5. 添加流水 6. 添加日志 :return: """ print("提现功能：") while True: # 1. 输入提现金额 money = input(‘请输入提现金额：>>‘).strip() if not money: continue if not money.isdigit(): print("输入金额不合法") continue money = int(money) # 2. 读取用户信息 # 读取当前用户的信息 path = settings.BASE_DIR path_file = os.path.join(path, ‘db‘, ‘%s.json‘ % user_data[‘username‘]) with open(path_file, ‘r‘, encoding=‘utf-8‘) as f: user_dic = json.load(f) # 判断金额是否足够 if user_dic[‘account‘] >= money * 1.05: # 修改金额 user_dic[‘account‘] -= money * 1.05 # 再次写入文件 with open(path_file, ‘w‘, encoding=‘utf-8‘) as f: json.dump(user_dic, f) print("提现成功，欢迎老板~") break else: print("穷逼一个") continue # 转账 @common.login_auth def transfer(): """ 分析转账功能： 1. 输入转账的账户，转账的账户必须存在 2. 输入转账金额 3. 检测自己的账户金额是否足够 4. 增加对方账户的金额，自己账户的金额要减掉 5. 写流水 6. 写日志 :return: """ print("转账功能：") while True: # 1. 输入转账的账户 to_user = input(‘请输入转账的账户用户名:>>‘).strip() # 2. 验证账户是否存在 path = settings.BASE_DIR to_path_file = os.path.join(path, ‘db‘, ‘%s.json‘ % to_user) # 判断转账的用户是否存在 if not os.path.exists(to_path_file): print("用户不存在") continue # 3. 输入转账金额 transfer_money = int(input(‘请输入转账金额：>>‘).strip()) # 4. 读取自己的账户金额是否足够 from_path_file = os.path.join(path, ‘db‘, ‘%s.json‘ % user_data[‘username‘]) with open(from_path_file, ‘r‘, encoding=‘utf-8‘) as f: from_user_dic = json.load(f) # 比较金额 if from_user_dic[‘account‘] >= transfer_money: # 对方账户增加金额 with open(to_path_file, ‘r‘, encoding=‘utf-8‘) as f1: to_user_dic = json.load(f1) to_user_dic[‘account‘] += transfer_money to_user_dic[‘flow‘].append(‘收到了%s的一笔转账%s元‘ % (from_user_dic[‘username‘], transfer_money)) with open(to_path_file, ‘w‘, encoding=‘utf-8‘) as f2: json.dump(to_user_dic, f2, ensure_ascii=False) # 自己的账户减掉金额 from_user_dic[‘account‘] -= transfer_money from_user_dic[‘flow‘].append(‘向%s转了一笔钱%s元‘ % (to_user_dic[‘username‘], transfer_money)) with open(from_path_file, ‘w‘, encoding=‘utf-8‘) as f3: json.dump(from_user_dic, f3, ensure_ascii=False) print("转账成功") break else: print("没钱啊，别装了。。。") """ 充值必须要登录，就要有一个验证登录的装饰器 """ # 充值 @common.login_auth def pay(): """ 分析充值的需求： 1. 要输入充值的金额 2. 要充值的账号必须存在 3. 把账户的余额修改掉 """ print("充值金额：") while True: # 1. 输入充值金额 money = input(‘请输入充值金额：>>‘).strip() # 验证输入的金额是否为数字 if not money.isdigit(): print("输入金额不是数字") continue money = int(money) """ 1. 拿到你现在的余额 2. 用你现在的余额加上money """ # 读取当前用户的信息 path = settings.BASE_DIR path_file = os.path.join(path, ‘db‘, ‘%s.json‘ % user_data[‘username‘]) # 判断当前用户是否存在 if not os.path.exists(path_file): print("用户不存在") continue # 读取用户信息 with open(path_file, ‘r‘, encoding=‘utf-8‘) as f: user_dic = json.load(f) # 获取当前时间 # ctime = time.time() # 2021- now = time.strftime(‘%Y-%m-%d %X‘) # 增加账号金额 user_dic[‘account‘] += money user_dic[‘flow‘].append(‘%s在%s充值了%s元‘ % (user_data[‘username‘], now, money)) # user_dic[‘flow‘].append(‘%s在%s充值了%s元‘) # {"username": "ly", "password": "123", "account": 10000, "locked": true} # with open(path_file, ‘w‘, encoding=‘utf-8‘) as f: json.dump(user_dic, f, ensure_ascii=False) # 写入流水 print("充值成功, 赶紧潇洒哦") break # 查看余额 def check_money(): pass # 查看流水 def check_flow(): pass # 遍历 # 加入购物车 @common.login_auth def add_cart(): """ 1. 得先有商品 goods = [ [‘商品名称‘, 100], [‘商品名称‘, 100], [‘商品名称‘, 100], ] 2. 让用户输入商品序号 3. 定义一个全局变量， 判断字典里是不是已经有了，没有的话，‘商品名称‘：{‘price‘：100，‘count‘: 3} 4. 在用户信息中，增加一个key: shopping_cart:{} 5. 日志 数据存储格式： {‘商品名称‘：{‘price‘：100，‘count‘: 2},‘商品名称1‘：{‘price‘：200，‘count‘: 1}} :return: """ print("加入购物车：") shopping_cart = {} while True: goods = [ [‘hulatang‘, 100], [‘tea‘, 200], [‘coffee‘, 300], [‘water‘, 400], ] """ 0. hulatang 100 1. tea 200 2. coffee 300 3. water 400 """ for k, v in enumerate(goods): print("%s %s %s" % (k, v[0], v[1])) num = input(‘请输入商品序号（q to quit）：>>‘).strip() # 0, 1, 2 ,3 if num.isdigit(): num = int(num) # 商品名称 goods_name = goods[num][0] goods_price = goods[num][1] # 去shopping_cart中判断是否存在 if goods_name not in shopping_cart: # {‘tea‘:{‘price‘: 100, ‘count‘:1}} shopping_cart[goods_name] = {‘price‘: goods_price, ‘count‘: 1} else: # {‘tea‘:{‘price‘: 100, ‘count‘:2}} shopping_cart[goods_name][‘count‘] += 1 print("请继续购物~") elif num == ‘q‘: # 把用户选择的商品写入文件中 if shopping_cart: # 读取用户信息 # 读取当前用户的信息 path = settings.BASE_DIR path_file = os.path.join(path, ‘db‘, ‘%s.json‘ % user_data[‘username‘]) with open(path_file, ‘r‘, encoding=‘utf-8‘) as f: user_dic = json.load(f) user_dic[‘shopping_cart‘] = shopping_cart with open(path_file, ‘w‘, encoding=‘utf-8‘) as f: json.dump(user_dic, f) print("欢迎下次再来") break else: print("输入序号不合法") # 查看购物车 def check_shop(): pass func_dic = { ‘1‘: register, ‘2‘: login, ‘3‘: withdraw, ‘4‘: transfer, ‘5‘: pay, ‘6‘: check_money, ‘7‘: check_flow, ‘8‘: add_cart, ‘9‘: check_shop, } def run(): while True: print("""" ############################################ 1. 注册 2. 登录 3. 提现 4. 转账 5. 充值 6. 查看余额 7. 查看流水 8. 加入购物车 9. 查看购物车 ############################################ """) choice = input(‘请输入你的选择:>>‘).strip() # if choice == 1: # register() # elif choice == 2: # login() if choice not in func_dic: continue func_dic[choice]()
common
# 验证登录的装饰器 from core import src def login_auth(func): # func => pay def inner(*args, **kwargs): # 验证是否登录 if src.user_data[‘is_login‘]: return func(*args, **kwargs) else: # 没有登录 src.login() return inner import logging import os from conf import settings # 一：日志配置 logging.basicConfig( # 1、日志输出位置：1、终端 2、文件 # filename=‘access.log‘, # 不指定，默认打印到终端 filename=os.path.join(settings.BASE_DIR, ‘log‘, ‘log.log‘), # 不指定，默认打印到终端 # 2、日志格式 format=‘%(asctime)s - %(name)s - %(levelname)s -%(module)s: %(message)s‘, # 3、时间格式 datefmt=‘%Y-%m-%d %H:%M:%S %p‘, # 4、日志级别 # critical => 50 # error => 40 # warning => 30 # info => 20 # debug => 10 level=10, ) # 二：输出日志 # logging.debug(‘调试debug‘) # logging.info(‘消息info‘) # logging.warning(‘警告warn‘) # logging.error(‘错误error‘) # logging.critical(‘严重critical‘) def write_log(info): logging.debug(info) ‘‘‘ # 注意下面的root是默认的日志名字 WARNING:root:警告warn ERROR:root:错误error CRITICAL:root:严重critical ‘‘‘
settings
import os BASE_DIR = os.path.dirname(os.path.dirname(__file__))