lonelyowl13/owlrandomshitbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

restrict prompting to myself

Browse Source
This commit is contained in:
Owl 2025-03-05 15:43:03 +07:00
parent 2bb1e2dccb
commit aa1fb245f5
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
bot.py
View File

@ -3,7 +3,7 @@ from telegram.ext import ApplicationBuilder, ContextTypes, CommandHandler, Messa
CallbackContext, CallbackQueryHandler
import logging
from tgbot.config import TOKEN
from tgbot.config import TOKEN, PROMPTING_USERS
from tgbot.shit.handlers import handle_xitter, handle_red_ebalo, handle_hentai, handle_cute_button
from tgbot.shit.prompting import gen_image
from tgbot.shit.render import render_text_on_image, image_to_string
@ -23,7 +23,7 @@ async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
async def handle_text(update: Update, context: ContextTypes.DEFAULT_TYPE):
if not update.message.text.startswith("prompting"):
if not update.message.text.startswith("prompting") or update.message.from_user["username"] not in PROMPTING_USERS:
file = render_text_on_image("tgbot/assets/red_ebalo.png", update.message.text)
await context.bot.send_sticker(chat_id=update.effective_chat.id, sticker=file)
else:
@ -57,6 +57,7 @@ async def callback_query_handler(update: Update, context: CallbackContext):
else:
await query.answer()
async def handle_inline(update: Update, context: ContextTypes.DEFAULT_TYPE):
query = update.inline_query.query
if not query:
@ -76,6 +77,7 @@ async def handle_inline(update: Update, context: ContextTypes.DEFAULT_TYPE):
await handle_red_ebalo(update, context)
def main():
application = ApplicationBuilder().token(TOKEN).build()

2
tgbot/config.py
View File

@ -12,3 +12,5 @@ DANBOORU_API_KEY = os.getenv("DANBOORU_API_KEY")
IMAGE_GENERATOR_TOKEN: str = os.getenv("IMAGE_GENERATOR_TOKEN")
IMAGE_GENERATOR_API_HOST: str = os.getenv("IMAGE_GENERATOR_API_HOST")
CDN_URL: str = os.getenv("CDN_URL")
PROMPTING_USERS: list = os.getenv("PROMPTING_USERS").split(" ")