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

admin_api.voice: handle UniqueViolationError

This commit is contained in:
Luna 2019-03-04 01:26:53 -03:00
parent 2d7dc05453
commit 9d80cb5564
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
litecord/blueprints/admin_api/voice.py
View File

@ -17,11 +17,13 @@ along with this program. If not, see <http://www.gnu.org/licenses/>.
""" """
import asyncpg
from quart import Blueprint, jsonify, current_app as app, request from quart import Blueprint, jsonify, current_app as app, request
from litecord.auth import admin_check from litecord.auth import admin_check
from litecord.schemas import validate from litecord.schemas import validate
from litecord.admin_schemas import VOICE_SERVER, VOICE_REGION from litecord.admin_schemas import VOICE_SERVER, VOICE_REGION
from litecord.errors import BadRequest
bp = Blueprint('voice_admin', __name__) bp = Blueprint('voice_admin', __name__)
@ -58,10 +60,13 @@ async def put_region_server(region):
await admin_check() await admin_check()
j = validate(await request.get_json(), VOICE_SERVER) j = validate(await request.get_json(), VOICE_SERVER)
await app.db.execute(""" try:
INSERT INTO voice_servers (hostname, region) await app.db.execute("""
VALUES ($1, $2) INSERT INTO voice_servers (hostname, region_id)
""", j['hostname'], region) VALUES ($1, $2)
""", j['hostname'], region)
except asyncpg.UniqueViolationError:
raise BadRequest('voice server already exists with given hostname')
return '', 204 return '', 204