misskey2telegram/defs/announcement.py
2023-07-03 22:39:52 +08:00

84 lines
2.3 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]