将数组传递给Python中的类

Question

我是一名PHP开发者，对Python不是很了解……所以，如果我的问题很傻，请多包涵。

我有两个PHP脚本（client.php和worker.php），它们使用了Gearman，我需要把它们转换成Python版本。我部分完成了这个转换，但现在遇到了瓶颈。首先，给你们看看我的两个脚本：

client.py

#!/usr/bin/env python
import gearman
import json

RBLS = [
    'b.barracudacentral.org',
    'bl.emailbasura.org'
]

IP = '1.2.3.4'

def check_request_status(job_request):
    if job_request.complete:
        print "Job %s finished!  Result: %s - %s" % (job_request.job.unique, job_request.state, job_request.result)
    elif job_request.timed_out:
        print "Job %s timed out!" % job_request.unique
    elif job_request.state == JOB_UNKNOWN:
        print "Job %s connection failed!" % job_request.unique


data = {"ip": IP, "rbls": RBLS}
serialized_data = json.dumps(data)

gm_client = gearman.GearmanClient(['localhost:4730'])
completed_job_request = gm_client.submit_job("runcheck", serialized_data)
check_request_status(completed_job_request)

worker.py

#!/usr/bin/env python

import gearman
import sys
import socket
import re
import json
from dns.resolver import Resolver, NXDOMAIN, NoNameservers, Timeout, NoAnswer
from threading import Thread

# This hardcoded RBLS need to be passed by gearman client script
# RBLS = ['xbl.spamhaus.org', 'zen.spamhaus.org']

class Lookup(Thread):
    def __init__(self, host, dnslist, listed, resolver):
        Thread.__init__(self)
        self.host = host
        self.listed = listed
        self.dnslist = dnslist
        self.resolver = resolver

    def run(self):
        try:
            host_record = self.resolver.query(self.host, "A")
            if len(host_record) > 0:
                self.listed[self.dnslist]['LISTED'] = True
                self.listed[self.dnslist]['HOST'] = host_record[0].address
                text_record = self.resolver.query(self.host, "TXT")
                if len(text_record) > 0:
                    self.listed[self.dnslist]['TEXT'] = "\n".join(text_record[0].strings)
            self.listed[self.dnslist]['ERROR'] = False
        except NXDOMAIN:
            self.listed[self.dnslist]['ERROR'] = True
            self.listed[self.dnslist]['ERRORTYPE'] = NXDOMAIN
        except NoNameservers:
            self.listed[self.dnslist]['ERROR'] = True
            self.listed[self.dnslist]['ERRORTYPE'] = NoNameservers
        except Timeout:
            self.listed[self.dnslist]['ERROR'] = True
            self.listed[self.dnslist]['ERRORTYPE'] = Timeout
        except NameError:
            self.listed[self.dnslist]['ERROR'] = True
            self.listed[self.dnslist]['ERRORTYPE'] = NameError
        except NoAnswer:
            self.listed[self.dnslist]['ERROR'] = True
            self.listed[self.dnslist]['ERRORTYPE'] = NoAnswer

class RBLSearch(object):
    def __init__(self, lookup_host):
        self.lookup_host = lookup_host
        self._listed = None
        self.resolver = Resolver()
        self.resolver.timeout = 0.2
        self.resolver.lifetime = 1.0

    def search(self):
        if self._listed is not None:
            pass
        else:
            host = self.lookup_host.split(".")
            host = ".".join(list(reversed(host)))
            self._listed = {'SEARCH_HOST': self.lookup_host}
            threads = []
            for LIST in RBLS:
                self._listed[LIST] = {'LISTED': False}
                query = Lookup("%s.%s" % (host, LIST), LIST, self._listed, self.resolver)
                threads.append(query)
                query.start()
            for thread in threads:
                thread.join()
        return self._listed
    listed = property(search)

    def print_results(self):
        listed = self.listed
        print("")
        print("--- DNSBL Report for %s ---" % listed['SEARCH_HOST'])
        for key in listed:
            if key == 'SEARCH_HOST':
                continue
            if not listed[key].get('ERROR'):
                if listed[key]['LISTED']:
                    print("Results for %s: %s" % (key, listed[key]['LISTED']))
                    print("  + Host information: %s" % \
                          (listed[key]['HOST']))
                if 'TEXT' in listed[key].keys():
                    print("    + Additional information: %s" % \
                          (listed[key]['TEXT']))
            else:
                #print "*** Error contacting %s ***" % key
                pass


def task_listener_runcheck(gearman_worker, gearman_job):
    jdata = json.loads(gearman_job.data)
    host = jdata['ip']
    ip = host
    RBLS = jdata['rbls']

    print("Looking up: %s (please wait)" % host)
    pat = re.compile("\d{1,3}.\d{1,3}.\d{1,3}.\d{1,3}")
    is_ip_address = pat.match(host)
    if not is_ip_address:
        try:
            ip = socket.gethostbyname(host)
            print("Hostname %s resolved to ip %s" % (host,ip))
        except socket.error:
            print("Hostname %s can't be resolved" % host)
            ip = ""
    if ip:
        searcher = RBLSearch(ip)
        searcher.print_results()
    return "RunCheck was successfull"


gm_worker = gearman.GearmanWorker(['localhost:4730'])

gm_worker.set_client_id('python-worker')
gm_worker.register_task('runcheck', task_listener_runcheck)
gm_worker.work()

这两个脚本的工作原理是这样的：client.py把IP地址和一个RBL（黑名单）数组传给worker.py。然后，worker会获取这个IP地址，并检查它是否在所有的RBL中。

问题是，我不知道如何在RBLSearch类里面使用这些RBL。如果我在脚本开头硬编码（直接写死）RBL，它是可以工作的(见worker.py，第12行)，但如果我在task_listener_runcheck里面定义RBL，就不行了。

黑名单脚本转换数组传递类的使用 gearman PHP到Python RBL检查任务监听

将数组传递给Python中的类

1 个回答

撰写回答