From bd8309e1d7cefe44645952e71f37d72852954475 Mon Sep 17 00:00:00 2001 From: rlaphoenix Date: Mon, 1 Apr 2024 17:30:08 +0100 Subject: [PATCH] fix(Basic): Make query case-insensitive Fixes #88 --- devine/core/proxies/basic.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/devine/core/proxies/basic.py b/devine/core/proxies/basic.py index fb2a7ce..56dde86 100644 --- a/devine/core/proxies/basic.py +++ b/devine/core/proxies/basic.py @@ -1,13 +1,16 @@ import random -from typing import Optional +from typing import Optional, Union from devine.core.proxies.proxy import Proxy class Basic(Proxy): - def __init__(self, **countries): + def __init__(self, **countries: dict[str, Union[str, list[str]]]): """Basic Proxy Service using Proxies specified in the config.""" - self.countries = countries + self.countries = { + k.lower(): v + for k, v in countries.items() + } def __repr__(self) -> str: countries = len(self.countries) @@ -17,6 +20,8 @@ class Basic(Proxy): def get_proxy(self, query: str) -> Optional[str]: """Get a proxy URI from the config.""" + query = query.lower() + servers = self.countries.get(query) if not servers: return