Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import telebot
- import sqlite3
- from telebot import types
- bot = telebot.TeleBot('YOUR_BOT_TOKEN')
- @bot.message_handler(commands=['start'])
- def start(message):
- conn = sqlite3.connect('bimbimbim.sql')
- cur = conn.cursor()
- cur.execute(
- 'CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY AUTOINCREMENT, name VARCHAR(50), pass VARCHAR(50))')
- conn.commit()
- cur.close()
- conn.close()
- bot.send_message(message.chat.id, 'Привет, напиши своё имя, чтобы зарегистрироваться.')
- bot.register_next_step_handler(message, save_name)
- def save_name(message):
- user_name = message.text
- conn = sqlite3.connect('bimbimbim.sql')
- cur = conn.cursor()
- cur.execute('INSERT INTO users (name) VALUES (?)', (user_name,))
- conn.commit()
- cur.close()
- conn.close()
- bot.send_message(message.chat.id, f'Спасибо, {user_name}! Теперь введи свой пароль.')
- # Pass user_name as a parameter to the save_password function
- bot.register_next_step_handler(message, save_password, user_name)
- def save_password(message, user_name):
- user_password = message.text
- conn = sqlite3.connect('bimbimbim.sql')
- cur = conn.cursor()
- cur.execute('UPDATE users SET pass = ? WHERE name = ?', (user_password, user_name))
- conn.commit()
- cur.close()
- conn.close()
- markup = telebot.types.InlineKeyboardMarkup()
- markup.add(telebot.types.InlineKeyboardButton('Список всех пользователей', callback_data='bimbimbim'))
- bot.send_message(message.chat.id, f'Спасибо, {user_name}! Теперь ты зарегистрирован.', reply_markup=markup)
- @bot.callback_query_handler(func=lambda call: call.data == 'bimbimbim')
- def show_users_list(call):
- conn = sqlite3.connect('bimbimbim.sql')
- cur = conn.cursor()
- cur.execute('SELECT name FROM users')
- users = cur.fetchall()
- if users:
- user_list = "\n".join([user[0] for user in users])
- bot.send_message(call.message.chat.id, f'Список всех пользователей:\n{user_list}')
- else:
- bot.send_message(call.message.chat.id, 'Список пользователей пуст.')
- cur.close()
- conn.close()
- bot.polling(none_stop=True)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement