from datetime import datetime from typing import Optional from mipac import Route from init import bot from misskey_init import MisskeyBot announcement_template = """Misskey Announcement {0} {1}""" class Announcement: def __init__(self, data, misskey_bot: MisskeyBot): self.misskey_bot = misskey_bot self.id = data["id"] self.title = data["title"] self.text = data["text"] self.is_read = data["is_read"] self.image_url = data["image_url"] self._created_at = data["created_at"] self._updated_at = data["updated_at"] @property def created_at(self) -> datetime: return datetime.strptime(self._created_at, "%Y-%m-%dT%H:%M:%S.%fZ") @property def updated_at(self) -> Optional[datetime]: return ( datetime.strptime(self._updated_at, "%Y-%m-%dT%H:%M:%S.%fZ") if self._updated_at else None ) async def send_notice(self): if not self.image_url: await bot.send_message( self.misskey_bot.tg_user.chat_id, announcement_template.format( self.title, self.text[:1000], ), reply_to_message_id=self.misskey_bot.tg_user.notice_topic, ) else: await bot.send_photo( self.misskey_bot.tg_user.chat_id, self.image_url, caption=announcement_template.format( self.title, self.text[:1000], ), reply_to_message_id=self.misskey_bot.tg_user.notice_topic, ) async def mark_as_read(self): data = { "announcementId": self.id, } await self.misskey_bot.core.http.request( Route("POST", "/api/i/read-announcement"), json=data, auth=True, lower=True, ) async def get_unread_announcements(misskey_bot: MisskeyBot): data = { "limit": 10, "withUnreads": True, } req = await misskey_bot.core.http.request( Route("POST", "/api/announcements"), json=data, auth=True, lower=True, ) return [Announcement(i, misskey_bot) for i in req]