49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
from typing import Optional
|
|
from .data import (
|
|
ProfilesData,
|
|
UserProfile,
|
|
DiscordProfileLink,
|
|
TwitchProfileLink,
|
|
Community,
|
|
DiscordCommunityLink,
|
|
TwitchCommunityLink,
|
|
)
|
|
|
|
|
|
class ProfilesRegistry:
|
|
VERSION = ProfilesData.VERSION
|
|
|
|
def __init__(self, data: ProfilesData):
|
|
self.data = data
|
|
|
|
async def init(self):
|
|
await self.data.init()
|
|
|
|
async def get_profile(self, profileid: int) -> Optional[UserProfile]:
|
|
return await UserProfile.fetch(profileid)
|
|
|
|
async def get_profile_discord(self, userid: int) -> Optional[UserProfile]:
|
|
link = await DiscordProfileLink.fetch(userid)
|
|
if link:
|
|
return await UserProfile.fetch(link.profileid)
|
|
|
|
async def get_profile_twitch(self, userid: str) -> Optional[UserProfile]:
|
|
userid = str(userid)
|
|
link = await TwitchProfileLink.fetch(userid)
|
|
if link:
|
|
return await UserProfile.fetch(link.profileid)
|
|
|
|
async def get_community(self, communityid: int) -> Optional[Community]:
|
|
return await Community.fetch(communityid)
|
|
|
|
async def get_community_discord(self, guildid: int) -> Optional[Community]:
|
|
link = await DiscordCommunityLink.fetch(guildid)
|
|
if link:
|
|
return await Community.fetch(link.communityid)
|
|
|
|
async def get_community_twitch(self, channelid: str) -> Optional[Community]:
|
|
channelid = str(channelid)
|
|
link = await TwitchCommunityLink.fetch(channelid)
|
|
if link:
|
|
return await Community.fetch(link.communityid)
|