tgbots/TeleDiffusionBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add TG Chat DB (KB not working currently)

Browse Source
This commit is contained in:
seel2304
2023-05-05 02:10:45 +03:00
parent 1653e184a8
commit 0bc37fc87f
2 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
bot/handlers/txt2img/txt2img.py
View File

@@ -1,4 +1,5 @@
import re import re
from bot.common import bot
from aiogram import types from aiogram import types
from bot.db import db, DBTables from bot.db import db, DBTables
from bot.utils.cooldown import throttle from bot.utils.cooldown import throttle
@@ -8,6 +9,8 @@ from bot.modules.api.objects.prompt_request import Generated
from bot.modules.api.status import wait_for_status from bot.modules.api.status import wait_for_status
from bot.keyboards.image_info import get_img_info_keyboard from bot.keyboards.image_info import get_img_info_keyboard
from bot.utils.errorable_command import wrap_exception from bot.utils.errorable_command import wrap_exception
from bot.callbacks.factories.image_info import (prompt_only, full_prompt, import_prompt, back)
@wrap_exception([ValueError], custom_loading=True) @wrap_exception([ValueError], custom_loading=True)
@@ -43,6 +46,16 @@ async def generate_command(message: types.Message):
db[DBTables.queue]['n'] = db[DBTables.queue].get('n', 1) - 1 db[DBTables.queue]['n'] = db[DBTables.queue].get('n', 1) - 1
image = await txt2img(prompt) image = await txt2img(prompt)
image_message = await message.reply_photo(photo=image[0]) image_message = await message.reply_photo(photo=image[0])
#Send photo to SD Image Archive
# message_data =
archive_message = f'User ID: {message.from_id} \n \
User nickname: {message.from_user.full_name} \n \
User username: @{message.from_user.username}'
# archive_message = f'{callback_data=full_prompt.new(p_id=f"{image_message.photo[0].file_unique_id}")} + "User ID: " + message.from_id + " User nickname: " + message.from_user.full_name + " User username: " + message.from_user.username'
await bot.send_photo(-929754401, photo=image[0], caption=archive_message, reply_markup=get_img_info_keyboard(image_message.photo[0].file_unique_id))
db[DBTables.generated][image_message.photo[0].file_unique_id] = Generated( db[DBTables.generated][image_message.photo[0].file_unique_id] = Generated(
prompt=prompt, prompt=prompt,

7
bot/utils/private_keyboard.py
View File

@@ -1,10 +1,17 @@
from aiogram import types from aiogram import types
from bot.config import ADMIN
async def other_user(call: types.CallbackQuery) -> bool: async def other_user(call: types.CallbackQuery) -> bool:
if not hasattr(call.message.reply_to_message, 'from_id'): if not hasattr(call.message.reply_to_message, 'from_id'):
await call.answer('Error, original call was removed', show_alert=True) await call.answer('Error, original call was removed', show_alert=True)
return True return True
elif call.from_user.id == ADMIN:
print(f"call.from_user.id: {call.from_user.id}")
print(f"ADMIN: {ADMIN}")
print("User is Admin")
return False
elif call.message.reply_to_message.from_id != call.from_user.id: elif call.message.reply_to_message.from_id != call.from_user.id:
await call.answer('It is not your menu!', show_alert=True) await call.answer('It is not your menu!', show_alert=True)
return True return True