def start(bot, update, chat_data, args):
tg_user = update.message.from_user
chat_id = tg_user.id
# Get or create the user from/in database
User.from_telegram_object(tg_user)
if isinstance(args, list) and len(args) > 0:
# CATEGORY BY ID
try:
cat = Category.get(Category.id == args[0])
from botlistbot.components.explore import send_category
return send_category(bot, update, chat_data, cat)
except (ValueError, Category.DoesNotExist):
pass
query = " ".join(args).lower()
# SPECIFIC DEEP-LINKED QUERIES
if query == const.DeepLinkingActions.CONTRIBUTING:
return help.contributing(bot, update, quote=False)
elif query == const.DeepLinkingActions.EXAMPLES:
return help.examples(bot, update, quote=False)
elif query == const.DeepLinkingActions.RULES:
return help.rules(bot, update, quote=False)
elif query == const.DeepLinkingActions.SEARCH:
return search_handler(bot, update, chat_data)
# SEARCH QUERY
search_query(bot, update, chat_data, query)
else:
bot.sendSticker(
chat_id,
open(
os.path.join(appglobals.ROOT_DIR, "assets", "sticker",
"greetings-humanoids.webp"),
"rb",
),
)
help.help(bot, update)
util.wait(bot, update)
if util.is_private_message(update):
main_menu(bot, update)
return ConversationHandler.END
def access_token(bot, update):
user = User.from_telegram_object(update.effective_user)
rd_token = binascii.hexlify(os.urandom(32)).decode('utf-8')
try:
db_token = APIAccess.get(user=user).token
except:
db_token = "You have no token."
text = "Random token: {}\n" \
"Database token (use this for api calls):\n{}".format(rd_token, db_token)
update.message.reply_text(text)
return ConversationHandler.END
def notify_bot_offline(bot, update, args=None):
tg_user = update.message.from_user
user = User.from_telegram_object(tg_user)
reply_to = util.original_reply_id(update)
if args:
text = " ".join(args)
else:
text = update.message.text
command_no_args = (
len(re.findall(r"^/new\s*$", text)) > 0
or text.lower().strip() == "/offline@botlistbot"
)
if command_no_args:
update.message.reply_text(
util.action_hint(
"Please use this command with an argument. For example:\n/offline @mybot"
),
reply_to_message_id=reply_to,
)
return
# `#offline` is already checked by handler
try:
username = re.match(settings.REGEX_BOT_IN_TEXT, text).groups()[0]
if username == "@" + settings.SELF_BOT_NAME:
log.info("Ignoring {}".format(text))
return
except AttributeError:
if args:
update.message.reply_text(
util.failure("Sorry, but you didn't send me a bot `@username`."),
quote=True,
parse_mode=ParseMode.MARKDOWN,
reply_to_message_id=reply_to,
)
else:
log.info("Ignoring {}".format(text))
# no bot username, ignore update
pass
return
try:
offline_bot = Bot.get(
fn.lower(Bot.username) ** username.lower(), Bot.approved == True
)
try:
Suggestion.get(action="offline", subject=offline_bot)
except Suggestion.DoesNotExist:
suggestion = Suggestion(
user=user,
action="offline",
date=datetime.date.today(),
subject=offline_bot,
)
suggestion.save()
update.message.reply_text(
util.success(
"Thank you! We will review your suggestion and set the bot offline."
),
reply_to_message_id=reply_to,
)
except Bot.DoesNotExist:
update.message.reply_text(
util.action_hint("The bot you sent me is not in the @BotList."),
reply_to_message_id=reply_to,
)
return ConversationHandler.END
def new_bot_submission(bot, update, chat_data, args=None, bot_checker=None):
tg_user = update.message.from_user
user = User.from_telegram_object(tg_user)
if util.stop_banned(update, user):
return
reply_to = util.original_reply_id(update)
if args:
text = " ".join(args)
else:
text = update.message.text
command_no_args = (
len(re.findall(r"^/new\s*$", text)) > 0
or text.lower().strip() == "/new@botlistbot"
)
if command_no_args:
update.message.reply_text(
util.action_hint(
"Please use this command with an argument. For example:\n/new @mybot 🔎"
),
reply_to_message_id=reply_to,
)
return
# `#new` is already checked by handler
try:
username = re.match(settings.REGEX_BOT_IN_TEXT, text).groups()[0]
if username.lower() == "@" + settings.SELF_BOT_NAME.lower():
log.info("Ignoring {}".format(text))
return
except AttributeError:
if args:
update.message.reply_text(
util.failure("Sorry, but you didn't send me a bot `@username`."),
quote=True,
parse_mode=ParseMode.MARKDOWN,
reply_to_message_id=reply_to,
)
log.info("Ignoring {}".format(text))
# no bot username, ignore update
return
try:
new_bot = Bot.by_username(username, include_disabled=True)
if new_bot.disabled:
update.message.reply_text(
util.failure(
"{} is banned from the @BotList.".format(new_bot.username)
),
reply_to_message_id=reply_to,
)
elif new_bot.approved:
update.message.reply_text(
util.action_hint(
"Sorry fool, but {} is already in the @BotList 😉".format(
new_bot.username
)
),
reply_to_message_id=reply_to,
)
else:
update.message.reply_text(
util.action_hint(
"{} has already been submitted. Please have patience...".format(
new_bot.username
)
),
reply_to_message_id=reply_to,
)
return
except Bot.DoesNotExist:
new_bot = Bot(
revision=Revision.get_instance().next,
approved=False,
username=username,
submitted_by=user,
)
new_bot.inlinequeries = "🔎" in text
new_bot.official = "🔹" in text
# find language
languages = Country.select().execute()
for lang in languages:
if lang.emoji in text:
new_bot.country = lang
new_bot.date_added = datetime.date.today()
description_reg = re.match(settings.REGEX_BOT_IN_TEXT + " -\s?(.*)", text)
description_notify = ""
if description_reg:
description = description_reg.group(2)
new_bot.description = description
description_notify = " Your description was included."
new_bot.save()
if (
util.is_private_message(update)
and util.uid_from_update(update) in settings.MODERATORS
):
from botlistbot.components.explore import send_bot_details
send_bot_details(bot, update, chat_data, new_bot)
else:
update.message.reply_text(
util.success(
"You submitted {} for approval.{}".format(new_bot, description_notify)
),
parse_mode=ParseMode.MARKDOWN,
reply_to_message_id=reply_to,
)
# Ask the user to fill in the bot details
util.send_md_message(
bot,
update.effective_user.id,
"Congratulations, you just submitted a bot to the @BotList. Please help us fill in the details below:",
)
edit_bot(bot, update, chat_data, to_edit=new_bot)
try:
check_submission(bot, bot_checker, new_bot)
except Exception as e:
log.exception(e)
return ConversationHandler.END