aho/welcomecenterbot
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixx

Browse Source
This commit is contained in:
tonyrewin
2023-04-16 21:04:12 +03:00
parent 94fa599c33
commit 6ce2018b2b
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
api/index.py
View File

@@ -6,9 +6,10 @@ from sanic.response import json, text
app = Sanic() app = Sanic()
WEBHOOK = os.environ.get('VERCEL_URL') WEBHOOK = os.environ.get('VERCEL_URL')
CHAT_ID = os.environ.get('CHAT_ID').replace("-", "-100") CHAT_ID = os.environ.get('CHAT_ID').replace("-", "-100")
WELCOME_MSG = os.environ.get('WELCOME_MSG') or 'Welcome! Press the wright button' WELCOME_MSG = os.environ.get('WELCOME_MSG') or 'Welcome! Press the button'
BUTTON_OK = os.environ.get('BUTTON_OK') or 'Ok' BUTTON_OK = os.environ.get('BUTTON_OK') or 'Ok'
BUTTON_NO = os.environ.get('BUTTON_NO') or 'No' BUTTON_NO = os.environ.get('BUTTON_NO') or 'No'
@@ -28,9 +29,10 @@ async def handle(req):
print(req) print(req)
try: try:
update = req.json update = req.json
print(update) # print(update)
msg = update.get('message', update.get('my_chat_member')) msg = update.get('message', update.get('my_chat_member'))
if str(msg['chat']['id']) == CHAT_ID: if str(msg['chat']['id']) == CHAT_ID:
print(f'message in chat')
if 'new_chat_member' in msg: if 'new_chat_member' in msg:
chat_id = str(msg['chat']['id']) chat_id = str(msg['chat']['id'])
member_id = str(msg['new_chat_member']['id']) member_id = str(msg['new_chat_member']['id'])
@@ -62,11 +64,12 @@ async def handle(req):
else: else:
delete_message(msg['message_id']) delete_message(msg['message_id'])
if 'callback_query' in update: if 'callback_query' in update:
msg = update['callback_query'] print(f'callback_query')
chat_id = str(msg['message']['chat']['id']) callback_query = update['callback_query']
chat_id = str(callback_query['message']['chat']['id'])
if chat_id == CHAT_ID: if chat_id == CHAT_ID:
member_id = str(msg['from']['id']) member_id = str(callback_query['from']['id'])
callback_data = msg['data'] callback_data = callback_query['data']
if callback_data == BUTTON_NO: if callback_data == BUTTON_NO:
ban_member(member_id, CHAT_ID) ban_member(member_id, CHAT_ID)
elif callback_data == BUTTON_OK: elif callback_data == BUTTON_OK: