Browse Source

Apply suggestions from reviews

https: //github.com/Rapptz/discord.py/pull/9260#pullrequestreview-2085585270
Co-Authored-By: DA344 <108473820+DA-344@users.noreply.github.com>
pull/10226/head
Soheab_ 5 days ago
parent
commit
cbb83f0c20
  1. 19
      discord/onboarding.py
  2. 12
      discord/types/onboarding.py

19
discord/onboarding.py

@ -89,7 +89,7 @@ class PartialOnboardingPromptOption:
def __init__( def __init__(
self, self,
title: str, title: str,
emoji: Union[Emoji, PartialEmoji, str], emoji: Union[Emoji, PartialEmoji, str] = MISSING,
description: Optional[str] = None, description: Optional[str] = None,
channel_ids: Iterable[int] = MISSING, channel_ids: Iterable[int] = MISSING,
role_ids: Iterable[int] = MISSING, role_ids: Iterable[int] = MISSING,
@ -101,14 +101,17 @@ class PartialOnboardingPromptOption:
self.role_ids: Set[int] = set(role_ids or []) self.role_ids: Set[int] = set(role_ids or [])
def to_dict(self, *, id: int = MISSING) -> PromptOptionPayload: def to_dict(self, *, id: int = MISSING) -> PromptOptionPayload:
if isinstance(self.emoji, str): if self.emoji is not MISSING:
emoji_payload = {"emoji_name": self.emoji} if isinstance(self.emoji, str):
emoji_payload = {"emoji_name": self.emoji}
else:
emoji_payload = {
"emoji_id": self.emoji.id,
"emoji_name": self.emoji.name,
"emoji_animated": self.emoji.animated,
}
else: else:
emoji_payload = { emoji_payload = {}
"emoji_id": self.emoji.id,
"emoji_name": self.emoji.name,
"emoji_animated": self.emoji.animated,
}
return { return {
'id': id or os.urandom(16).hex(), 'id': id or os.urandom(16).hex(),

12
discord/types/onboarding.py

@ -24,7 +24,7 @@ DEALINGS IN THE SOFTWARE.
from __future__ import annotations from __future__ import annotations
from typing import Literal, Optional, TypedDict from typing import Literal, Optional, TypedDict, List
from .emoji import PartialEmoji from .emoji import PartialEmoji
from .snowflake import Snowflake from .snowflake import Snowflake
@ -36,8 +36,8 @@ OnboardingMode = Literal[0, 1]
class PromptOption(TypedDict): class PromptOption(TypedDict):
id: Snowflake id: Snowflake
channel_ids: list[Snowflake] channel_ids: List[Snowflake]
role_ids: list[Snowflake] role_ids: List[Snowflake]
emoji: PartialEmoji emoji: PartialEmoji
title: str title: str
description: Optional[str] description: Optional[str]
@ -45,7 +45,7 @@ class PromptOption(TypedDict):
class Prompt(TypedDict): class Prompt(TypedDict):
id: Snowflake id: Snowflake
options: list[PromptOption] options: List[PromptOption]
title: str title: str
single_select: bool single_select: bool
required: bool required: bool
@ -55,7 +55,7 @@ class Prompt(TypedDict):
class Onboarding(TypedDict): class Onboarding(TypedDict):
guild_id: Snowflake guild_id: Snowflake
prompts: list[Prompt] prompts: List[Prompt]
default_channel_ids: list[Snowflake] default_channel_ids: List[Snowflake]
enabled: bool enabled: bool
mode: OnboardingMode mode: OnboardingMode

Loading…
Cancel
Save