|
|
import os |
|
|
import logging |
|
|
import random |
|
|
from dotenv import load_dotenv |
|
|
|
|
|
load_dotenv() |
|
|
|
|
|
IGNORED_MODEL_NAMES = ["gpt-4", "gpt-3.5", "websearch", "dall-e-3", "gpt-4o"] |
|
|
IMAGE_MODEL_NAMES = ["dalle3", "dalle-3", "dall-e-3"] |
|
|
AUTH_TOKEN = os.getenv("AUTHORIZATION") |
|
|
|
|
|
G_TOKEN = None |
|
|
HISTORY_MSG_LIMIT = os.getenv("HISTORY_MSG_LIMIT", 0) |
|
|
HTTP_PROXIES = os.getenv("HTTP_PROXIES") |
|
|
HTTPS_PROXIES = os.getenv("HTTPS_PROXIES") |
|
|
SOCKS_PROXIES = os.getenv("SOCKS_PROXIES") |
|
|
RECAPTCHA_SECRET = os.getenv("RECAPTCHA_SECRET") |
|
|
POPAI_BASE_URL = "https://www.popai.pro/" |
|
|
log_level = os.getenv('LOG_LEVEL', 'INFO').upper() |
|
|
|
|
|
|
|
|
def configure_logging(): |
|
|
extended_log_format = ( |
|
|
'%(asctime)s | %(levelname)s | %(name)s | ' |
|
|
'%(process)d | %(filename)s:%(lineno)d | %(funcName)s | %(message)s' |
|
|
) |
|
|
logging.basicConfig(level=log_level, format=extended_log_format) |
|
|
|
|
|
|
|
|
def _get_proxies_from_env(env_var): |
|
|
proxies = os.getenv(env_var, '') |
|
|
return [proxy.strip() for proxy in proxies.split(',') if proxy.strip()] |
|
|
|
|
|
|
|
|
class ProxyPool: |
|
|
def __init__(self): |
|
|
self.http_proxies = _get_proxies_from_env('HTTP_PROXIES') |
|
|
self.https_proxies = _get_proxies_from_env('HTTPS_PROXIES') |
|
|
self.socks_proxies = _get_proxies_from_env('SOCKS_PROXIES') |
|
|
|
|
|
def get_random_proxy(self): |
|
|
proxy = {} |
|
|
if self.http_proxies: |
|
|
proxy['http'] = random.choice(self.http_proxies) |
|
|
if self.https_proxies: |
|
|
proxy['https'] = random.choice(self.https_proxies) |
|
|
if self.socks_proxies: |
|
|
socks_proxy = random.choice(self.socks_proxies) |
|
|
proxy['http'] = socks_proxy |
|
|
proxy['https'] = socks_proxy |
|
|
|
|
|
logging.info("proxy URL %s", proxy) |
|
|
|
|
|
return proxy if proxy else None |
|
|
|
