accept discord.Member type to get_user_color

This commit is contained in:
LBlend committed 2022-11-14 04:06:29 +01:00
1 parent c0ebacf1ed
commit 33342c5257
1 file changed
+4 -3
+4 -3
View File
@@ -1,17 +1,18 @@
import discord
def get_user_color(user: discord.User) -> discord.Color:
def get_user_color(user: discord.User | discord.Member) -> discord.Color:
"""
Returns the user's top role color
Parameters
-----------
user (discord.User): A discord user or member object
user (discord.User|discord.Member): A discord user or member object
Returns
-----------
(discord.Color): The user's displayed color
"""
if str(user.color) != '#000000':
return user.color