nex/college-bot-v2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Activity Actions

Properly allow configuring screenshot proxy

Browse source
This commit is contained in:
Nexus 2024-03-17 16:15:59 +00:00
parent 273404a615
commit 8ef909c33d
Signed by: nex
GPG key ID: 0FA334385D0B689F
2 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/cogs/screenshot.py
View file

@ -16,6 +16,8 @@ from selenium import webdriver
from selenium.webdriver.chrome.options import Options as ChromeOptions
from selenium.webdriver.chrome.service import Service as ChromeService
from conf import CONFIG
class ScreenshotCog(commands.Cog):
def __init__(self, bot: commands.Bot):
@ -107,8 +109,8 @@ class ScreenshotCog(commands.Cog):
start_init = time.time()
try:
options = copy.copy(self.chrome_options)
if use_proxy:
options.add_argument("--proxy-server=http://127.0.0.1:8888")
if use_proxy and (server := CONFIG["screenshot"].get("proxy")):
options.add_argument("--proxy-server=" + server)
service = await asyncio.to_thread(ChromeService)
driver: webdriver.Chrome = await asyncio.to_thread(
webdriver.Chrome,

1
src/conf.py
View file

@ -28,6 +28,7 @@ try:
CONFIG.setdefault("jimmy", {})
CONFIG.setdefault("ollama", {})
CONFIG.setdefault("rss", {"meta": {"channel": None}})
CONFIG.setdefault("screenshot", {})
CONFIG.setdefault(
"server",
{