From d578e7471d9f1b67c42fd31e318157daadff2692 Mon Sep 17 00:00:00 2001 From: Conatum Date: Sun, 20 Aug 2023 12:29:56 +0300 Subject: [PATCH] feat: Add basic proxy support. --- src/bot.py | 1 + src/core/data.py | 5 ++++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/src/bot.py b/src/bot.py index 72765fdf..c66d1ce9 100644 --- a/src/bot.py +++ b/src/bot.py @@ -68,6 +68,7 @@ async def main(): shard_id=sharding.shard_number, shard_count=sharding.shard_count, help_command=None, + proxy=conf.bot.get('proxy', None), translator=translator ) as lionbot: ctx_bot.set(lionbot) diff --git a/src/core/data.py b/src/core/data.py index 6f409332..4de8f455 100644 --- a/src/core/data.py +++ b/src/core/data.py @@ -4,6 +4,7 @@ from psycopg import sql from cachetools import TTLCache import discord +from meta import conf from data import Table, Registry, Column, RowModel, RegisterEnum from data.models import WeakCache from data.columns import Integer, String, Bool, Timestamp @@ -367,4 +368,6 @@ class CoreData(Registry, name="core"): token = String() def as_webhook(self, **kwargs): - return discord.Webhook.partial(self.webhookid, self.token, **kwargs) + webhook = discord.Webhook.partial(self.webhookid, self.token, **kwargs) + webhook.proxy = conf.bot.get('proxy', None) + return webhook