From 5ded527f73d4b285fbda20b67677fcd041c64fc7 Mon Sep 17 00:00:00 2001 From: monosans Date: Sat, 23 Mar 2024 09:47:19 +0300 Subject: [PATCH] Fix possible KeyError --- proxy_scraper_checker/scraper.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/proxy_scraper_checker/scraper.py b/proxy_scraper_checker/scraper.py index d7219d1..2544380 100644 --- a/proxy_scraper_checker/scraper.py +++ b/proxy_scraper_checker/scraper.py @@ -60,9 +60,7 @@ async def scrape_one( for proxy in itertools.chain((proxy,), proxies): # noqa: B020 try: protocol = ProxyType[ - "HTTP" - if (p := proxy.group("protocol").upper()) == "HTTPS" - else p + proxy.group("protocol").upper().strip("S") ] except AttributeError: protocol = proto