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__(
self,
title: str,
emoji: Union[Emoji, PartialEmoji, str],
emoji: Union[Emoji, PartialEmoji, str] = MISSING,
description: Optional[str] = None,
channel_ids: Iterable[int] = MISSING,
role_ids: Iterable[int] = MISSING,
@ -101,14 +101,17 @@ class PartialOnboardingPromptOption:
self.role_ids: Set[int] = set(role_ids or [])
def to_dict(self, *, id: int = MISSING) -> PromptOptionPayload:
if isinstance(self.emoji, str):
emoji_payload = {"emoji_name": self.emoji}
if self.emoji is not MISSING:
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:
emoji_payload = {
"emoji_id": self.emoji.id,
"emoji_name": self.emoji.name,
"emoji_animated": self.emoji.animated,
}
emoji_payload = {}
return {
'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 typing import Literal, Optional, TypedDict
from typing import Literal, Optional, TypedDict, List
from .emoji import PartialEmoji
from .snowflake import Snowflake
@ -36,8 +36,8 @@ OnboardingMode = Literal[0, 1]
class PromptOption(TypedDict):
id: Snowflake
channel_ids: list[Snowflake]
role_ids: list[Snowflake]
channel_ids: List[Snowflake]
role_ids: List[Snowflake]
emoji: PartialEmoji
title: str
description: Optional[str]
@ -45,7 +45,7 @@ class PromptOption(TypedDict):
class Prompt(TypedDict):
id: Snowflake
options: list[PromptOption]
options: List[PromptOption]
title: str
single_select: bool
required: bool
@ -55,7 +55,7 @@ class Prompt(TypedDict):
class Onboarding(TypedDict):
guild_id: Snowflake
prompts: list[Prompt]
default_channel_ids: list[Snowflake]
prompts: List[Prompt]
default_channel_ids: List[Snowflake]
enabled: bool
mode: OnboardingMode

Loading…
Cancel
Save