150 lines
4.9 KiB
Python
150 lines
4.9 KiB
Python
from tgbot.api import send_message, forward_message, delete_message, ban_member
|
|
from tgbot.config import FEEDBACK_CHAT_ID, WELCOME_MSG, BUTTON_NO, \
|
|
BUTTON_OK, CHAT_ID, BUTTON_OK2, REDIS_URL
|
|
import json
|
|
import redis
|
|
|
|
# сохраняет сессии и пересылаемые сообщения между перезагрузками
|
|
storage = redis.from_url(REDIS_URL)
|
|
|
|
|
|
def handle_feedback(msg):
|
|
mid = msg['message_id']
|
|
cid = msg['chat']['id']
|
|
r = forward_message(cid, mid, FEEDBACK_CHAT_ID).json()
|
|
support_msg_id = r['result']['message_id']
|
|
# store private chat message id
|
|
# fbk-<support-chat-message-id> -> <private-chat-id>:<private-message-id>
|
|
storage.set(f'fbk-{support_msg_id}', json.dumps({
|
|
"message_id": mid,
|
|
"chat_id": cid
|
|
}))
|
|
|
|
|
|
def handle_answer(msg):
|
|
print(f'handle answer from support')
|
|
support_msg_id = str(msg['reply_to_message']['message_id'])
|
|
# get stored private chat id
|
|
stored_feedback = storage.get(f'fbk-{support_msg_id}')
|
|
stored_feedback = json.loads(stored_feedback)
|
|
r = send_message(f'{stored_feedback["chat_id"]}', msg['text'], reply_to=stored_feedback["message_id"]) # notice 'u' before private chat ID
|
|
print(r.json())
|
|
|
|
|
|
def handle_welcome(msg):
|
|
chat_id = str(msg['chat']['id'])
|
|
from_id = str(msg['from']['id'])
|
|
member_id = str(msg['new_chat_member']['id'])
|
|
s = {}
|
|
if from_id == member_id:
|
|
print(f'new self-joined member {member_id}')
|
|
reply_markup = {
|
|
"inline_keyboard": [
|
|
[
|
|
{"text": BUTTON_NO, "callback_data": BUTTON_NO},
|
|
{"text": BUTTON_OK, "callback_data": BUTTON_OK}
|
|
]
|
|
]
|
|
}
|
|
r = send_message(
|
|
chat_id,
|
|
WELCOME_MSG,
|
|
reply_to=msg['message_id'],
|
|
reply_markup=reply_markup
|
|
)
|
|
welcome_msg_id = r.json()['result']['message_id']
|
|
print(f'welcome message id: {welcome_msg_id}')
|
|
s["newcomer"] = True
|
|
s["welcome_id"] = welcome_msg_id
|
|
else:
|
|
s['newcomer'] = False
|
|
|
|
# create new member session
|
|
storage.set(f'usr-{member_id}', json.dumps(s))
|
|
|
|
|
|
def handle_left(msg):
|
|
member_id = msg["left_chat_member"]["id"]
|
|
|
|
# read member session
|
|
s = storage.get(f'usr-{member_id}')
|
|
if s:
|
|
s = json.loads(s)
|
|
r = delete_message(CHAT_ID, s['welcome_id'])
|
|
print(r.json())
|
|
|
|
# remove left member session
|
|
storage.delete(f'usr-{member_id}')
|
|
|
|
|
|
def handle_text(msg):
|
|
member_id = str(msg['from']['id'])
|
|
|
|
# check if author is self-joined newcomer
|
|
author = storage.get(f'usr-{member_id}')
|
|
|
|
if author:
|
|
author = json.loads(author)
|
|
if author.get("newcomer"):
|
|
print(f'new member speaks {msg["text"]}')
|
|
answer = msg['text']
|
|
if BUTTON_OK.lower() in answer.lower() or \
|
|
BUTTON_OK2.lower() in answer.lower():
|
|
print('found answer, cleanup')
|
|
r = delete_message(CHAT_ID, author["welcome_id"])
|
|
print(r.json())
|
|
author["newcomer"] = False
|
|
|
|
# set author as not a newcomer
|
|
storage.set(f'usr-{member_id}', json.dumps(author))
|
|
|
|
else:
|
|
print('remove some message')
|
|
r = delete_message(CHAT_ID, msg['message_id'])
|
|
print(r.json())
|
|
else:
|
|
print(f'old member speaks {msg["text"]}')
|
|
|
|
|
|
def handle_button(callback_query):
|
|
member_id = str(callback_query['from']['id'])
|
|
callback_data = callback_query['data']
|
|
reply_owner = str(callback_query['message']['reply_to_message']['from']['id'])
|
|
welcome_msg_id = str(callback_query['message']['message_id'])
|
|
enter_msg_id = str(callback_query['message']['reply_to_message']['message_id'])
|
|
if reply_owner == member_id:
|
|
print(f'callback_query in {CHAT_ID}')
|
|
|
|
# read session
|
|
s = storage.get(f'usr-{member_id}')
|
|
if s:
|
|
s = json.loads(s)
|
|
else:
|
|
print('no user session found, create')
|
|
s = {
|
|
'newcomer': True,
|
|
'welcome_id': welcome_msg_id
|
|
}
|
|
storage.set(f'usr-{member_id}', json.dumps(s))
|
|
|
|
if callback_data == BUTTON_NO:
|
|
print('wrong answer, cleanup')
|
|
r = delete_message(CHAT_ID, enter_msg_id)
|
|
print(r.json())
|
|
r = delete_message(CHAT_ID, welcome_msg_id)
|
|
print(r.json())
|
|
|
|
# remove banned member session
|
|
storage.delete(f'usr-{member_id}')
|
|
|
|
print('ban member')
|
|
r = ban_member(CHAT_ID, member_id)
|
|
print(r.json())
|
|
elif callback_data == BUTTON_OK:
|
|
print('proper answer, cleanup')
|
|
r = delete_message(CHAT_ID, welcome_msg_id)
|
|
print(r.json())
|
|
s['newcomer'] = False
|
|
|
|
# store new member session
|
|
storage.set(f'usr-{member_id}', json.dumps(s)) |