python写小说阅读功能~

一、代码格式规范readme.md

-- 软件开发目录规范
    - bin
        - start  启动软件入口
    - conf
        - settings 配置文件
    - core
        - src  核心业务逻辑代码
    - db
        - 用于存放数据文件与操作数据的代码文件
        - db.file ----> db.txt
        - db_handler  操作数据的代码
    -lib 
        - common 存放公共的功能
    - log
        - log.txt  存放日志的文件

二、正式代码：

'''
start.py
项目启动入口
'''
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()

'''
setting.py
配置文件路径
'''
import os

# 获取根目录
BASE_PATH = os.path.dirname(os.path.dirname(__file__))
# 获取db目录
DB_PATH = os.path.join(BASE_PATH,'db')
# 获取db目录下文件
DB_TXT_PATH = os.path.join(DB_PATH,"db.txt")

# 获取db目录下小说的的路径
STORY_PATH = os.path.join(DB_PATH,'story_class.txt')

# 日志文件的路径
LOG_PATH = os.path.join(BASE_PATH,"log","log.txt")

'''
src.py
主逻辑代码
'''

from db import db_hanlder
from lib import common
import time

login_user = None

# 注册功能
def register():
    print("注册！")
    while 1:
        user_name = input("请输入你要注册的账户:").strip()
        # 判断用户名是否存在
        user_data_or_None = db_hanlder.select(user_name)

        if user_data_or_None:
            print("账户名存在，请重新输入！")
            continue
        password = input("请输入注册账户的密码:").strip()
        re_password = input("请再次输入注册账户的密码:").strip()

        if password == re_password:
            print("用户【%s】注册成功！"%user_name)
            db_hanlder.save(user_name,password)
            break
        else:
            print("两次密码不一致，请从新输入")

# 登录功能
def login():
    print("登录!")
    while 1:
        user_name = input("请输入要登录的用户名：").strip()
        user_data= db_hanlder.select(user_name)
        if not user_data:
            print("用户不存在！")
            continue
        password = input("请输入用户名密码：").strip()

        if  password == user_data[1]:
            global login_user
            login_user = user_name
            print("【%s】登录成功！"%user_name)
            break
        else:
            print("用户名或密码错误！")

# 充值功能
@common.login_auth
def recharge():
    print("充值！")
    while 1:
        balance = input("请输入要充值的金额：").strip()
        if not balance.isdigit():
            print("请输入数字！")
            continue
        balance = int(balance)
        user,pwd,bal =db_hanlder.select(login_user)
        old_data = '%s:%s:%s'%(user,pwd,bal)
        bal = int(bal)
        bal += balance
        new_data = '%s:%s:%s'%(user,pwd,bal)
        db_hanlder.update(old_data,new_data)
        print("当前用户【%s】充值金额【%s】元！"%(login_user,balance))

        now_time = time.strftime("%Y-%m-%d %X")
        login_data = "%s 用户名%s 充值金额:%s ...."%(now_time,login_user,balance)
        common.append_log(login_data)
        break

# 阅读功能
@common.login_auth
def reader():
    print("阅读")
    story_dic = db_hanlder.get_all_story()

    if  not  story_dic:
        print("没有小说，联系管理员上传！")
        return

    while 1:
        # 打印小说目录
        print('''
        ====== 小说 ======
            0、玄幻小说
            1、都市爱情
            2、考试学习
        ''')
        choice = input("请输入要阅读的小说类别：").strip()

        if choice not in story_dic:
            print("输入有误，重新输入！")
            continue
        fiction_dic = story_dic.get(choice)

        for number,fiction_list in fiction_dic.items():
            name,price = fiction_list
            print("小说编号【%s】小说名字【%s】小说价格【%s】"%(number,name,price))
        while 1:
            choice2 = input("请输入要购买的小说:").strip()
            if choice2 not in fiction_dic:
                print("输入有误，请重新输入")
                continue
            name,price = fiction_dic.get(choice2)
            choice3 = input("是否购买当前小说【%s】价格是【%s】,请输入（Y/N）:"%(name,price)).strip().upper()
            if choice3 == "Y":
                # 读取用户数据
                user,pwd,bal = db_hanlder.select(login_user)
                bal = int(bal)
                price = int(price)
                if bal < price:
                    print("余额不足，请充值！！！")
                    break
                old_data = '%s:%s:%s'%(user,pwd,bal)
                bal -=price
                new_data = '%s:%s:%s'%(user,pwd,bal)
                db_hanlder.update(old_data,new_data)
                print("小说购买成功！")

                now_time = time.strftime("%Y-%m-%d %X")
                login_data = "%s 用户名%s 购买小说:%s 消费金额:%s...." % (now_time, login_user, name,price)
                common.append_log(login_data)

                res = db_hanlder.show_fiction_data(name)
                print(res)

                break

            break


# 函数字典
func_dic = {
    "0":register,
    "1":login,
    "2":recharge,
    "3":reader
}


def run():
    '''

    :return: 程序运行开始显示：
            0、账号注册
            1、登录
            2、充值
            3、阅读小说
            4、退出
    '''
    while 1:
        print("""
        ======== 小说阅读器 ========
                0、账号注册
                1、登录
                2、充值
                3、阅读小说
                4、退出
        """)

        choice =  input("请输入你要做的操作：").strip()
        if choice == "4":break
        if choice not in func_dic:
            print("你输入的有误，请重新输入！")
            continue
        func_dic[choice]()

#!/usr/bin/python
# -*- coding: UTF-8 -*-
'''
db_hanlder.py
数据操作
'''

from conf import settings
import os

# 查询用户是否存在
def select(name):
    with open(settings.DB_TXT_PATH,"r",encoding="utf8") as r_f:
        for line in r_f:
            if name in line:
                use_data = line.strip().split(":")
                return use_data

# 注册写入功能
def save(name,pwd,bal=0):
    with open(settings.DB_TXT_PATH,"a",encoding="utf8") as w_f:
        w_f.write("%s:%s:%s
"%(name,pwd,bal))

# 数据更新
def update(old_data,new_data):

    new_path = os.path.join(settings.DB_PATH,"new.txt")
    with open(settings.DB_TXT_PATH,"r",encoding="utf-8") as r_f,
        open(new_path,"w",encoding="utf-8") as w_f:

        all_user_data = r_f.read()
        all_user_data = all_user_data.replace(old_data,new_data)
        w_f.write(all_user_data)
    os.remove(settings.DB_TXT_PATH)
    os.rename(new_path, settings.DB_TXT_PATH)


# 读取小说文件中的字典
def get_all_story():
    with open(settings.STORY_PATH,"r",encoding="utf8") as f:
        story_dic = eval(f.read())
        return story_dic

def show_fiction_data(name):
    story_data_path = os.path.join(settings.DB_PATH,"fictions",name)
    with open(story_data_path,"r",encoding="utf8") as f:
        res = f.read()
        return res

'''
common.py
公共模块
'''
from conf import settings
from core import src
import time,os

def login_auth(func):
    def wrapper(*args,**kwargs):
        if src.login_user:
            res = func(*args,**kwargs)
            return res
        else:
            print("未登录，暂不支持此功能")
            src.login()
    return wrapper

def append_log(log_data):
    with open(settings.LOG_PATH,"a",encoding="utf8") as f:
        f.write(log_data+'
')