DRMTalks
/
devine
mirror of https://github.com/devine-dl/devine.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

fix(Basic): Make query case-insensitive 

Fixes #88
This commit is contained in:
rlaphoenix 2024-04-01 17:30:08 +01:00
parent f25d2419cf
commit bd8309e1d7
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
devine/core/proxies/basic.py
View File

@ -1,13 +1,16 @@
import random import random
from typing import Optional from typing import Optional, Union
from devine.core.proxies.proxy import Proxy from devine.core.proxies.proxy import Proxy
class Basic(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.""" """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: def __repr__(self) -> str:
countries = len(self.countries) countries = len(self.countries)
@ -17,6 +20,8 @@ class Basic(Proxy):
def get_proxy(self, query: str) -> Optional[str]: def get_proxy(self, query: str) -> Optional[str]:
"""Get a proxy URI from the config.""" """Get a proxy URI from the config."""
query = query.lower()
servers = self.countries.get(query) servers = self.countries.get(query)
if not servers: if not servers:
return return