mirror of
https://github.com/devine-dl/devine.git
synced 2025-04-30 01:59:44 +00:00
31 lines
929 B
Python
31 lines
929 B
Python
import random
|
|
from typing import Optional
|
|
|
|
from devine.core.proxies.proxy import Proxy
|
|
|
|
|
|
class Basic(Proxy):
|
|
def __init__(self, **countries):
|
|
"""Basic Proxy Service using Proxies specified in the config."""
|
|
self.countries = countries
|
|
|
|
def __repr__(self) -> str:
|
|
countries = len(self.countries)
|
|
servers = len(self.countries.values())
|
|
|
|
return f"{countries} Countr{['ies', 'y'][countries == 1]} ({servers} Server{['s', ''][servers == 1]})"
|
|
|
|
def get_proxy(self, query: str) -> Optional[str]:
|
|
"""Get a proxy URI from the config."""
|
|
servers = self.countries.get(query)
|
|
if not servers:
|
|
return
|
|
|
|
proxy = random.choice(servers)
|
|
|
|
if "://" not in proxy:
|
|
# TODO: Improve the test for a valid URI
|
|
raise ValueError(f"The proxy '{proxy}' is not a valid proxy URI supported by Python-Requests.")
|
|
|
|
return proxy
|