模型分离--让代码更方便管理
新建models.py,将模型定义全部放到这个独立的文件中。
from werkzeug.security import generate_password_hash,check_password_hash
from datetime import datetime
from exts import db
class 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)#内部使用
nickname = db.Column(db.String(50))
# email=db.Column(db.String(100))
@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 result
class 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)
create_time=db.Column(db.DateTime,default=datetime.now)
author_id =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'))
create_time = db.Column(db.DateTime, default=datetime.now)
detail = db.Column(db.Text, nullable=False)
question=db.relationship('Question',backref=db.backref('comments',order_by=create_time.desc))
author=db.relationship('User',backref=db.backref('comments'))
#db.create_all()
新建exts.py,将db = SQLAlchemy()的定义放到这个独立的文件中。
from flask_sqlalchemy import SQLAlchemy
db=SQLAlchemy()
models.py和主py文件,都从exts.py中导入db。
from exts import db
from exts import db
在主py文件中,对db进行始化,db.init_app(app)。
db.init_app(app)