模型分离（选做）

模型分离--让代码更方便管理

新建models.py，将模型定义全部放到这个独立的文件中。

新建exts.py，将db = SQLAlchemy()的定义放到这个独立的文件中。

models.py和主py文件，都从exts.py中导入db。

在主py文件中，对db进行始化，db.init_app(app)。

 

主py:

from flask import Flask, render_template, request, redirect, url_for, session
import config
from functools import wraps
from sqlalchemy import or_, and_
from models import User,Fabu,Comment
from exts import db


app = Flask(__name__)  # 创建Flask对象
app.config.from_object(config)  # 关联config.py文件进来
db.init_app(app)   # 建立和数据库的关系映射

...

models.py:

from datetime import datetime
from werkzeug.security import generate_password_hash,check_password_hash
from exts import db


class User(db.Model):  # 创建类User
    __tablename__ = 'user'  # 类对应的表名user
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)  # autoincrement自增长
    username = db.Column(db.String(20), nullable=False)  # nullable是否为空
    _password = db.Column(db.String(200), nullable=False)   # 密码加密内部使用
    nickname = db.Column(db.String(20), nullable=True)

    @property   # 定义函数，需要用属性时可以用函数代替
    def password(self):   # 密码加密外部使用
        return self._password

    @password.setter
    def password(self,row_password):   # 密码进来时进行加密,generate_password_hash是一个密码加盐哈希函数，生成的哈希值可通过check_password_hash()进行验证。
        self._password = generate_password_hash(row_password)

    def check_password(self,row_password):   # check_password_hash函数用于验证经过generate_password_hash哈希的密码。若密码匹配，则返回真，否则返回假。
        result = check_password_hash(self._password,row_password)
        return result


class Fabu(db.Model):
    __tablename__ = 'fabu'
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    title = db.Column(db.String(100), nullable=False)
    detail = db.Column(db.Text, nullable=False)
    creat_time = db.Column(db.DateTime, default=datetime.now)  # 提交时间会自己赋值
    author_id = db.Column(db.Integer, db.ForeignKey('user.id'))  # 数据类型是db.Integer，db.ForeignKey参数指定外键是哪个表中哪个id
    author = db.relationship('User', backref=db.backref('fabu'))  # 建立关联，其author属性将返回与问答相关联的用户实例，相当于数据库中的表连接
    # 第一个参数表明这个关系的另一端是哪个类，第二个参数backref，将向User类中添加一个fabu属性，从而定义反向关系，这一属性可访问Fabu类，获取的是模型对象


class Comment(db.Model):
    __tablename__ = 'comment'
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    author_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    fabu_id = db.Column(db.Integer, db.ForeignKey('fabu.id'))
    creat_time = db.Column(db.DateTime, default=datetime.now)
    detail = db.Column(db.Text, nullable=False)
    fabu = db.relationship('Fabu',
                           backref=db.backref('comments', order_by=creat_time.desc))  # order_by=creat_time.desc按时间降序
    author = db.relationship('User', backref=db.backref('comments'))


# db.create_all()  # 测试是否连接成功

exts.py:

from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()  # 建立和数据库的关系映射