from telegram import Update
from telegram.ext import CommandHandler, ContextTypes
from db import SessionLocal
from models import Button

async def edit_button(update: Update, context: ContextTypes.DEFAULT_TYPE):
    args = context.args
    if len(args) < 3:
        await update.message.reply_text('Используйте: /edit_button <название> <новое_название> <новая_ссылка>')
        return
    name, new_name, new_url = args[0], args[1], args[2]
    session = SessionLocal()
    button = session.query(Button).filter_by(name=name).first()
    if not button:
        await update.message.reply_text('Кнопка не найдена.')
        session.close()
        return
    button.name = new_name
    button.url = new_url
    session.commit()
    session.close()
    await update.message.reply_text(f'Кнопка "{name}" изменена.')
