Python使用创建类获取Thread线程函数返回值只能获取一个结果

Z时代
2024-01-10
分类：技术分享

想要获取线程函数中Thread(target=fun, ...)中fun函数的返回值，根据网上的方法创建了一个类获得：

但是不解的是，get_result()方法只能获取到一个值，而run方法，print(self._result)是可以打印出所有的值的。
这是什么原因呢？请各位大佬指教。
下面附代码：
import nmap
import sendemailtool as tool
from optparse import OptionParser
from threading import Thread

class CustomTask(Thread):

def __init__(self, tgthost, tgtport):
    Thread.__init__(self)
    self._result = None
    self.tgthost = tgthost
    self.tgtport = tgtport
def run(self):
    result = nmScan(self.tgthost, self.tgtport)
    self._result = result
    print(self._result)
def get_result(self):
    # print(self._result)
    return self._result

def nmScan(tgthost, tgtport):

whitelist = []
result = ''
nmscan = nmap.PortScanner()
tmp = nmscan.scan(tgthost, tgtport)
result = result + "<h2>ip地址:%s ...... %s</h2><hr>" % (tgthost, tmp['scan'][tgthost]['status']['state'])
try:
    ports = tmp['scan'][tgthost]['tcp'].keys()
    for port in ports:
        if port not in whitelist:
            info = '<span><font color=red>Alert:非预期端口</font></span>&nbsp;&nbsp;'
        else:
            info = '<span><font color=green>Info:正常开放端口</font></span>&nbsp;&nbsp;'
        portinfo = "%s <span>port</span> : %s &nbsp;&nbsp;<span>state</span> : %s &nbsp;&nbsp;<span>product</span> : %s <br>" % (info, port, tmp['scan'][tgthost]['tcp'][int(port)]['state'], tmp['scan'][tgthost]['tcp'][int(port)]['product'])
        result = result + portinfo
except KeyError as e:
    if whitelist:
         whitestr = ','.join(whitelist)
         result = result + "未扫到开放端口！请检查%s端口对应的服务状态" % whitestr
    else:
        result = result + "扫描结果正常，无暴露端口"
return result

def main():

f1 = open('/home/angie/PycharmProjects/python_work/iplist.txt', 'ab')
parser = OptionParser()
parser.add_option('-H', dest='tgthost', type='string', help='specify target host')
parser.add_option('-f', dest='file', default='', help='host list', metavar='LIST')
(options, args) = parser.parse_args()
tgthost = options.tgthost
tgtports = [20, 21, 22, 23, 25, 69, 80, 109, 110, 139, 179, 443, 445, 544, 1080, 1433, 1434, 1521, 1158, 2100,3306, 3389, 7001, 8080, 8081, 9080, 9090]
if options.file == '':
    for tgtport in tgtports:
        ct = CustomTask(str(tgthost), str(tgtport))
        ct.start()
        ct.join()
    result = ct.get_result()
    # mailcontent = '<br>'.join(result)
    # print(result)
    tool.sendemail(result)
else:
    f = open(options.file, 'r')
    current_host = f.readline().strip()
    while (current_host):
        for tgtport in tgtports:
            ct = CustomTask(str(current_host), str(tgtport))
            ct.start()
            ct.join()
        current_host = f.readline().strip()
    # mailcontent = '<br>'.join(result)
    result = ct.get_result()
    f.close()
    # print(result)
    tool.sendemail(result)
f1.close()

if name == '__main__':

main()

回答：

问题已经解决了。
自己的理解是能够打印出所有的结果是因为每次循环都打印了一次，但是在每次运行的时候，self._result = result 只能接收一次的结果，所以最后存储到的只有最后一个的扫描结果。

解决的方法是：

创建list results，利用append方法每次存一个结果。
最后再利用类的get_result方法，再循环存储到list mailcontent中。
但还有一点奇怪的是不知道为什么用.join()方法无法存储所有的结果。。只好再用了append。
但用了append之后list不能解码为html，于是转化为str之后再传输到发送邮件的模块里发送。