File size: 1,779 Bytes
2a81db1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
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 = os.getenv("G_TOKEN")
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
|