"""
@author: luojunhui
"""
import json
import requests

from applications.decorator import retryOnNone


class WeixinSpider(object):
    """
    Update account articles
    """
    # ip = "8.217.190.241"
    # ip = "47.98.154.124"
    # port = "8888"
    base_url = "http://crawler-cn.aiddit.com/crawler/wei_xin"
    headers = {
        "Content-Type": "application/json"
    }

    @classmethod
    @retryOnNone()
    def search_articles(cls, title) -> dict:
        """
        search articles in wx
        :return:
        """
        url = "{}/keyword".format(cls.base_url)
        payload = json.dumps({
            "keyword": title,
            "cursor": "1"
        })
        response = requests.request("POST", url, headers=cls.headers, data=payload, timeout=120)
        return response.json()

    @classmethod
    def get_article_text(cls, content_link, is_count=False) -> dict:
        """
        获取文章
        :param is_count:
        :param content_link:
        :return:
        """
        url = "{}/detail".format(cls.base_url)
        payload = json.dumps({
            "content_link": content_link,
            "is_count": is_count,
            "is_ad": False
        })
        response = requests.request("POST", url, headers=cls.headers, data=payload, timeout=120)
        return response.json()

    @classmethod
    # @retryOnNone()
    def update_msg_list(cls, ghId, index) -> dict:
        """
        :return:
        """
        url = '{}/blogger'.format(cls.base_url)
        payload = {
            'account_id': ghId,
            'cursor': index,
        }
        response = requests.post(url=url, headers=cls.headers, data=json.dumps(payload), timeout=120)
        return response.json()

    @classmethod
    @retryOnNone()
    def get_account_by_url(cls, content_url) -> dict:
        """
        通过文章获取账号信息
        :param content_url:
        :return:
        """
        url = '{}/account_info'.format(cls.base_url)
        data = {"content_link": content_url}
        response = requests.request("POST", url=url, headers=cls.headers, json=data, timeout=120)
        return response.json()

    @classmethod
    def get_recommend_articles(cls, content_link) -> dict:
        """
        use content link to get recommend articles
        :param content_link:
        :return:
        """
        url = "{}/recommend".format(cls.base_url)
        payload = json.dumps(
            {"content_link": content_link}
        )
        response = requests.request("POST", url=url, headers=cls.headers, data=payload, timeout=120)
        return response.json()