1.peewee模型类建立——自关联的多对多关系
由于关注和粉丝之间的关系都是来源于用户表，并且是多对多的关系
代码展示
from peewee import *
from forum import database
from datetime import datetime
class BaseModel(Model):
create_time = DateTimeField(default = datetime.now)
def to_json(self) -> dict:
r = {}
for k in self.__data__.keys():
if k == 'create_time':
r[k] = str(getattr(self,k))
else:
r[k] = getattr(self,k)
return r
class Meta:
database = database
class UserModel(BaseModel):
id = CharField(primary_key = True)
email = CharField(max_length=32, verbose_name='账号')
nick_name = CharField(max_length=32, verbose_name='昵称',null=True)
password = CharField(max_length=128, verbose_name='密码')
gender = CharField(verbose_name='性别',null=True)
signatrue = CharField(max_length=32, verbose_name='签名',null=True)
pic = CharField(max_length=512, verbose_name='头像',null=True)
status = CharField(verbose_name='账号状态', default=1)
class Meta:
table_name = 't_user'
class FollowModel(BaseModel):
to_user = ForeignKeyField(UserModel, backref='to_user')
from_user = ForeignKeyField(UserModel, backref='from_user')
class Meta:
table_name = 't_follow'
primary_key = CompositeKey('to_user','from_user')
创建代码
FollowModel.create_table(True)
2.用户关注——tornado增加接口
2.1查看前端需要的接口和传递的数据
; 2.2代码展示
接口
from forum.handler.BaseHandler import BaseHandler
from forum import manager
from forum.models import UserModel, FollowModel
from forum.decorators import login_required_async
class AddFollowHandler(BaseHandler):
@login_required_async
async def post(self):
uid = self.get_body_argument('id')
print("uid:",uid,"----------",type(uid))
to_user = await manager.get(UserModel, id= uid)
print("to_user:",to_user,"----------",type(to_user))
from_user = await manager.get(UserModel, id = self._user_id)
await manager.create(FollowModel, to_user=to_user, from_user=from_user)
self.finish({'code':200,'msg':'关注成功！！'})
2.3 获取关注状态概述
当我们点击完关注，应该更换成一个取消关注的图标。此时，查看前端代码，前端是通过flag这个变量进行控制。
; 2.4查看前端需要的接口和传递的数据——tornado查询接口
我们发现，在前端，还有一个接口是专门控制查看是否关注。
2.5代码展示
from forum.handler.BaseHandler import BaseHandler
from forum import manager
from forum.models import UserModel, FollowModel
from forum.decorators import login_required_async
class GetFollowHandler(BaseHandler):
@login_required_async
async def post(self):
try:
uid = self.get_body_argument('id')
fid = self._user_id
follow = await manager.execute(FollowModel.select().where(FollowModel.to_user==uid,FollowModel.from_user==fid))
if follow:
self.finish({'code':200,'msg':'已关注','flag':True})
else:
self.finish({'code':200,'msg':'未关注','flag':False})
except Exception as e:
print(e)
self.finish({'code':200,'msg':'未关注','flag':False})
3.取消关注——tornado删除接口
3.1查看前端需要的接口和传递的数据
; 3.2代码展示
接口
class DeleteFollowHandler(BaseHandler):
@login_required_async
async def delete(self):
try:
data = loads(self.request.body)
uid = data.get('id')
to_user = await manager.get(UserModel, id= uid)
from_user = await manager.get(UserModel, self._user_id)
rs = await manager.execute(FollowModel.delete().where(FollowModel.to_user==uid, FollowModel.from_user==self._user_id))
self.finish({'code':200,'msg':'取消关注成功'})
except Exception as e:
self.finish({'code':500,'msg':'取消关注失败'})
4.我的关注列表——tornado查询接口
4.1查看前端需要的接口和传递的数据
; 4.2代码展示
class GetFollowHandler(BaseHandler):
@login_required_async
async def get(self):
follows = await manager.execute(FollowModel.select(FollowModel.create_time,UserModel.id,UserModel.pic,UserModel.nick_name).join(UserModel,on=FollowModel.to_user).where(FollowModel.from_user==self._user_id))
data = [{'create_time':str(f[0]),'id':f[1], 'pic':f[2], 'nick_name':f[3] } for f in follows._rows ]
self.finish({'code':200,'msg':'获取关注列表成功！！','friends':data})
5.显示关注与粉丝数——tornado查询接口
5.1查看前端需要的接口和传递的数据
; 5.2代码展示
class FollowNumHandler(BaseHandler):
@login_required_async
async def post(self):
type_ = self.get_body_argument('type_')
if type_ == 'to':
to_user = await manager.execute(FollowModel.select().where(FollowModel.from_user == self._user_id))
self.finish({'code':200,'msg':'获取关注数成功！！！','num':len(to_user._rows)})
elif type_ == 'from':
from_user = await manager.execute(FollowModel.select().where(FollowModel.to_user == self._user_id))
self.finish({'code':200,'msg':'获取粉丝数成功！！！','num':len(from_user._rows)})
Original: https://blog.csdn.net/m0_63953077/article/details/127820751
Author: 想成为数据分析师的开发工程师
Title: 15.关注模块——peewee创建模型、tornado-peewee-async查询增加删除接口
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/658679/
转载文章受原作者版权保护。转载请注明原作者出处！