"""
@author: luojunhui
@description: 投流文章自动回复发布
"""
from applications import aiditApi


class PublishTouliuAutoReplyArticles(object):
    """
    @author: luojunhui
    @description: 投流文章自动回复发布
    @version: 0.0.1
    @doc:
        step1: 自动创建抓取计划
        step2: 绑定抓取计划到无限流发布计划（需要维护cookie）
        step3: 等待发布，可以通过发布计划id查到发布内容id，并且可以通过发布内容id查询发布状态
        step4: 通过爬虫抓回文章链接，并且获取其root_source_id， 通过链接替换自动回复系统中的链接
    """
    URL_DICT = {
        '小惠爱厨房': 'https://mp.weixin.qq.com/s?__biz=MzkwNTYxNDQ3Ng==&mid=2247490456&idx=1&sn=e500cbc6829a1e61cd8546a27594b2f0&chksm=c0f45e7bf783d76d0266a380c183609c345494e20a12240778c5fed088ae0c3300897c5a6cda#rd',
        '小阳看天下': 'https://mp.weixin.qq.com/s?__biz=MzkwNjYxMjM1Mg==&mid=2247493253&idx=1&sn=fc8b86d72c0be412817a9849aae1c441&chksm=c0e77d6af790f47cc8f868c10e0641b755017aa3d0b823bfb5d6ff537ac155b02cd74d70661a#rd',
        '银发生活畅谈': 'https://mp.weixin.qq.com/s?__biz=Mzk0Mjg2NTY3Ng==&mid=2247483897&idx=1&sn=3922ba195db859c985e8b59977d493fb&chksm=c33ded9af44a648c28d2f41ce540df1da61a45d7b96081ab1539e4cdac26b3ade7477ff27d79#rd',
        '生活慢时光': 'https://mp.weixin.qq.com/s?__biz=MzkzNTUwNzYzMQ==&mid=2247492803&idx=1&sn=8985c7da427318466b18ad73d9714db4&chksm=c2ae4703f5d9ce1599333dc00f58554c15062b1ebf674619258913e7ada570f06e7a310c83f1#rd',
        '历史长河流淌': 'https://mp.weixin.qq.com/s?__biz=MzkzMjczNDAwNQ==&mid=2247486220&idx=1&sn=6b2e35b3e38d72bf623c901f7f5016d7&chksm=c2567e56f521f740365843a4b87bd5d12c4aeb9b5e85500879168e8b1db76a14136ef03c7a22#rd',
        '暖心一隅': 'https://mp.weixin.qq.com/s?__biz=Mzk0OTcyNjYxNg==&mid=2247487310&idx=1&sn=0e0ac28189280f186dea412caed8aa2c&chksm=c352b3b9f4253aaf73836e46f13f75491fd45fe5fadce273be1a854eb211c9820b180f206d60#rd',
        '史趣探秘': 'https://mp.weixin.qq.com/s?__biz=MzkzNDcyMzU5NA==&mid=2247486858&idx=1&sn=d7dff664a6a5d9d073bafbc194547b4d&chksm=c2b9a4eff5ce2df94cd20ba1df03a7d25a7f0da20acf5171b538b12aabecf0767bdda17740cc#rd'
    }
    PUBLISH_ID_DICT = {
        "小阳看天下": "20241119084409473585377",
        "小惠爱厨房": "20241119073710385694858",
        "银发生活畅谈": "20241119125508677670625",
        "生活慢时光": "20241119125344679385053",
        "历史长河流淌": "20241119125244408356973",
        "暖心一隅": "20241119125002792704222",
        "史趣探秘": "20241119124554214209549"
    }

    def create_crawler_plan(self, account_name):
        """
        创建抓取计划
        :return:
        crawler_plan_id = crawler_plan_response['data']['id']
        crawler_plan_name = crawler_plan_response['data']['name']
        """
        article_url = self.URL_DICT.get(account_name)
        if article_url:
            crawler_plan_response = aiditApi.auto_create_crawler_task(
                plan_id=None,
                plan_name="{}--自动回复文章".format(account_name),
                plan_tag="投流自动回复文章",
                url_list=[article_url]
            )
            return crawler_plan_response
        else:
            return None

    def do_job(self, account_name):
        """

        :param account_name:
        :return:
        """
        crawler_plan_response = self.create_crawler_plan(account_name)
        if crawler_plan_response:
            crawler_plan_id = crawler_plan_response['data']['id']
            crawler_plan_name = crawler_plan_response['data']['name']
            publish_plan_response = aiditApi.bind_crawler_task_to_publish_task(
                target_publish_task_id=self.PUBLISH_ID_DICT.get(account_name),
                crawler_task_name=crawler_plan_name,
                crawler_task_id=crawler_plan_id
            )
            return publish_plan_response


p = PublishTouliuAutoReplyArticles()
for key in p.URL_DICT.keys():
    # print(key)
    p.create_crawler_plan(account_name=key)