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

Add user property

Browse Source
This commit is contained in:
thecosmos
2019-05-21 00:15:15 +05:30
parent 71404b1d93
commit c85b84bd0a
2 changed files with 20 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
flask_discord/client.py
View File

@@ -85,7 +85,7 @@ class DiscordOAuth2Session(_http.DiscordOAuth2HttpClient):
return self._make_session().authorized return self._make_session().authorized
def fetch_user(self) -> models.User: def fetch_user(self) -> models.User:
"""This method requests user data from discord, caches native :py:class:`flask_discord.models.User` """This method requests current user data from discord, caches native :py:class:`flask_discord.models.User`
to flask `session <http://flask.pocoo.org/docs/1.0/api/#flask.session>`_ object. to flask `session <http://flask.pocoo.org/docs/1.0/api/#flask.session>`_ object.
Returns Returns
@@ -96,6 +96,24 @@ class DiscordOAuth2Session(_http.DiscordOAuth2HttpClient):
session["discord_user"] = models.User(self.get("/users/@me")) session["discord_user"] = models.User(self.get("/users/@me"))
return session["discord_user"] return session["discord_user"]
@property
def user(self) -> models.User:
"""A property which returns cached current :py:class:`flask_discord.models.User` from flask
`session <http://flask.pocoo.org/docs/1.0/api/#flask.session>`_ object.
Note
----
If user is not present in flask `session <http://flask.pocoo.org/docs/1.0/api/#flask.session>`_
object, it requests user data from discord, caches user to session and then returns user object.
Returns
-------
flask_discord.models.User
Cached discord user object form flask `session <http://flask.pocoo.org/docs/1.0/api/#flask.session>`_.
"""
return session.get("discord_user") or self.fetch_user()
def fetch_connections(self) -> models.UserConnection: def fetch_connections(self) -> models.UserConnection:
"""Requests and returns connections of current user from discord. """Requests and returns connections of current user from discord.

2
tests/test_app.py
View File

@@ -24,7 +24,7 @@ def callback():
@app.route("/me/") @app.route("/me/")
def me(): def me():
user = discord.fetch_user() user = discord.user
return f""" return f"""
<html> <html>
<head> <head>