From 829316dae9e7c796ca4c5b9006781db025c0c2de Mon Sep 17 00:00:00 2001 From: Luna Date: Wed, 1 Sep 2021 23:36:59 -0300 Subject: [PATCH] dms: add support for v9 create dm --- litecord/blueprints/dms.py | 9 +++++++-- litecord/schemas.py | 4 ++-- 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/litecord/blueprints/dms.py b/litecord/blueprints/dms.py index 79a728b..3afe316 100644 --- a/litecord/blueprints/dms.py +++ b/litecord/blueprints/dms.py @@ -105,8 +105,13 @@ async def create_dm(user_id, recipient_id): async def start_dm(): """Create a DM with a user.""" user_id = await token_check() - j = validate(await request.get_json(), CREATE_DM) - recipient_id = j["recipient_id"] + j = validate( + await request.get_json(), + CREATE_GROUP_DM_V9 if request.discord_api_version == 9 else CREATE_DM, + ) + recipient_id: str = ( + j["recipients"][0] if request.discord_api_version == 9 else j["recipient_id"] + ) return await create_dm(user_id, recipient_id) diff --git a/litecord/schemas.py b/litecord/schemas.py index 26b3b98..0413d45 100644 --- a/litecord/schemas.py +++ b/litecord/schemas.py @@ -534,11 +534,11 @@ RELATIONSHIP = { CREATE_DM = {"recipient_id": {"type": "snowflake", "required": True}} CREATE_GROUP_DM = { - "recipients": {"type": "list", "required": True, "schema": {"type": "snowflake"}} + "recipient_id": {"type": "list", "required": True, "schema": {"type": "snowflake"}} } CREATE_GROUP_DM_V9 = { - "recipient_id": {"type": "list", "required": True, "schema": {"type": "snowflake"}} + "recipients": {"type": "list", "required": True, "schema": {"type": "snowflake"}} } GROUP_DM_UPDATE = {