@DEUS9

Как передать id строки по цепочке диалога для контроля вставки данных?

Код диалога и мне нужно хранить введенные значение в базу. При первой вставке я получу id строки и дальше буду делать update столбцов этой строки. Как мне передать данные (id строки) дальше по цепочки диалога? Каждый раз дергать последнюю вставленную строку - не правильно.
Код
from telegram import (ReplyKeyboardMarkup, ReplyKeyboardRemove)
from telegram.ext import (Updater, CommandHandler, MessageHandler, Filters, RegexHandler,
                          ConversationHandler)
 
import logging
 
# Enable logging
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
                    level=logging.INFO)
 
logger = logging.getLogger(__name__)
 
GENDER, PHOTO, LOCATION, BIO = range(4)
 
 
def start(bot, update):
    reply_keyboard = [['Boy', 'Girl', 'Other']]
    
    update.message.reply_text(
        'Hi! My name is Professor Bot. I will hold a conversation with you. '
        'Send /cancel to stop talking to me.\n\n'
        'Are you a boy or a girl?',
        reply_markup=ReplyKeyboardMarkup(reply_keyboard, one_time_keyboard=True))
 
    return GENDER
 
 
def gender(bot, update):
    user = update.message.from_user
    gend = update.message.text
    conn = psycopg2.connect(Tokens.postgresql('live'))
    cur = conn.cursor()
    sql = "INSERT INTO users(gend) VALUES (%s)"
    cur.execute(sql, (gend, ))
     
    logging.info('ДОБАВЛЕН')
    conn.commit()
    #Получаем id вставленной строки
    #id='5'
    cur.close()
    logger.info("Gender of %s: %s", user.first_name, update.message.text)
    update.message.reply_text('I see! Please send me a photo of yourself, '
                              'so I know what you look like, or send /skip if you don\'t want to.',
                              reply_markup=ReplyKeyboardRemove())
 
    return PHOTO
 
 
def photo(bot, update):
    user = update.message.from_user
    photo_file = bot.get_file(update.message.photo[-1].file_id)
    photo_file.download('user_photo.jpg')
    logger.info("Photo of %s: %s", user.first_name, 'user_photo.jpg')
    update.message.reply_text('Gorgeous! Now, send me your location please, '
                              'or send /skip if you don\'t want to.')
 
    return BIO
def bio(bot, update):
    user = update.message.from_user
    logger.info("Bio of %s: %s", user.first_name, update.message.text)
    update.message.reply_text('Thank you! I hope we can talk again some day.')
 
    return ConversationHandler.END
def main():
    # Create the EventHandler and pass it your bot's token.
    updater = Updater("TOKEN")
 
    # Get the dispatcher to register handlers
    dp = updater.dispatcher
 
    # Add conversation handler with the states GENDER, PHOTO, LOCATION and BIO
    conv_handler = ConversationHandler(
        entry_points=[CommandHandler('start', start)],
 
        states={
            GENDER: [RegexHandler('^(Boy|Girl|Other)$', gender)],
 
            PHOTO: [MessageHandler(Filters.photo, photo),
                    CommandHandler('skip', skip_photo)],
 
            LOCATION: [MessageHandler(Filters.location, location),
                       CommandHandler('skip', skip_location)],
 
            BIO: [MessageHandler(Filters.text, bio)]
        },
 
        fallbacks=[CommandHandler('cancel', cancel)]
    )
 
    dp.add_handler(conv_handler)
 
    # log all errors
    dp.add_error_handler(error)
 
    # Start the Bot
    updater.start_polling()
 
    # Run the bot until you press Ctrl-C or the process receives SIGINT,
    # SIGTERM or SIGABRT. This should be used most of the time, since
    # start_polling() is non-blocking and will stop the bot gracefully.
    updater.idle()
 
 
if __name__ == '__main__':
    main()

  • Вопрос задан
  • 124 просмотра
Пригласить эксперта
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Войти через центр авторизации
Похожие вопросы