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

guilds: fix region update vip check

This commit is contained in:
Luna 2019-03-15 03:46:12 -03:00
parent 448b74f91b
commit 2bc3817a4e
1 changed files with 18 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
litecord/blueprints/guilds.py
View File

@ -238,6 +238,21 @@ async def _guild_update_icon(scope: str, guild_id: int,
""", new_icon.icon_hash, guild_id) """, new_icon.icon_hash, guild_id)
async def _guild_update_region(guild_id, region):
is_vip = region.vip
can_vip = await app.storage.has_feature(guild_id, 'VIP_REGIONS')
if is_vip and not can_vip:
raise BadRequest('can not assign guild to vip-only region')
await app.db.execute("""
UPDATE guilds
SET region = $1
WHERE id = $2
""", region.id, guild_id)
@bp.route('/<int:guild_id>', methods=['PATCH']) @bp.route('/<int:guild_id>', methods=['PATCH'])
async def _update_guild(guild_id): async def _update_guild(guild_id):
user_id = await token_check() user_id = await token_check()
@ -263,17 +278,10 @@ async def _update_guild(guild_id):
""", j['name'], guild_id) """, j['name'], guild_id)
if 'region' in j: if 'region' in j:
is_vip = app.voice.lvsp.region(j['region']).vip region = app.voice.lvsp.region(j['region'])
can_vip = await app.storage.has_feature(guild_id, 'VIP_REGIONS')
if is_vip and not can_vip: if region is not None:
raise BadRequest('can not assign guild to vip-only region') await _guild_update_region(guild_id, region)
await app.db.execute("""
UPDATE guilds
SET region = $1
WHERE id = $2
""", j['region'], guild_id)
if 'icon' in j: if 'icon' in j:
await _guild_update_icon( await _guild_update_icon(