From 318e2318b8f554041ab1fdaf4b72df43e0715a0a Mon Sep 17 00:00:00 2001 From: thec0sm0s Date: Fri, 15 May 2020 18:08:27 +0530 Subject: [PATCH] Add import to Integration --- flask_discord/_http.py | 4 ++++ flask_discord/models/__init__.py | 1 + 2 files changed, 5 insertions(+) diff --git a/flask_discord/_http.py b/flask_discord/_http.py index 697f40c..e874251 100644 --- a/flask_discord/_http.py +++ b/flask_discord/_http.py @@ -52,6 +52,10 @@ class DiscordOAuth2HttpClient(abc.ABC): os.environ["OAUTHLIB_INSECURE_TRANSPORT"] = "true" app.discord = self + @property + def user_id(self): + """A property which returns Discord user ID if it exists in flask :py:attr:`flask.session` object.""" + @staticmethod def _token_updater(token): session["DISCORD_OAUTH2_TOKEN"] = token diff --git a/flask_discord/models/__init__.py b/flask_discord/models/__init__.py index 85a8be1..c3839d3 100644 --- a/flask_discord/models/__init__.py +++ b/flask_discord/models/__init__.py @@ -1,4 +1,5 @@ from .connections import UserConnection +from .integration import Integration from .user import User, Bot from .guild import Guild