ynerant
/
lglog
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Automatically register users

This commit is contained in:
Yohann D'ANELLO 2020-11-16 04:00:23 +01:00
parent 358f8d446a
commit fc18f93ff8
5 changed files with 48 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
lg/models.py
View File

@ -1,12 +1,34 @@
from django.contrib.auth.models import User from django.contrib.auth.models import AbstractUser
from django.db import models from django.db import models
from django.utils.translation import gettext_lazy as _ from django.utils.translation import gettext_lazy as _
from polymorphic.models import PolymorphicModel from polymorphic.models import PolymorphicModel
class DiscordUser(AbstractUser):
discord_id = models.CharField(
max_length=32,
primary_key=True,
verbose_name=_("discord id"),
)
discriminator = models.CharField(
max_length=4,
verbose_name=_("discriminator"),
)
avatar_id = models.CharField(
max_length=32,
verbose_name=_("avatar id"),
)
@property
def avatar_url(self):
return f"https://cdn.discordapp.com/avatars/{self.discord_id}/{self.avatar_id}.png"
class Player(PolymorphicModel): class Player(PolymorphicModel):
user = models.OneToOneField( user = models.OneToOneField(
User, DiscordUser,
on_delete=models.CASCADE, on_delete=models.CASCADE,
verbose_name=_("user"), verbose_name=_("user"),
) )

18
lglog/middlewares.py
View File

@ -2,9 +2,12 @@ from authlib.integrations.base_client import OAuthError
from authlib.integrations.django_client import OAuth from authlib.integrations.django_client import OAuth
from authlib.oauth2.rfc6749 import OAuth2Token from authlib.oauth2.rfc6749 import OAuth2Token
from django.conf import settings from django.conf import settings
from django.contrib.auth import login
from django.shortcuts import redirect from django.shortcuts import redirect
from django.utils.deprecation import MiddlewareMixin from django.utils.deprecation import MiddlewareMixin
from lg.models import DiscordUser
class OAuthMiddleware(MiddlewareMixin): class OAuthMiddleware(MiddlewareMixin):
@ -37,6 +40,7 @@ class OAuthMiddleware(MiddlewareMixin):
if request.session.get('token', None) is not None: if request.session.get('token', None) is not None:
current_user = self.get_current_user(sso_client, request) current_user = self.get_current_user(sso_client, request)
login(request, current_user)
if current_user is not None: if current_user is not None:
return self.get_response(request) return self.get_response(request)
@ -54,13 +58,21 @@ class OAuthMiddleware(MiddlewareMixin):
return None return None
if not OAuth2Token.from_dict(token).is_expired() and 'user' in request.session: if not OAuth2Token.from_dict(token).is_expired() and 'user' in request.session:
return request.session['user'] return DiscordUser.objects.get(discord_id=request.session['user'])
try: try:
res = sso_client.get(settings.OAUTH_CLIENT['userinfo_endpoint'], token=OAuth2Token(token)) res = sso_client.get(settings.OAUTH_CLIENT['userinfo_endpoint'], token=OAuth2Token(token))
if res.ok: if res.ok:
request.session['user'] = res.json() user_json = res.json()
return res.json() discord_id = user_json['id']
user, _ = DiscordUser.objects.get_or_create(discord_id=discord_id)
user.username = user_json['username']
user.discriminator = user_json['discriminator']
user.email = user_json['email']
user.avatar_id = user_json['avatar']
user.save()
request.session['user'] = user.discord_id
return user
raise Exception(res, str(res.__dict__)) raise Exception(res, str(res.__dict__))
except OAuthError as e: except OAuthError as e:
print(e) print(e)

5
lglog/settings.py
View File

@ -141,8 +141,11 @@ CRISPY_TEMPLATE_PACK = 'bootstrap4'
DJANGO_TABLES2_TEMPLATE = 'django_tables2/bootstrap4.html' DJANGO_TABLES2_TEMPLATE = 'django_tables2/bootstrap4.html'
AUTH_USER_MODEL = 'lg.discorduser'
# OAuth Settings # OAuth Settings
OAUTH_URL_WHITELISTS = [] OAUTH_URL_WHITELISTS = [
]
OAUTH_CLIENT_NAME = 'discord' OAUTH_CLIENT_NAME = 'discord'

5
lglog/templates/base.html
View File

@ -49,6 +49,11 @@
<body class="d-flex w-100 h-100 flex-column"> <body class="d-flex w-100 h-100 flex-column">
<main class="mb-auto"> <main class="mb-auto">
<nav class="navbar navbar-expand-md navbar-light bg-light fixed-navbar shadow-sm"> <nav class="navbar navbar-expand-md navbar-light bg-light fixed-navbar shadow-sm">
<div class="container-fluid">
<div class="navbar-nav ml-auto">
{{ user.username }}#{{ user.discriminator }} <img src="{{ user.avatar_url }}" alt="avatar" style="width: 32px; height: 32px;" />
</div>
</div>
</nav> </nav>
{% block fullcontent %} {% block fullcontent %}
<div class="{% block containertype %}container{% endblock %} my-3"> <div class="{% block containertype %}container{% endblock %} my-3">

2
lglog/templates/index.html
View File

@ -6,6 +6,4 @@
{% block content %} {% block content %}
Historique de loup-garou Historique de loup-garou
{{ request.session.user }}
{% endblock %} {% endblock %}