proxy_pool/Schedule/ProxyValidSchedule.py

# -*- coding: utf-8 -*-
"""
-------------------------------------------------
   File Name：     ProxyValidSchedule.py
   Description :  验证useful_proxy_queue中的代理,将不可用的移出
   Author :       JHao
   date：          2017/3/31
-------------------------------------------------
   Change Activity:
                   2017/3/31: 验证useful_proxy_queue中的代理
-------------------------------------------------
"""
__author__ = 'JHao'

import sys
from time import sleep

sys.path.append('../')

from Util.utilFunction import validUsefulProxy
from Manager.ProxyManager import ProxyManager
from Util.LogHandler import LogHandler


class ProxyValidSchedule(ProxyManager):
    def __init__(self):
        ProxyManager.__init__(self)
        self.log = LogHandler('valid_schedule')

    def __validProxy(self):
        """
        验证代理
        :return:
        """
        while True:
            self.db.changeTable(self.useful_proxy_queue)
            for each_proxy in self.db.getAll():
                if isinstance(each_proxy, bytes):
                    # 兼容PY3
                    each_proxy = each_proxy.decode('utf-8')

                value = self.db.getvalue(each_proxy)
                if validUsefulProxy(each_proxy):
                    # 成功计数器加1
                    if value and int(value) < 1:
                        self.db.inckey(each_proxy, 1)
                    self.log.info('ProxyValidSchedule: {} validation pass'.format(each_proxy))
                else:
                    # 失败计数器减一
                    if value and int(value) < -5:
                        # 计数器小于-5删除该代理
                        self.db.delete(each_proxy)
                    else:
                        self.db.inckey(each_proxy, -1)
                    self.log.info('ProxyValidSchedule: {} validation fail'.format(each_proxy))

            self.log.info('ProxyValidSchedule running normal')
            sleep(60 * 1)

    def main(self):
        self.__validProxy()


def run():
    p = ProxyValidSchedule()
    p.main()


if __name__ == '__main__':
    p = ProxyValidSchedule()
    p.main()