小说阅读器
启动模块bin/start.py 1 import os
2 import sys
3
4 BASE_PATH=os.path.dirname(os.path.dirname(__file__))
5 sys.path.append(BASE_PATH)
6
7
8 from core import src
9
10 if __name__== \'__main__\':
11 src.run()
核心代码模块core/src.py
1 from db import db_handler as db
2 from lib import common
3
4
5 login_user = None
6
7
8
9
10 def register():
11 \'\'\'
12 注册功能
13 :return:
14 \'\'\'
15 while True:
16 inp_name = input(\'输入用户名\').strip()
17 name, psd, remind = db.getinfo(inp_name)
18 if inp_name in name:
19 print(\'用户已存在\')
20 password = input(\'输入密码\')
21 re_psd = input(\'再次输入密码\')
22 if password == re_psd:
23 db.save_reg(inp_name, password)
24 print(\'注册成功\')
25 break
26 else:
27 print(\'两次密码不一致，请重新输入\')
28
29
30
31 def login():
32 \'\'\'
33 登录功能
34 :return:
35 \'\'\'
36 while True:
37 inp_name = input(\'输入用户名\')
38 name, psd, remind = db.getinfo(inp_name)
39 if inp_name not in name:
40 print(\'用户名不存在，请先注册\')
41 continue
42 password = input(\'输入密码\').strip()
43 if password == psd:
44 global login_user
45 login_user = inp_name
46 print(\'登录成功,亲爱的{}用户\'.format(login_user))
47 break
48 else:
49 print(\'账号密码错误\')
50
51
52
53 @common.log
54 @common.auth
55 def add_credit():
56 \'\'\'
57 充值功能
58 :return:
59 \'\'\'
60 print(\'欢迎使用充值功能，亲爱的{}用户\'.format(login_user))
61 while True:
62 amount = input(\'输入充值金额\')
63 if not amount.isdigit():
64 print(\'请输入数字\')
65 continue
66 comf = input(\'确定要充值{}元吗,按Y/y确认,按R/r\'.format(amount))
67 if comf == \'R\' or comf == \'r\':
68 continue
69 if comf == \'Y\' or comf == \'y\':
70 name, psd, remind = db.getinfo(login_user)
71 remind=int(remind)
72 old_data = \'{}:{}:{}\'.format(name, psd, remind)
73 remind += int(amount)
74 new_data = \'{}:{}:{}\'.format(name, psd, remind)
75 db.update(old_data, new_data)
76 print(\'已成功充值{}\'.format(amount))
77 break
78 return amount
79
80
81 @common.log
82 @common.auth
83 def reader():
84 \'\'\'
85 小说功能
86 :return:
87 \'\'\'
88 book_dic=db.get_book()
89 if not book_dic:
90 print(\'库存没有小说了\')
91 return
92 while True:
93 print(\'\'\'
94 === 欢迎来到阅读小说功能主页 ===
95 0 玄幻武侠
96 1 都市爱情
97 2 高效养猪36技
98 ============ end ============
99 \'\'\')
100 cmd=input(\'输入你想阅读的小说种类编号\').strip()
101 if cmd not in book_dic:
102 print(\'请正确输入编号\')
103 second_dic =book_dic[cmd]
104 for num,booklist in second_dic.item():
105 name,price=booklist
106 print(\'小说编号[{}],小说名《{}》,小说价格[{}]\'.format(num,name,price))
107 while True:
108 cmd = input(\'输入你想购买的小说编号\')
109 if cmd not in second_dic:
110 print(\'请正确输入编号\')
111 continue
112 if cmd in second_dic:
113 name=second_dic[cmd][0]
114 price=second_dic[cmd][1]
115 cmd = input(\'当前选择的小说名为: 《{name}》，商品单价为: [{price}]， 请输入Y/y购买：\'.format(name,price)).strip()
116 if cmd == \'Y\' or cmd==\'y\':
117 name,psd,remind = db.getinfo(login_user)
118 remind=int(remind)
119 if remind<int(price):
120 price(\'余额不足，请先充值\')
121 break
122 old_data = \'{}:{}:{}\'.format(name, psd, remind)
123 remind -= int(price)
124 new_data = \'{}:{}:{}\'.format(name, psd, remind)
125 db.update(old_data, new_data)
126 print(\'购买成功,请欣赏内容\')
127 fictiondata=db.show(name)
128 print(\'\'\'
129 -------开始阅读--------
130 {}
131 \'\'\'.format(fictiondata))
132
133
134
135 def logout():
136 \'\'\'
137 登出功能
138 :return:
139 \'\'\'
140 global login_user
141 login_user=None
142
143
144 #运行菜单
145 func_dic={
146 \'0\':[\'注册\',register],
147 \'1\':[\'登录\',login],
148 \'2\':[\'充值\',add_credit],
149 \'3\':[\'小说\',reader],
150 \'4\':[\'登出\',logout],
151 }
152
153 def run():
154 while True:
155 print(\'\'\'
156 ~~~~~~~欢迎来到小说世界~~~~~~~
157 0 账号注册
158 1 账号登录
159 2 充值功能
160 3 阅读小说
161 4 登出
162 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~
163 \'\'\')
164 cmd= input(\'请输入功能编号，按4退出\').strip()
165 if cmd ==\'4\':
166 logout()
167 break
168 if cmd not in func_dic:
169 print(\'请正确输入功能编号\')
170 continue
171 func_dic[cmd][1]()
172
173 run()
数据库操作模块db/db_handler.py
1 from conf import settings
2 import os
3
4
5 def getinfo(username):
6 \'\'\'
7 取出文件中用户的数据
8 :param username:
9 :return:
10 \'\'\'
11 with open(settings.DB_TXT_PATH, \'rt\', encoding=\'utf-8\')as f:
12 for line in f:
13 if username in line:
14 name, psd, remind = line.strip().split(\':\')
15 return name, psd, remind
16
17
18 def save_reg(name, psd, balance=0):
19 \'\'\'
20 存注册用户的数据
21 :param name:
22 :param psd:
23 :param balance:
24 :return:
25 \'\'\'
26 with open(settings.DB_TXT_PATH, \'at\', encoding=\'utf-8\')as f:
27 f.write(\'{}:{}:{}\'.format(name, psd, balance))
28
29
30
31 def update(old_data, new_data):
32 \'\'\'
33 更新用户数据
34 :param old_data:
35 :param new_data:
36 :return:
37 \'\'\'
38 with open(settings.DB_TXT_PATH, \'rt\', encoding=\'utf-8\')as rf:
39 alldata = rf.read()
40 alldata = alldata.replace(old_data, new_data)
41 with open(\'db.txt.swap\', \'wt\', encoding=\'utf-8\')as wf:
42 wf.write(alldata)
43 os.rename(\'db.txt.swap\', settings.DB_TXT_PATH)
44
45
46 def get_book():
47 \'\'\'
48 获取小说数据
49 :return:
50 \'\'\'
51 with open(settings.STORY_PATH,\'rt\',encoding=\'utf-8\')as f:
52 book_dic=eval(f.read())
53 return book_dic
54
55
56 def show(bookname):
57 \'\'\'
58 查看小说
59 :param bookname:
60 :return:
61 \'\'\'
62 book_path = os.path.join(settings.BOOK_DIR, bookname)
63 with open(book_path,\'rt\',encoding=\'utf-8\')as f:
64 data=f.read()
65 return data
公共模块（装饰器）lib/common.py
1 from conf import settings
2 import time
3 from core.src import login_user
4 from core.src import login
5
6 def auth(func):
7 \'\'\'
8 认证装饰器
9 :param func:
10 :return:
11 \'\'\'
12 def wrapper(*args, **kwargs):
13 if login_user:
14 res = func(*args, **kwargs)
15 return res
16 else:
17 print(\'请先登录再使用功能\')
18 login()
19
20 return wrapper
21
22
23
24 def log(func):
25 \'\'\'
26 记录日志装饰器
27 :param func:
28 :return:
29 \'\'\'
30 def wrapper(*args, **kwargs):
31 res = func(*args, **kwargs)
32 now_time = time.strftime(\'%Y-%m-%d %X\')
33 log_data = \'{} {} {} {}\'.format(now_time, login_user, func.__name__, res)
34 with open(\'time.log\', \'ab\')as logf:
35 logf.write(log_data.encode(\'utf-8\'))
36
37 return res
38
39 return wrapper
配置模块conf/settings.py
import os
# 获取项目根目录reader_sys根目录
BASE_PATH = os.path.dirname(os.path.dirname(__file__))
# 获取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\')
# 小说存放目录
BOOK_DIR = os.path.join(DB_PATH, \'book\')
# 日志文件的路径
LOG_PATH = os.path.join(BASE_PATH, \'log\', \'user.log\')
面条形式的一个文件下的小说阅读程序实现
1.更新了小说阅读模块
2.重写了用户存取数据的方式：将字典模式改为函数功能，方便之后规范化放入数据库操作模块中
3.独立了登录模块，将认证功能装饰器简化
import os
import time
login_user = None
# common模块
# 认证装饰器
def auth(func):
def wrapper(*args, **kwargs):
if login_user:
res = func(*args, **kwargs)
return res
else:
print(\'请先登录再使用功能\')
login()
return wrapper
# 记录日志装饰器
def log(func):
def wrapper(*args, **kwargs):
res = func(*args, **kwargs)
now_time = time.strftime(\'%Y-%m-%d %X\')
log_data = \'{} {} {} {}\'.format(now_time, login_user, func.__name__, res)
with open(\'time.log\', \'ab\')as logf:
logf.write(log_data.encode(\'utf-8\'))
return res
return wrapper
# 数据库模块
# 取数据
def getinfo(username):
with open(\'db.txt\', \'rt\', encoding=\'utf-8\')as f:
for line in f:
if username in line:
name, psd, remind = line.strip().split(\':\')
return name, psd, remind
# 存注册数据
def save_reg(name, psd, balance=0):
with open(\'db.txt\', \'at\', encoding=\'utf-8\')as f:
f.write(\'{}:{}:{}\'.format(name, psd, balance))
# 更新数据
def update(old_data, new_data):
with open(\'db.txt\', \'rt\', encoding=\'utf-8\')as rf:
alldata = rf.read()
alldata = alldata.replace(old_data, new_data)
with open(\'db.txt.swap\', \'wt\', encoding=\'utf-8\')as wf:
wf.write(alldata)
os.rename(\'db.txt.swap\', \'db.txt\')
# 注册功能
def register():
while True:
inp_name = input(\'输入用户名\').strip()
name, psd, remind = getinfo(inp_name)
if inp_name in name:
print(\'用户已存在\')
password = input(\'输入密码\')
re_psd = input(\'再次输入密码\')
if password == re_psd:
save_reg(inp_name, password)
print(\'注册成功\')
break
else:
print(\'两次密码不一致，请重新输入\')
# 登录功能
def login():
while True:
inp_name = input(\'输入用户名\')
name, psd, remind = getinfo(inp_name)
if inp_name not in name:
print(\'用户名不存在，请先注册\')
continue
password = input(\'输入密码\').strip()
if password == psd:
global login_user
login_user = inp_name
print(\'登录成功,亲爱的{}用户\'.format(login_user))
break
else:
print(\'账号密码错误\')
# 充值功能
@log
@auth
def add_credit():
print(\'欢迎使用充值功能，亲爱的{}用户\'.format(login_user))
while True:
amount = input(\'输入充值金额\')
if not amount.isdigit():
print(\'请输入数字\')
continue
comf = input(\'确定要充值{}元吗,按Y/y确认,按R/r\'.format(amount))
if comf == \'R\' or comf == \'r\':
continue
if comf == \'Y\' or comf == \'y\':
name, psd, remind = getinfo(login_user)
remind=int(remind)
old_data = \'{}:{}:{}\'.format(name, psd, remind)
remind += int(amount)
new_data = \'{}:{}:{}\'.format(name, psd, remind)
update(old_data, new_data)
print(\'已成功充值{}\'.format(amount))
break
return amount
#获取小说数据
def get_book():
with open(\'story_class.txt\',\'rt\',encoding=\'utf-8\')as f:
book_dic=eval(f.read())
return story_dic
#查看小说
def show(bookname):
with open(\'\',\'rt\',encoding=\'utf-8\')as f:
data=f.read()
return data
# 小说功能
@log
@auth
def reader():
book_dic=get_book()
if not book_dic:
print(\'库存没有小说了\')
return
while True:
print(\'\'\'
=== 欢迎来到阅读小说功能主页 ===
0 玄幻武侠
1 都市爱情
2 高效养猪36技
============ end ============
\'\'\')
cmd=input(\'输入你想阅读的小说种类编号\').strip()
if cmd not in book_dic:
print(\'请正确输入编号\')
second_dic =book_dic[cmd]
for num,booklist in second_dic.item():
name,price=booklist
print(\'小说编号[{}],小说名《{}》,小说价格[{}]\'.format(num,name,price))
while True:
cmd = input(\'输入你想购买的小说编号\')
if cmd not in second_dic:
print(\'请正确输入编号\')
continue
if cmd in second_dic:
name=second_dic[cmd][0]
price=second_dic[cmd][1]
cmd = input(\'当前选择的小说名为: 《{name}》，商品单价为: [{price}]， 请输入Y/y购买：\'.format(name,price)).strip()
if cmd == \'Y\' or cmd==\'y\':
name,psd,remind = getinfo(login_user)
remind=int(remind)
if remind<int(price):
price(\'余额不足，请先充值\')
break
old_data = \'{}:{}:{}\'.format(name, psd, remind)
remind -= int(price)
new_data = \'{}:{}:{}\'.format(name, psd, remind)
update(old_data, new_data)
print(\'购买成功,请欣赏内容\')
fictiondata=show(name)
print(\'\'\'
-------开始阅读--------
{}
\'\'\'.format(fictiondata))
#登出功能
def logout():
global login_user
login_user=None
#运行菜单
func_dic={
\'0\':[\'注册\',register],
\'1\':[\'登录\',login],
\'2\':[\'充值\',add_credit],
\'3\':[\'小说\',reader],
\'4\':[\'登出\',logout],
}
def run():
while True:
print(\'\'\'
~~~~~~~欢迎来到小说世界~~~~~~~
0 账号注册
1 账号登录
2 充值功能
3 阅读小说
4 登出
~~~~~~~~~~~~~~~~~~~~~~~~~~~~
\'\'\')
cmd= input(\'请输入功能编号，按4退出\').strip()
if cmd ==\'4\':
logout()
break
if cmd not in func_dic:
print(\'请正确输入功能编号\')
continue
func_dic[cmd][1]()
run()
# 编写小说阅读程序实现下属功能
# # 一：程序运行开始时显示
# 0 账号注册
# 1 充值功能
# 2 阅读小说
#
#
# # 二： 针对文件db.txt，内容格式为："用户名:密码:金额",完成下述功能
# 2.1、账号注册
# 2.2、充值功能
#
# # 三：文件story_class.txt存放类别与小说文件路径，如下,读出来后可用eval反解出字典
# {"0":{"0":["倚天屠狗记.txt",3],"1":["沙雕英雄转.txt",10]},"1":{"0":["令人羞耻的爱.txt",6],"1":["二狗的妻子与大草原的故事.txt",5]},}
#
# 3.1、用户登录成功后显示如下内容，根据用户选择，显示对应品类的小说编号、小说名字、以及小说的价格
# """
# 0 玄幻武侠
# 1 都市爱情
# 2 高效养猪36技
# """
#
# 3.2、用户输入具体的小说编号，提示是否付费，用户输入y确定后，扣费并显示小说内容，如果余额不足则提示余额不足
#
# # 四：为功能2.2、3.1、3.2编写认证功能装饰器，要求必须登录后才能执行操作
#
# # 五：为功能2.2、3.2编写记录日志的装饰器，日志格式为："时间 用户名 操作(充值or消费) 金额"
#
#
#
# # 附加：
# # 可以拓展作者模块，作者可以上传自己的作品
#
#
import time
userstatus = {\'username\': None}
def auth(func):
\'\'\'
认证功能装饰器
:return:
\'\'\'
def wrapper(*args, **kwargs):
if userstatus[\'username\'] is None:
inp_name = input(\'输入用户名\')
inp_psd = input(\'输入密码\')
with open(\'db.txt\', \'rt\', encoding=\'utf-8\')as rf:
userinfo = {}
for line in rf:
name, psd, remind = line.strip().split(\':\')
userinfo[name] = [psd, int(remind)]
if inp_name not in userinfo:
print(\'用户不存在\')
return
if inp_name in userinfo and userinfo[inp_name][0]==inp_psd:
print(\'登录成功，亲爱的{}用户\'.format(inp_name))
userstatus[\'username\']=inp_name
res=func(*args, **kwargs)
return res
else:
print(\'账号密码错误\')
return
else:
res = func(*args, **kwargs)
return res
return wrapper
def record_log(func):
\'\'\'
日志记录功能装饰器
:return:
\'\'\'
def wrapper(*args, **kwargs):
res = func(*args, **kwargs)
with open(\'record.log\', \'ab\')as log_f:
op_time = time.time()
amount = func.amount
log_f.write(\'{} {} {} {}\n\'.format(op_time, userstatus[\'username\'], func.__name__, amount).encode(\'utf-8\'))
return res
return wrapper
def register():
\'\'\'
注册功能
:return:
\'\'\'
with open(\'db.txt\', \'rt\', encoding=\'utf-8\')as rf:
userinfo = {}
for line in rf:
name, psd, remind = line.strip().split(\':\')
userinfo[name] = [psd, int(remind)]
inp_name = input(\'输入注册名\')
inp_psd = input(\'输入密码\')
if inp_name in userinfo:
print(\'用户名已存在\')
return
re_psd = input(\'再次输入密码\')
if inp_psd == re_psd:
print(\'注册成功\')
with open(\'db.txt\', \'at\', encoding=\'utf-8\')as af:
af.write(\'{}:{}:0\n\'.format(inp_name, inp_psd))
else:
print(\'两次密码不一致\')
return
@record_log
@auth
def add_credit():
\'\'\'
充值功能
:return:
\'\'\'
print(\'欢迎使用充值功能，亲爱的{}\'.format(userstatus[\'username\']))
with open(\'db.txt\', \'rt\', encoding=\'utf-8\')as rf:
userinfo = {}
for line in rf:
name, psd, remind = line.strip().split(\':\')
userinfo[name] = [psd, int(remind)]
while True:
amount = input(\'请输入充值的金额\').strip()
conf = input(\'确定要向{}用户充值{}吗，确定输入Y/y，退出N/n\'.format(userstatus[\'username\'], amount))
if conf == \'Y\' or conf == \'y\':
print(\'充值成功，已为您充值{}元\'.format(amount))
userinfo[userstatus[\'username\']][1] += int(amount)
break
elif conf == \'N\' or conf == \'n\':
break
with open(\'db.txt\', \'wt\', encoding=\'utf-8\')as wf:
for name in userinfo:
wf.write(\'{}:{}:{}\n\'.format(userinfo[name], userinfo[name][0], userinfo[name][1]))
@record_log
@auth
def read():
\'\'\'
阅读小说功能
:return:
\'\'\'
pass
def logout():
\'\'\'
登出
:return:
\'\'\'
userstatus[\'username\'] = None
return
def menu():
fuc_dic = {
\'0\': (\'注册功能\', register),
\'1\': (\'充值功能\', add_credit),
\'2\': (\'阅读小说\', read),
\'3\': (\'登出\',logout)
}
tag= True
while tag:
cmd = input(\'\'\'
0 账号注册
1 充值功能
2 阅读小说
3 登出
\'\'\')
int(cmd)
if not cmd.isdigit():
print(\'请输入数字\')
if cmd.isdigit():
fuc_dic[cmd][1]()
if cmd == 3:
tag=False
menu()