模型分离--让代码更方便管理
新建models.py,将模型定义全部放到这个独立的文件中。
新建exts.py,将db = SQLAlchemy()的定义放到这个独立的文件中。
models.py和主py文件,都从exts.py中导入db。
在主py文件中,对db进行始化,db.init_app(app)。
models.py:
from datetime import datetimefrom werkzeug.security import generate_password_hash, check_password_hashfrom exts import dbclass User(db.Model): __tablename__ = 'User' id = db.Column(db.Integer, primary_key=True, autoincrement=True) username = db.Column(db.String(20), nullable=False) _password = db.Column(db.String(200), nullable=False) # 内部使用 @property def password(self): return self._password @password.setter def password(self, row_password): self._password = generate_password_hash(row_password) def check_password(self, row_password): result = check_password_hash(self._password, row_password) return resultclass Question(db.Model): __tablename__ = 'question' 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) creatTime = db.Column(db.DateTime,default=datetime.now) authorID = db.Column(db.Integer,db.ForeignKey('User.id')) author = db.relationship('User',backref=db.backref('question'))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')) question_id = db.Column(db.Integer,db.ForeignKey('question.id')) detail = db.Column(db.Text,nullable=False) creatTime = db.Column(db.DateTime,default=datetime.now) question = db.relationship('Question',backref=db.backref('comments',order_by=creatTime.desc)) author = db.relationship('User',backref=db.backref('comments'))
exts.py:
from flask_sqlalchemy import SQLAlchemydb = SQLAlchemy()
主py:
from flask import Flask, render_template,request,redirect,url_for,sessionimport configfrom functools import wrapsfrom exts import dbfrom sqlalchemy import or_, and_from models import Question,User,Commentapp = Flask(__name__)app.config.from_object(config)db.init_app(app)