近期在阅读Flask WEB 开发(基于Python的Web应用开发实践) ,从读者的github源码上下载下来并进行实践。
在实践过程中发现其中一个部分的代码bug,导致部分功能实现不了(并不是什么大问题,但是有些读者不加测试的话就会引用原code导致不必要的bug):
Bug重现:
关注用户部分,作者定义了数据库模型类Follow如下:
里边定义关注者id和被关注者id,以及关注时间。
class Follow(db.Model):
__tablename__ = 'follows'
follower_id = db.Column(db.Integer, db.ForeignKey('users.id'),
primary_key=True)
followed_id = db.Column(db.Integer, db.ForeignKey('users.id'),
primary_key=True)
timestamp = db.Column(db.DateTime, default=datetime.utcnow)
在数据库User模型中follow函数引用了该Follow模型进行实例化:
def follow(self, user):
if not self.is_following(user):
f = Follow(follower=self, followed=user)
db.session.add(f)
发现问题没有?
在Follow实例化过程中并没有应用定义时定义的具体的列,导致传入的参数类型问题f = Follow(follower=self, followed=user)。
应当传入当前用户的id和关注用户的id。
改进:
def follow(self, user):
if not self.is_following(user):
f = Follow(follower_id=self.id, followed_id=user.id)
db.session.add(f)
之后程序在测试过程中就不会出现函数参数类型报错了。