yawaflua/Flask-Discord
1
0
Fork 0
mirror of https://github.com/yawaflua/Flask-Discord.git synced 2026-02-04 10:14:15 +02:00
Code Issues Packages Projects Releases Wiki Activity

✔ Add proxy support

Browse Source
This commit is contained in:
thec0sm0s
2020-09-13 23:58:08 +05:30
parent 5bd875cae9
commit bf53bf825f
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
flask_discord/_http.py
View File

@@ -28,13 +28,15 @@ class DiscordOAuth2HttpClient(abc.ABC):
def __init__( def __init__(
self, app=None, self, app=None,
client_id=None, client_secret=None, redirect_uri=None, client_id=None, client_secret=None, redirect_uri=None,
bot_token=None, users_cache=None bot_token=None, users_cache=None, proxy=None, proxy_auth=None
): ):
self.client_id = client_id self.client_id = client_id
self.__client_secret = client_secret self.__client_secret = client_secret
self.redirect_uri = redirect_uri self.redirect_uri = redirect_uri
self.__bot_token = bot_token self.__bot_token = bot_token
self.users_cache = users_cache self.users_cache = users_cache
self.proxy = proxy
self.proxy_auth = proxy_auth
if app is not None: if app is not None:
self.init_app(app) self.init_app(app)
@@ -58,6 +60,8 @@ class DiscordOAuth2HttpClient(abc.ABC):
) if self.users_cache is None else self.users_cache ) if self.users_cache is None else self.users_cache
if not issubclass(self.users_cache.__class__, Mapping): if not issubclass(self.users_cache.__class__, Mapping):
raise ValueError("Instance users_cache must be a mapping like object.") raise ValueError("Instance users_cache must be a mapping like object.")
self.proxy = self.proxy or app.config.get("DISCORD_PROXY_SETTINGS")
self.proxy_auth = self.proxy_auth or app.config.get("DISCORD_PROXY_AUTH_SETTINGS")
app.discord = self app.discord = self
@property @property
@@ -157,6 +161,12 @@ class DiscordOAuth2HttpClient(abc.ABC):
""" """
route = configs.DISCORD_API_BASE_URL + route route = configs.DISCORD_API_BASE_URL + route
if self.proxy is not None:
kwargs["proxy"] = self.proxy
if self.proxy_auth is not None:
kwargs["proxy_auth"] = self.proxy_auth
response = self._make_session( response = self._make_session(
).request(method, route, data, **kwargs) if oauth else requests.request(method, route, data=data, **kwargs) ).request(method, route, data, **kwargs) if oauth else requests.request(method, route, data=data, **kwargs)