Fix possible KeyError

2024-03-23 09:47:19 +03:00
parent 998cf21518
commit 5ded527f73
1 changed files with 1 additions and 3 deletions
@@ -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