From 57ddb9e27f71ce6bf42709ec8e2c66f9651c5c65 Mon Sep 17 00:00:00 2001 From: thec0sm0s Date: Thu, 14 May 2020 20:58:01 +0530 Subject: [PATCH] Save discord user id in flask session --- flask_discord/models/user.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/flask_discord/models/user.py b/flask_discord/models/user.py index 60f1cbd..fd53c7b 100644 --- a/flask_discord/models/user.py +++ b/flask_discord/models/user.py @@ -110,10 +110,13 @@ class User(DiscordModelsBase): [cls, ...] List of instances of this model when many of these models exist.""" self = super().fetch_from_api() + session["DISCORD_USER_ID"] = self.id + if guilds: self.fetch_guilds() if connections: self.fetch_connections() + return self def add_to_guild(self, guild_id) -> dict: