Croc/croccybot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Add basic proxy support.

Browse Source
This commit is contained in:
Conatum
2023-08-20 12:29:56 +03:00
parent 267eb97fe4
commit d578e7471d
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/bot.py
View File

@@ -68,6 +68,7 @@ async def main():
shard_id=sharding.shard_number, shard_id=sharding.shard_number,
shard_count=sharding.shard_count, shard_count=sharding.shard_count,
help_command=None, help_command=None,
proxy=conf.bot.get('proxy', None),
translator=translator translator=translator
) as lionbot: ) as lionbot:
ctx_bot.set(lionbot) ctx_bot.set(lionbot)

5
src/core/data.py
View File

@@ -4,6 +4,7 @@ from psycopg import sql
from cachetools import TTLCache from cachetools import TTLCache
import discord import discord
from meta import conf
from data import Table, Registry, Column, RowModel, RegisterEnum from data import Table, Registry, Column, RowModel, RegisterEnum
from data.models import WeakCache from data.models import WeakCache
from data.columns import Integer, String, Bool, Timestamp from data.columns import Integer, String, Bool, Timestamp
@@ -367,4 +368,6 @@ class CoreData(Registry, name="core"):
token = String() token = String()
def as_webhook(self, **kwargs): 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