mirror of
https://github.com/nexus-stc/hyperboria
synced 2024-12-12 04:37:50 +01:00
dd23846059
- [nexus] Switch bot - [bot] Added extra receivers functionality GitOrigin-RevId: 68fc32d3e79ff411758f54f435fe8680fc42dead
62 lines
2.2 KiB
Python
62 lines
2.2 KiB
Python
import asyncio
|
|
import logging
|
|
|
|
from izihawa_utils.pb_to_json import MessageToDict
|
|
from library.telegram.base import RequestContext
|
|
from nexus.models.proto.operation_pb2 import \
|
|
DocumentOperation as DocumentOperationPb
|
|
from nexus.models.proto.operation_pb2 import Vote as VotePb
|
|
from nexus.translations import t
|
|
from telethon import events
|
|
|
|
from .base import BaseCallbackQueryHandler
|
|
|
|
|
|
class VoteHandler(BaseCallbackQueryHandler):
|
|
filter = events.CallbackQuery(pattern='^/vote([ab])?_([A-Za-z0-9]+)_([0-9]+)_([bo])$')
|
|
|
|
def parse_pattern(self, event: events.ChatAction):
|
|
short_index_alias = event.pattern_match.group(1)
|
|
index_alias = self.short_index_alias_to_index_alias(short_index_alias.decode()) if short_index_alias else None
|
|
session_id = event.pattern_match.group(2).decode()
|
|
document_id = int(event.pattern_match.group(3).decode())
|
|
vote = event.pattern_match.group(4).decode()
|
|
vote_value = {'b': -1, 'o': 1}[vote]
|
|
|
|
return index_alias, session_id, document_id, vote, vote_value
|
|
|
|
async def handler(self, event: events.ChatAction, request_context: RequestContext):
|
|
index_alias, session_id, document_id, vote, vote_value = self.parse_pattern(event)
|
|
|
|
request_context.add_default_fields(mode='vote', session_id=session_id)
|
|
request_context.statbox(
|
|
action='vote',
|
|
document_id=document_id,
|
|
query=vote,
|
|
index_alias=index_alias,
|
|
)
|
|
|
|
document_operation_pb = DocumentOperationPb(
|
|
vote=VotePb(
|
|
document_id=document_id,
|
|
value=vote_value,
|
|
voter_id=request_context.chat.chat_id,
|
|
),
|
|
)
|
|
logging.getLogger('operation').info(
|
|
msg=MessageToDict(document_operation_pb, preserving_proto_field_name=True),
|
|
)
|
|
|
|
message = await event.get_message()
|
|
|
|
# ToDo: Generalize nexus.views.telegram.common.remove_button and use it here
|
|
return await asyncio.gather(
|
|
self.application.telegram_client.edit_message(
|
|
request_context.chat.chat_id,
|
|
message.id,
|
|
message.text,
|
|
buttons=None,
|
|
),
|
|
event.answer(t('TANKS_BRUH')),
|
|
)
|