misskey2telegram/defs/announcement.py
2023-01-27 20:36:41 +08:00

76 lines
2.2 KiB
Python

from datetime import datetime
from typing import Optional
from mipac import Route
from init import bot
from misskey_init import MisskeyBot
announcement_template = """<b>Misskey Announcement</b>
<b>{0}</b>
{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]