Pārlūkot izejas kodu

[update] show proxy count

jhao 5 gadi atpakaļ
vecāks
revīzija
3ccabd2530
1 mainītis faili ar 5 papildinājumiem un 4 dzēšanām
  1. 5 4
      test/testProxyFetcher.py

+ 5 - 4
test/testProxyFetcher.py

@@ -19,13 +19,14 @@ from handler.configHandler import ConfigHandler
 def testProxyFetcher():
     conf = ConfigHandler()
     proxy_getter_functions = conf.fetchers
+    proxy_counter = {_: 0 for _ in proxy_getter_functions}
     for proxyGetter in proxy_getter_functions:
-        proxy_count = 0
         for proxy in getattr(ProxyFetcher, proxyGetter.strip())():
             if proxy:
-                print('{func}: fetch proxy {proxy},proxy_count:{proxy_count}'.format(func=proxyGetter, proxy=proxy,
-                                                                                     proxy_count=proxy_count))
-                proxy_count += 1
+                print('{func}: fetch proxy {proxy}'.format(func=proxyGetter, proxy=proxy))
+                proxy_counter[proxyGetter] = proxy_counter.get(proxyGetter) + 1
+    for key, value in proxy_counter.items():
+        print(key, value)
 
 
 if __name__ == '__main__':