From a37c7373109746dd7a11194d53f9284245386dcb Mon Sep 17 00:00:00 2001 From: LWR Date: Wed, 26 Apr 2023 00:18:15 +0800 Subject: [PATCH] feat: Add follow user http api --- starbot/core/server.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/starbot/core/server.py b/starbot/core/server.py index a7c96af..52b6371 100644 --- a/starbot/core/server.py +++ b/starbot/core/server.py @@ -8,8 +8,10 @@ from loguru import logger from .datasource import DataSource from .model import Message, PushType +from .user import User, RelationType from ..exception import DataSourceException from ..utils import config +from ..utils.utils import get_credential routes = web.RouteTableDef() datasource: Optional[DataSource] = None @@ -65,6 +67,17 @@ async def send_by_bot(request: aiohttp.web.Request) -> aiohttp.web.Response: return await send(request, int(request.match_info['bot'])) +@routes.get("/user/follow/{uid}") +async def follow(request: aiohttp.web.Request) -> aiohttp.web.Response: + if not str(request.match_info['uid']).isdigit(): + logger.warning("关注用户失败, 传入的 UID 格式不正确") + return web.Response(text="fail") + + uid = int(request.match_info['uid']) + u = User(uid, get_credential()) + await u.modify_relation(RelationType.SUBSCRIBE) + + def get_routes() -> RouteTableDef: """ 获取路由，可用于外部扩展功能