from asyncio import sleep from mipa import Note from module_interface import ModuleInterface from mipa.ext import tasks from mipac import ICreatePoll, MiPoll import random polls = [ ["신기해 보이는 것들", "여러분, 어떤 게 가장 신기할 것 같나요?"], ["맛있을 것 같은 거", "여러분은 어떤 게 가장 맛있을 것 같아요?"], ["무거울 것 같은 것", "이 중에서 어느 게 가장 무거울 것 같나요?"], ["갖고 싶은 거", "여러분, 이 중에서 어느 게 가장 갖고 싶으세요?"], ["무인도에 가져가고 싶은 것", "여러분은 무인도에 한 가지를 가져갈 수 있다면 어떤 걸 가져가고 싶으세요?"], ["집에 장식하고 싶은 것", "여러분은 집에 어떤 장식을 다시나요?"], ["사업 아이템", "여러분은 어떤 게 가장 잘 팔릴 것 같아요?"], ["내렸으면 하는 것들", "여러분, 다음 중에서 어떤 게 하늘에서 내렸으면 좋겠어요?"], ["폰으로 하고 싶은 것", "여러분은 어떤 걸 핸드폰으로 갖고 싶어요?"], ["상품화하고 싶은 것", "여러분들은 상품화 하면 어떤 거 할래요?"], ["발굴될 만한 것", "여러분은 유적에서 발굴될 것 같은 것은 무엇이라고 생각하시나요?"], ["향기가 좋을 것 같은 것", "여러분들은 어떤 게 가장 좋은 향기가 난다고 생각하시나요?"], ["뭔가 비쌀 것 같은 것", "여러분은 어떤 게 가장 고가에 거래될 꺼라고 생각하시나요?"], ["지구 궤도상에 있을 법한 것", "여러분은 어떤 게 지구 궤도를 떠다닐 것 같다고 생각하세요?"], ["선물하고 싶은 것", "여러분은 저에게 어떤 선물을 주고 싶으신가요?"], ["선물 받고 싶은 것", "여러분은 어떤 걸 선물로 받고 싶으세요?"], ["제가 갖고 있을 것 같은 것", "여러분은 제가 어떤 걸 갖고 있을 것 같나요?"], ["유행할 법한 것", "여러분, 어떤 게 유행할 것 같아요?"], ["아침밥", "여러분은 아침밥으로 어떤 걸 드시고 싶으신가요?"], ["점심밥", "여러분, 점심은 뭐로 드시고 싶으세요?"], ["저녁밥", "여러분은 저녁에 어떤 걸 먹고 싶어요?"], ["몸에 좋은 것", "여러분은 어떤 게 몸에 좋을 것 같아요?"], ["후세에 물려주고 싶은 것", "여러분은 어떤 걸 후세에 물려주고 싶으세요?"], ["악기가 될 만한 것", "이 중에서 어떤 걸 악기로 쓸 수 있을까요?"], ["부대찌게에 넣을 만한 것", "여러분, 이 중에서 부대찌게에 넣을 만한 건 어떤 게 있을까요?"], ["후리카케 대용으로 쓰고 싶은 것", "여러분은 어떤 걸 밥에 뿌리고 싶으세요?"], ["자주 보는 것", "여러분, 어떤 걸 자주 보시나요"], ["길가에 떨어져 있을 만한 것", "길가에서 주울 수 있는 건 어떤 게 있을까요?"], ["미술관에 있을법한 것", "여러분은 이 중에서 어떤 게 미술관에 있을 것 같아요?"], ["교실에 있을 것 같은 것", "여러분은 어떤 게 교실에 있을 법한 것 같나요?"], ["이모지가 되었으면 하는 것", "이모지가 되었으면 하는 걸 골라주세요!"], ["CherryPick 본부에 있을 법한 것", "여러분, CherryPick 본부에는 어떤 게 있을 것 같나요?"], ["타는 쓰레기", "이 중에서 어느 게 타는 쓰레기일까요?"], ["좋아하는 삼각김밥 맛", "여러분이 좋아하는 삼각김밥 맛은 무엇인가요?"], ] class Amumal(ModuleInterface): def __init__(self): super().__init__() self.name = "Amumal_Poll" self.regex_pattern = None print("[Amumal_Poll] Random Poll module, Amumal_Poll V1 loaded.") def module_ready(self): print("[Amumal_Poll] Module Ready.") self.amumal_task.start() @tasks.loop(seconds=(60.0 * 60)) async def amumal_task(self): gen_aitem = self.manager.require("Ai-tem").get_func("generate") if random.choice([True, False, False, False, False]): generated_title = random.choice(polls) gen_poll = MiPoll( ICreatePoll( expired_after=1000 * 60 * 60, choices=[gen_aitem(), gen_aitem(), gen_aitem(), gen_aitem()], ) ) sent_note = await self.manager.bot.client.note.action.send( content=generated_title[1], visibility="home", poll=gen_poll ) print("[Amumal_Poll] Poll started") await sleep(60 * 60) await self._poll_end(generated_title[0], sent_note) async def _poll_end(self, title, p_note: Note): p_choices = ( await self.manager.bot.client.note.action.get(p_note.id) ).poll.choices kjosa = self.manager.require("K-Josa").get_func("process_josa") mostVotedChoice = None for p_choice in p_choices: if mostVotedChoice == None: mostVotedChoice = p_choice continue if p_choice.votes > mostVotedChoice.votes: mostVotedChoice = p_choice mostVotedChoices = list( filter(lambda x: x.votes == mostVotedChoice.votes, p_choices) ) if mostVotedChoice.votes == 0: await self.manager.bot.client.note.action.send( content="아무도 투표하지 않았어요... :meow_melt_cry:", visibility="home", renote_id=p_note.id, ) elif len(mostVotedChoices) == 1: p_content = mostVotedChoice.text else: p_content = " 그리고 ".join((i.text for i in mostVotedChoices)) p_content = kjosa(p_content, "이었", "였") await self.manager.bot.client.note.action.send( cw=f"{title} 설문조사 결과 발표에요!", content=f"결과는 {mostVotedChoice.votes}표의 {p_content}어요.", renote_id=p_note.id, ) print( "[Amumal_Poll] Poll ended :" + ", ".join((f"{i.text}:{i.votes}" for i in mostVotedChoices)) )