set bot command

#124
This commit is contained in:
BennyThink 2022-06-24 19:26:07 +08:00
parent 86e5a02729
commit a0c48d505e
No known key found for this signature in database
GPG Key ID: 6CD0DBDA5235D481
4 changed files with 21 additions and 5 deletions

View File

@ -225,7 +225,7 @@ docker-compose -f worker.yml up -d
# Command
```
```cmd
start - Let's start
about - What's this bot?
ping - Bot running status

View File

@ -1,6 +1,6 @@
pyrogram==1.4.16
tgcrypto==1.2.3
yt-dlp==2022.5.18
yt-dlp==2022.6.22.1
APScheduler==3.9.1
beautifultable==1.1.0
ffmpeg-python==0.2.0
@ -19,4 +19,4 @@ requests==2.28.0
tqdm==4.64.0
requests-toolbelt==0.9.1
ffpb==0.4.1
youtube-search-python==1.6.5
youtube-search-python==1.6.6

View File

@ -8,6 +8,8 @@
__author__ = "Benny <benny.think@gmail.com>"
import os
import pathlib
import re
import time
from config import (AFD_LINK, COFFEE_LINK, ENABLE_CELERY, ENABLE_VIP, EX,
@ -149,3 +151,7 @@ Sending format: **{1}**
text += f"{status}{hostname} **{active}** {load} {rev}\n"
return text
with open(pathlib.Path(__file__).parent.parent.joinpath("README.md"), encoding="u8") as f:
COMMANDS = re.findall(r"```cmd([\s|\S]*)```", f.read())[0].strip()

View File

@ -15,18 +15,19 @@ import time
import traceback
import typing
from io import BytesIO
from youtubesearchpython import VideosSearch
import pyrogram.errors
from apscheduler.schedulers.background import BackgroundScheduler
from pyrogram import Client, filters, types
from pyrogram.errors.exceptions.bad_request_400 import UserNotParticipant
from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup
from tgbot_ping import get_runtime
from youtubesearchpython import VideosSearch
from client_init import create_app
from config import (AUTHORIZED_USER, ENABLE_CELERY, ENABLE_VIP, OWNER,
REQUIRED_MEMBERSHIP)
from constant import BotText
from constant import COMMANDS, BotText
from db import InfluxDB, MySQL, Redis
from limit import VIP, verify_payment
from tasks import app as celery_app
@ -387,4 +388,13 @@ By @BennyThink, VIP mode: {ENABLE_VIP}, Distribution: {ENABLE_CELERY}
Version: {get_revision()}
"""
print(banner)
with app:
commands = [i for i in COMMANDS.split("\n") if i]
command_list = []
for cmd in commands:
split = cmd.split("-")
command_list.append(pyrogram.types.BotCommand(split[0].strip(), split[1].strip()))
app.set_bot_commands(command_list)
app.run()