1
0
mirror of https://github.com/internetarchive/warcprox.git synced 2025-01-18 13:22:09 +01:00

Add bad_hostnames_ports in PlaybackProxy

These vars are required also there in addition to
`SingleThreadedWarcProxy`.
This commit is contained in:
Vangelis Banos 2019-05-09 15:57:01 +00:00
parent 89d987a181
commit bbe41bc900

@ -42,6 +42,7 @@ from warcprox.mitmproxy import MitmProxyHandler
import warcprox import warcprox
import sqlite3 import sqlite3
import threading import threading
from cachetools import TTLCache
class PlaybackProxyHandler(MitmProxyHandler): class PlaybackProxyHandler(MitmProxyHandler):
logger = logging.getLogger("warcprox.playback.PlaybackProxyHandler") logger = logging.getLogger("warcprox.playback.PlaybackProxyHandler")
@ -219,6 +220,8 @@ class PlaybackProxy(socketserver.ThreadingMixIn, http_server.HTTPServer):
self.playback_index_db = playback_index_db self.playback_index_db = playback_index_db
self.warcs_dir = options.directory self.warcs_dir = options.directory
self.options = options self.options = options
self.bad_hostnames_ports = TTLCache(maxsize=1024, ttl=60)
self.bad_hostnames_ports_lock = threading.RLock()
def server_activate(self): def server_activate(self):
http_server.HTTPServer.server_activate(self) http_server.HTTPServer.server_activate(self)