'''
项目启动入口
'''
import os
import sys
# 将项目的根目录，添加到sys.path中
sys.path.append(
os.path.dirname(os.path.dirname(__file__))
)
from core import src
if __name__ == '__main__':
src.run()
'''
此处存放核心业务逻辑代码
'''
from db import db_hanlder
from lib import common
import time
# login_user: 若能进入该函数，证明用户已登录，能获取当前用户名
login_user = None
# 注册功能
def register():
print('注册功能执行中...')
while True:
username = input('请输入用户名 (输入q退出): ').strip()
# if username == 'q':
# break
# 1) 先校验用户是否存在
# 涉及数据的操作: 调用查看数据功能 ---> select
# 给select函数传入当前输入的用户名，用于判断该用户是否存在
# user_data ---> [user, pwd, bal] or None
user_data = db_hanlder.select(username)
# 2)若存在，让用户重新输入
if user_data:
print('当前输入的用户已存在，请重新输入!')
continue
password = input('请输入密码: ').strip()
re_password = input('请确认密码: ').strip()
# 3)两次密码是否一致的校验
if password == re_password:
# 4) 将当前用户数据写入文件中
db_hanlder.save(username, password)
print(f'用户[{username}]注册成功!')
break
else:
print('两次密码不一致，请重新输入!')
# 登录功能
def login():
print('登录功能执行中...')
while True:
username = input('请输入用户名 (输入q退出): ').strip()
# 1) 查看当前用户是否存在 [user, pwd, balance] or None
user_data = db_hanlder.select(username)
# 2)若不存在，让用户重新输入
if not user_data:
print('当前输入的用户不存在，请重新输入!')
continue
password = input('请输入密码: ').strip()
# 3)校验用户输入的密码是否与db.txt中的密码一致
if password == user_data[1]:
# 4)用户登录后记录登录状态
global login_user
login_user = username
print(f'用户[{username}]登录成功!')
break
else:
print('密码错误，登录失败!')
# 充值功能
@common.login_auth
def recharge():
print('充值功能执行中...')
while True:
# 1) 让用户输入充值金额
balance = input('请输入充值金额: ').strip()
# 2）判断用户输入的金额是否是数字
if not balance.isdigit():
print('请输入数字!')
continue
balance = int(balance) # 100
# 3) 修改当前用户的金额
# 3.1) 获取当前用户的数据
user, pwd, bal = db_hanlder.select(login_user) # [user, pwd, bal]
# 3.2) 先获取用户 "修改前" 的数据
old_data = f'{user}:{pwd}:{bal}'
# 3.3) 修改当前用户金额，做加钱操作
bal = int(bal)
bal += balance
# 3.4) 拼接 "修改后" 的当前用户数据
# 此时bal的值已经被修改
# 'tank:123:0' ---> 'tank:123:100'
new_data = f'{user}:{pwd}:{bal}'
# 3.5）调用修改数据功能
# 'tank:123:0' ---> 'tank:123:100'
db_hanlder.update(old_data, new_data)
print(f'当前用户: [{login_user}] 充值金额 [{balance}元] 成功！')
# ending: 做充值日志记录
now_time = time.strftime('%Y-%m-%d %X')
log_data = f'时间: {now_time} 用户名: {login_user} 充值金额:{balance}'
print(log_data)
common.append_log(log_data)
break
# 小说阅读功能
@common.login_auth
def reader():
'''
1、写该功能之前，先将小说数据，存放在story_class.txt文件中
2、先将story_class.txt文件中数据，获取出来，解析成字典类型
:return:
'''
# 获取story_class.txt文件中的字典数据
# story_dic ---> {小说数据字典} ---> ''
story_dic = db_hanlder.get_all_story()
# 判断story_class.txt文件中是否有小说数据
if not story_dic:
print('没有小说，请联系管理员上传~~')
return
'''story_dic
{
"0":{
"0":["倚天屠狗记.txt",3],
"1":["沙雕英雄转.txt",10]
},
"1":{
"0":["令人羞耻的爱.txt",6],
"1":["二狗的妻子与大草原的故事.txt",5]
},
"2":{
"0":["矮跟落叶归根养猪护理传.txt",100],
"1":["矮跟在猪圈雨后的小故事.txt",500]
},
}
'''
while True:
# 1) 打印小说种类选择信息
print('''
=== 欢迎来到阅读小说功能主页 ===
0 玄幻武侠
1 都市爱情
2 高效养猪36技
============ end ============
''')
# 2）让用户输入小说类型编号
choice1 = input('请输入小说类型编号：').strip()
# 3）判断当前用户选择的编号是否存在
# 若输入有误则，重新输入
if choice1 not in story_dic:
print('输入有误，请重新输入!!!')
continue
# 4)获取当前小说类型中的所有小说数据
fiction_dic = story_dic.get(choice1) # choice1 == '1'
'''fiction_dic
{
"0":["令人羞耻的爱.txt",6],
"1":["二狗的妻子与大草原的故事.txt",5]
},
'''
# 5）打印当前类型的所有小说
for number, fiction_list in fiction_dic.items():
name, price = fiction_list
print(f'小说编号: [{number}] 小说名字: [{name}] 小说价格: [{price}]')
# 6）让用户选择选择需要购买的小说
while True:
choice2 = input('请输入需要购买的小说编号: ').strip()
if choice2 not in fiction_dic:
print('输入有误，请重新输入!!!')
continue
# name, price = ["二狗的妻子与大草原的故事.txt",5]
# name --> 小说名字 price ---> 小说单价
name, price = fiction_dic.get(choice2)
# 7）让用户输入y选择是否购买商品
choice3 = input(f'当前选择的小说名为: [{name}]，商品单价为: [{price}]， 请输入y购买，或者退出程序：').strip()
# 8）判断用户输入的是否是y
if choice3 == 'y':
# 9）校验当前用户的金额是否 >= 小说单价
# 9.1）获取当前用户金额
user, pwd, bal = db_hanlder.select(login_user)
# 9.2）判断金额
# 当前用户金额
bal = int(bal)
# 小说单价
price = int(price)
if bal < price:
print('穷鬼，请先充值~~~')
break
# 10 开始扣费
# 10.1）拼接用户修改前的数据
old_data = f'{user}:{pwd}:{bal}'
# 10.2) 开始对用户金额做减钱操作
bal -= price
# 10.）拼接用户修改后的数据
new_data = f'{user}:{pwd}:{bal}'
db_hanlder.update(old_data, new_data)
print('当前小说购买成功，自动打开小说阅读~~')
# 11) 调用获取想说详细信息数据
fiction_data = db_hanlder.show_fiction_data(name)
print(
f'''
==== 当前小说数据如下 ====
{fiction_data}
'''
)
# 12） 记录购买成功日志
now_time = time.strftime('%Y-%m-%d %X')
log_data = f'时间: {now_time} 用户名: {login_user} 消费金额:{price}'
print(log_data)
common.append_log(log_data)
break
break
# 函数字典
func_dic = {
'0': register,
'1': login,
'2': recharge,
'3': reader,
}
# 启动函数
def run():
'''
# 一：程序运行开始时显示
0 账号注册
1 账号登录
2 充值功能
3 阅读小说
'''
while True:
print('''
===== 小说阅读器欢迎您 =====
0 账号注册
1 账号登录
2 充值功能
3 阅读小说
========== end ==========
''')
# choice --> '1'
choice = input('请输入功能编号 (温馨提示[输入q退出]): ').strip()
if choice == 'q':
break
# 判断用户输入的编号是否在函数字典中
if choice not in func_dic:
print('当前编号有误，请重新输入!')
continue
# func_dic.get('1')() --> register函数对象 ---> register()
func_dic.get(choice)()
'''
此处存放公共功能
'''
from conf import settings
# 登录认证装饰器
def login_auth(func):
# 解决问题: 循环导入问题
from core import src
def inner(*args, **kwargs):
if src.login_user:
res = func(*args, **kwargs)
return res
else:
print('未登录，不允许使用特殊功能, 请先登录~~~')
src.login()
return inner
# 记录日志，应该放在公共功能中
def append_log(log_data):
# 写入日志数据
with open(settings.LOG_PATH, 'a', encoding='utf-8') as f:
f.write(log_data + '
')
'''
用于存放操作数据代码
'''
from conf import settings
import os
# 查看数据
def select(username):
'''
- 接收用户输入的用户名
- 若该用户存在，则返回当前用户的所有数据
- 若不存在，则返回None
:param username:
:return:
'''
with open(settings.DB_TXT_PATH, 'r', encoding='utf-8') as f:
# 获取db.txt文件中的每一行数据
for line in f:
# 在每一行中，判断接收过来的用户名是否存在db.txt文件中
if username in line:
# 若用户存在，则在当前行中提取该用户的所有数据
# user_data ---> [user, pwd, bal]
user_data = line.strip().split(':')
# 将当前用户数据，返回给调用者
return user_data
# 保存数据
def save(username, password, balance=0):
'''
:param username: 注册用户名
:param password: 注册密码
:param balance: 注册用户初始金额设置为默认值
:return:
'''
with open(settings.DB_TXT_PATH, 'a', encoding='utf-8') as f:
f.write(f'{username}:{password}:{balance}
')
# 更新数据
def update(old_data, new_data):
'''
将旧数据old_data:'tank:123:0' 替换成new_data:---> 新数据'tank:123:100'
:return:
'''
# 1) 拼接新的文件路径
new_path = os.path.join(
settings.DB_PATH, 'new.txt'
)
# 2) 读取db.txt文件中数据进行修改，写入到新文件new.txt文件中，再更换为db.txt文件名
with open(settings.DB_TXT_PATH, 'r', encoding='utf-8') as r_f,
open(new_path, 'w', encoding='utf-8') as w_f:
# 2.1）新旧数据替换
# python终极奥秘: 尽量让代码更简洁
all_user_data = r_f.read()
all_user_data = all_user_data.replace(old_data, new_data)
# 2.2）将新数据写入新文件中
w_f.write(all_user_data)
# 3) 文件名的修改
# os.remove() 可以不写
# 这一步db.txt会将原来的覆盖掉
os.rename(new_path, settings.DB_TXT_PATH)
# 获取小说字典数据
def get_all_story():
with open(settings.STORY_PATH, 'r', encoding='utf-8') as f:
story_dic = eval(f.read())
return story_dic
# 查看单本小说
def show_fiction_data(fiction_name):
# 获取小说路径
fiction_path = os.path.join(
settings.FICTION_DIR, fiction_name
)
# 打开文件，获取文件数据，并返回给用户展示
with open(fiction_path, 'r', encoding='utf-8') as f:
fiction_data = f.read()
return fiction_data
'''
此处存放固定的配置信息
'''
import os
# 获取项目根目录reader_sys根目录
BASE_PATH = os.path.dirname(os.path.dirname(__file__))
# 获取db目录路径
# /python相关/python_files/01 python班级/reader_sys
# /python相关/python_files/01 python班级/reader_sys/db
DB_PATH = os.path.join(BASE_PATH, 'db')
# 获取db.txt的根目录
DB_TXT_PATH = os.path.join(DB_PATH, 'db.txt')
# story_class.txt文件目录路径
STORY_PATH = os.path.join(DB_PATH, 'story_class.txt')
# 小说存放目录
FICTION_DIR = os.path.join(DB_PATH, 'fictions')
# 日志文件的路径
LOG_PATH = os.path.join(BASE_PATH, 'log', 'log.txt')