LongArticlesJob-fork/applications/wxSpiderApi.py

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

from applications.decoratorApi import retryOnNone, retryOnTimeout


class WeixinSpider(object):
    """
    Update account articles
    """

    @classmethod
    @retryOnTimeout()
    @retryOnNone()
    def search_articles(cls, title):
        """
        search articles in wx
        :return:
        """
        url = "http://8.217.190.241:8888/crawler/wei_xin/keyword"
        payload = json.dumps({
            "keyword": title,
            "cursor": "1"
        })
        headers = {
            'Content-Type': 'application/json'
        }

        response = requests.request("POST", url, headers=headers, data=payload, timeout=10)
        return response.json()

    @classmethod
    @retryOnTimeout()
    @retryOnNone()
    def get_article_text(cls, content_link):
        """
        获取文章
        :param content_link:
        :return:
        """
        url = "http://8.217.190.241:8888/crawler/wei_xin/detail"
        payload = json.dumps({
            "content_link": content_link,
            "is_count": False,
            "is_ad": False
        })
        headers = {
            'Content-Type': 'application/json'
        }
        response = requests.request("POST", url, headers=headers, data=payload, timeout=10)
        return response.json()

    @classmethod
    @retryOnTimeout()
    @retryOnNone()
    def update_msg_list(cls, ghId, index):
        """
        :return:
        """
        url = 'http://8.217.190.241:8888/crawler/wei_xin/blogger'
        payload = {
            'account_id': ghId,
            'cursor': index,
        }
        headers = {
            'Content-Type': 'application/json'
        }
        response = requests.post(url, headers=headers, data=json.dumps(payload), timeout=10)
        return response.json()

    @classmethod
    @retryOnTimeout()
    @retryOnNone()
    def get_account_by_url(cls, content_url):
        """
        通过文章获取账号信息
        :param content_url:
        :return:
        """
        response = requests.request(
            "POST",
            url='http://8.217.190.241:8888/crawler/wei_xin/account_info',
            headers={'Content-Type': 'application/json'},
            json={"content_link": content_url},
            timeout=10
        )
        return response.json()