mirror of
https://github.com/PaiGramTeam/GramCore.git
synced 2024-11-24 07:10:37 +00:00
50 lines
2.3 KiB
Python
50 lines
2.3 KiB
Python
from typing import Optional
|
|
|
|
from gram_core.base_service import BaseService
|
|
from gram_core.services.channels.cache import ChannelAliasCache
|
|
from gram_core.services.channels.models import ChannelAliasDataBase as ChannelAlias
|
|
from gram_core.services.channels.repositories import ChannelAliasRepository
|
|
|
|
__all__ = ("ChannelAliasService",)
|
|
|
|
|
|
class ChannelAliasService(BaseService):
|
|
def __init__(self, channel_alias_repository: ChannelAliasRepository, cache: ChannelAliasCache):
|
|
self.channel_alias_repository = channel_alias_repository
|
|
self._cache = cache
|
|
|
|
async def initialize(self):
|
|
channels = await self.channel_alias_repository.get_all(is_valid=True)
|
|
for channel in channels:
|
|
if channel.chat_id and channel.user_id:
|
|
await self._cache.set_data(channel.chat_id, channel.user_id)
|
|
|
|
async def get_by_chat_id(self, chat_id: int, is_valid: Optional[bool] = None) -> Optional[ChannelAlias]:
|
|
return await self.channel_alias_repository.get_by_chat_id(chat_id, is_valid)
|
|
|
|
async def get_uid_by_chat_id(self, chat_id: int, is_valid: Optional[bool] = None) -> Optional[int]:
|
|
if uid := await self._cache.get_data(chat_id):
|
|
return uid
|
|
if channel := await self.get_by_chat_id(chat_id, is_valid):
|
|
await self._cache.set_data(channel.chat_id, channel.user_id)
|
|
return channel.user_id
|
|
await self._cache.set_data(chat_id, 0)
|
|
return None
|
|
|
|
async def add_channel_alias(self, channel_alias: ChannelAlias) -> ChannelAlias:
|
|
channel_alias = await self.channel_alias_repository.add(channel_alias)
|
|
await self._cache.set_data(channel_alias.chat_id, channel_alias.user_id)
|
|
return channel_alias
|
|
|
|
async def update_channel_alias(self, channel_alias: ChannelAlias) -> ChannelAlias:
|
|
channel_alias = await self.channel_alias_repository.update(channel_alias)
|
|
if channel_alias.is_valid:
|
|
await self._cache.set_data(channel_alias.chat_id, channel_alias.user_id)
|
|
else:
|
|
await self._cache.delete(channel_alias.chat_id)
|
|
return channel_alias
|
|
|
|
async def remove_channel_alias(self, channel_alias: ChannelAlias):
|
|
await self.channel_alias_repository.remove(channel_alias)
|
|
await self._cache.delete(channel_alias.chat_id)
|