LongArticleTaskServer/app/api/service/gzh_cookie_manager.py

from app.domains.crawler_tasks.crawler_gzh_fans import CrawlerGzhFansBase


class GzhCookieManager(CrawlerGzhFansBase):
    def __init__(self, pool, log_client):
        super().__init__(pool, log_client)

    async def deal(self, data):
        gh_id = data.get('gzh_id')
        if not gh_id:
            return {"error": "gh_id is required"}

        if not gh_id.startswith("gh_"):
            return {"error": "gh_id is invalid"}

        token = data.get("token")
        if not token:
            return {"error": "token is required"}

        cookie = data.get("cookie")
        if not cookie:
            return {"error": "cookie is required"}

        insert_row = await self.set_cookie_token_for_each_account(
            gh_id=gh_id, cookie=cookie, token=token
        )
        if not insert_row:
            return {"error": "insert row failed"}

        return {"success": "cookie and token set successfully"}