velvox
/
litecord
mirror of https://gitlab.com/litecord/litecord.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fix invites and switch to new format

This commit is contained in:
gabixdev 2018-12-04 00:46:45 +01:00
parent cf5ad107ad
commit 1497456856
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
litecord/blueprints/invites.py
View File

@ -1,5 +1,6 @@
import datetime import datetime
import hashlib import hashlib
import base64
import os import os
from quart import Blueprint, request, current_app as app, jsonify from quart import Blueprint, request, current_app as app, jsonify
@ -31,6 +32,7 @@ async def use_invite(user_id, invite_code):
if inv is None: if inv is None:
raise BadRequest('Unknown invite') raise BadRequest('Unknown invite')
if inv['max_age'] is not 0:
now = datetime.datetime.utcnow() now = datetime.datetime.utcnow()
delta_sec = (now - inv['created_at']).total_seconds() delta_sec = (now - inv['created_at']).total_seconds()
@ -38,7 +40,7 @@ async def use_invite(user_id, invite_code):
await delete_invite(invite_code) await delete_invite(invite_code)
raise BadRequest('Unknown invite (expiried)') raise BadRequest('Unknown invite (expiried)')
if inv['max_uses'] != -1 and inv['uses'] > inv['max_uses']: if inv['max_uses'] is not -1 and inv['uses'] > inv['max_uses']:
await delete_invite(invite_code) await delete_invite(invite_code)
raise BadRequest('Unknown invite (too many uses)') raise BadRequest('Unknown invite (too many uses)')
@ -109,7 +111,7 @@ async def create_invite(channel_id):
ChannelType.GUILD_VOICE.value): ChannelType.GUILD_VOICE.value):
raise BadRequest('Invalid channel type') raise BadRequest('Invalid channel type')
invite_code = hashlib.md5(os.urandom(64)).hexdigest()[:6] invite_code = base64.b64encode(hashlib.md5(os.urandom(64)).digest()).decode("utf-8").replace("/", "").replace("+", "")[:7]
await app.db.execute( await app.db.execute(
""" """