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

Use user_id shorthand property

Browse Source
This commit is contained in:
thec0sm0s
2020-05-15 18:31:39 +05:30
parent f970eb94bb
commit 252506e16c
4 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
flask_discord/models/connections.py
View File

@@ -1,6 +1,7 @@
from flask import current_app
from .base import DiscordModelsBase
from .integration import Integration
from flask import current_app, session
class UserConnection(DiscordModelsBase):
@@ -76,7 +77,7 @@ class UserConnection(DiscordModelsBase):
connections = super().fetch_from_api()
if cache:
user = current_app.discord.users_cache.get(session.get("DISCORD_USER_ID"))
user = current_app.discord.users_cache.get(current_app.discord.user_id)
try:
user.connections = connections
except AttributeError:

5
flask_discord/models/guild.py
View File

@@ -1,6 +1,5 @@
from flask import current_app, session
from .base import DiscordModelsBase
from flask import current_app
from .. import configs
@@ -63,7 +62,7 @@ class Guild(DiscordModelsBase):
guilds = super().fetch_from_api()
if cache:
user = current_app.discord.users_cache.get(session.get("DISCORD_USER_ID"))
user = current_app.discord.users_cache.get(current_app.discord.user_id)
try:
user.guilds = {guild.id: guild for guild in guilds}
except AttributeError:

2
flask_discord/models/user.py
View File

@@ -136,7 +136,7 @@ class User(DiscordModelsBase):
If the current doesn't exists in internal cache.
"""
return current_app.discord.users_cache.get(session.get("DISCORD_USER_ID"))
return current_app.discord.users_cache.get(current_app.discord.user_id)
def add_to_guild(self, guild_id) -> dict:
"""Method to add user to the guild, provided OAuth2 session has already been created with ``guilds.join`` scope.