From bd8309e1d7cefe44645952e71f37d72852954475 Mon Sep 17 00:00:00 2001
From: rlaphoenix <rlaphoenix@pm.me>
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