turns-00048.parquet:36993
9718efa779ee9777c2a6aa38degenerate_repetitionAbsentFinal dense release
USER
User: Теперь сделаем файл refferal_button. Данный файл будет создавать кнопку в меню "👥 Друзья". Ее можно будет изменить в message.py. Она отправляет сообщение:
"(Bold) 👥 Пригласите друзей и получайте по 3 Ирис 🍬
(Bold) 🔗 Ваша ссылка: (Тут нужна реферальная ссылка. Например, можно взять за основу как у другого бота, например: https://t.me/(Имя бота)?start=(Айди игрока)"
Если игрок, которому ссылка пренадлежит, или тот, кто уже по ней заходил еще раз по ней зайдет, игроку чья ссылка уже не начислиться 3 Ирис. Просто он перейдет по ссылке и ничего не произойдет. Нужно, чтоб аккаунт заходил в первые или не являлся владельцем ссылки. Так-же, главное чтобы переходил новый пользователь, который никогда не заходил в бота и не писал /start и т.д. Так-же добавь сразу подчет, сколько игрок пригласил друзей в бота, в будущем для профиля.
main.py:
import telebot
from tnik import TOKEN
from telebot.types import InlineKeyboardMarkup, InlineKeyboardButton
from message import (
WELCOME_MESSAGE, EARN_BUTTON_TEXT, PROFILE_BUTTON_TEXT,
NORMAL_CLICK_BUTTON, SUPER_CLICK_BUTTON,
REFRESH_STATS_BUTTON, MAX_SUPER_CLICKS_MESSAGE
)
from button_job import create_earn_button
from database import init_database, get_user_data, update_click_data, can_super_click
from profile_button import handle_profile
# Инициализируем бота
bot = telebot.TeleBot(TOKEN)
# Инициализация базы данных
init_database()
def create_inline_buttons():
markup = InlineKeyboardMarkup()
normal_click_button = InlineKeyboardButton(NORMAL_CLICK_BUTTON, callback_data='normal_click')
super_click_button = InlineKeyboardButton(SUPER_CLICK_BUTTON, callback_data='super_click')
markup.row(normal_click_button, super_click_button)
refresh_button = InlineKeyboardButton(REFRESH_STATS_BUTTON, callback_data='refresh')
markup.add(refresh_button)
return markup
@bot.message_handler(commands=['start'])
def send_welcome(message):
bot.send_message(
message.chat.id,
f"*{WELCOME_MESSAGE}*",
parse_mode='Markdown',
reply_markup=create_earn_button()
)
@bot.message_handler(func=lambda m: m.text == EARN_BUTTON_TEXT)
def send_earn_message(message):
user_id = message.from_user.id
simple_clicks, super_clicks, balance, _ = get_user_data(user_id)
earn_message = (
f"За каждый простой клик вы получите: 0.005 Ирис 🍬 🟢\n"
f"За каждый супер клик вы получите: 0.2 Ирис 🍬 🔴\n\n"
f"*Всего простых кликов: {simple_clicks} 🟢*\n"
f"*Всего супер кликов: {super_clicks} 🔴*"
)
bot.send_message(
message.chat.id,
earn_message,
parse_mode='Markdown',
reply_markup=create_inline_buttons()
)
@bot.message_handler(func=lambda m: m.text == PROFILE_BUTTON_TEXT)
def show_profile(message):
handle_profile(bot, message)
@bot.callback_query_handler(func=lambda call: True)
def callback_inline(call):
user_id = call.from_user.id
simple_clicks, super_clicks, balance, last_super_click = get_user_data(user_id)
if call.data == 'normal_click':
update_click_data(user_id, 'simple', 0.005)
bot.answer_callback_query(call.id, "Добавлено 0.005 Ирис")
elif call.data == 'super_click':
if super_clicks >= 5 and not can_super_click(last_super_click):
bot.answer_callback_query(call.id, MAX_SUPER_CLICKS_MESSAGE, show_alert=True)
else:
update_click_data(user_id, 'super', 0.2)
bot.answer_callback_query(call.id, "Добавлено 0.2 Ирис")
elif call.data == 'refresh':
simple_clicks, super_clicks, balance, _ = get_user_data(user_id)
new_earn_message = (
f"За каждый простой клик вы получите: 0.005 Ирис 🍬 🟢\n"
f"За каждый супер клик вы получите: 0.2 Ирис 🍬 🔴\n\n"
f"*Всего простых кликов: {simple_clicks} 🟢*\n"
f"*Всего супер кликов: {super_clicks} 🔴*"
)
try:
bot.edit_message_text(
chat_id=call.message.chat.id,
message_id=call.message.message_id,
text=new_earn_message,
parse_mode='Markdown',
reply_markup=create_inline_buttons()
)
except telebot.apihelper.ApiTelegramException as e:
if "message is not modified" in str(e):
pass
else:
raise
# Запускаем бота
if __name__ == '__main__':
bot.polling(none_stop=True)
database.py:
import sqlite3
import datetime
def init_database():
conn = sqlite3.connect('clicker.db')
cursor = conn.cursor()
cursor.execute('''
CREATE TABLE IF NOT EXISTS users (
user_id INTEGER PRIMARY KEY,
simple_clicks INTEGER DEFAULT 0,
super_clicks INTEGER DEFAULT 0,
balance REAL DEFAULT 0.0,
last_super_click TEXT
)''')
conn.commit()
conn.close()
def get_user_data(user_id):
conn = sqlite3.connect('clicker.db')
cursor = conn.cursor()
cursor.execute('SELECT simple_clicks, super_clicks, balance, last_super_click FROM users WHERE user_id = ?', (user_id,))
result = cursor.fetchone()
if not result:
cursor.execute('INSERT INTO users (user_id) VALUES (?)', (user_id,))
conn.commit()
return (0, 0, 0.0, None)
else:
return result
def update_click_data(user_id, click_type, value):
conn = sqlite3.connect('clicker.db')
cursor = conn.cursor()
if click_type == 'simple':
cursor.execute('UPDATE users SET simple_clicks = simple_clicks + 1, balance = balance + ? WHERE user_id = ?', (value, user_id))
elif click_type == 'super':
now = datetime.datetime.now().isoformat()
cursor.execute('UPDATE users SET super_clicks = super_clicks + 1, balance = balance + ?, last_super_click = ? WHERE user_id = ?', (value, now, user_id))
conn.commit()
conn.close()
def can_super_click(last_super_click):
if last_super_click is None:
return True
last_click_time = datetime.datetime.fromisoformat(last_super_click)
now = datetime.datetime.now()
return (now - last_click_time).days >= 1
button_job.py (Работа с меню, где и должна быть кнопка):
from telebot.types import ReplyKeyboardMarkup, KeyboardButton
from message import EARN_BUTTON_TEXT, PROFILE_BUTTON_TEXT
def create_earn_button():
markup = ReplyKeyboardMarkup(resize_keyboard=True)
button_earn = KeyboardButton(EARN_BUTTON_TEXT)
button_profile = KeyboardButton(PROFILE_BUTTON_TEXT)
markup.add(button_earn, button_profile)
return markup
Assistant: Чтобы добавить реферальную систему и кнопку "👥 Друзья" в ваш проект, вам нужно будет сделать следующие изменения в нескольких местах вашего кода. Ниже я покажу, как можно реализовать это.
### 1. Изменения в `database.py`
Добавьте новую таблицу для отслеживания рефералов и данные о приглашенных друзьях:
```python
def init_database():
conn = sqlite3.connect('clicker.db')
cursor = conn.cursor()
cursor.execute('''
CREATE TABLE IF NOT EXISTS users (
user_id INTEGER PRIMARY KEY,
simple_clicks INTEGER DEFAULT 0,
super_clicks INTEGER DEFAULT 0,
balance REAL DEFAULT 0.0,
last_super_click TEXT,
invited_friends INTEGER DEFAULT 0
)''')
cursor.execute('''
CREATE TABLE IF NOT EXISTS referrals (
referrer_id INTEGER,
referee_id INTEGER,
UNIQUE(referrer_id, referee_id)
)''')
conn.commit()
conn.close()
def add_referral(referrer_id, referee_id):
conn = sqlite3.connect('clicker.db')
cursor = conn.cursor()
try:
cursor.execute('INSERT INTO referrals (referrer_id, referee_id) VALUES (?, ?)', (referrer_id, referee_id))
cursor.execute('UPDATE users SET invited_friends = invited_friends + 1, balance = balance + 3.0 WHERE user_id = ?', (referrer_id,))
conn.commit()
except sqlite3.IntegrityError:
pass # Этот реферал уже существует
conn.close()
```
### 2. Создайте файл `referral_button.py`
Этот файл создаст кнопку и обработает команду.
```python
from telebot.types import ReplyKeyboardMarkup, KeyboardButton
from message import REFERRAL_BUTTON_TEXT
from telebot import TeleBot
def create_referral_button():
markup = ReplyKeyboardMarkup(resize_keyboard=True)
referral_button = KeyboardButton(REFERRAL_BUTTON_TEXT)
markup.add(referral_button)
return markup
def handle_referral(bot: TeleBot, message):
user_id = message.from_user.id
referral_link = f"https://t.me/{bot.get_me().username}?start={user_id}"
referral_message = (
"👥 *Пригласите друзей и получайте по 3 Ирис 🍬*\n\n"
f"🔗 *Ваша ссылка: {referral_link}*"
)
bot.send_message(message.chat.id, referral_message, parse_mode='Markdown')
```
### 3. Изменения в `message.py`
Добавьте текст для новой кнопки:
```python
REFERRAL_BUTTON_TEXT = "👥 Друзья"
```
### 4. Изменения в `button_job.py`
Добавьте кнопку "👥 Друзья" в главное меню:
```python
from message import EARN_BUTTON_TEXT, PROFILE_BUTTON_TEXT, REFERRAL_BUTTON_TEXT
def create_earn_button():
markup = ReplyKeyboardMarkup(resize_keyboard=True)
button_earn = KeyboardButton(EARN_BUTTON_TEXT)
button_profile = KeyboardButton(PROFILE_BUTTON_TEXT)
button_referral = KeyboardButton(REFERRAL_BUTTON_TEXT)
markup.add(button_earn, button_profile, button_referral)
return markup
```
### 5. Изменения в `main.py`
Добавьте обработчик для команды "/start" с реферальной ссылкой и для кнопки "👥 Друзья":
```python
from referral_button import handle_referral
from database import add_referral
@bot.message_handler(commands=['start'])
def send_welcome(message):
referrer_id = None
# Проверяем, есть ли у команды аргумент (реферальная ссылка)
if len(message.text.split()) > 1:
referrer_id = int(message.text.split()[1])
# Добавляем рефералла только если это не повторный запуск и не сам себе
if referrer_id != message.from_user.id:
add_referral(referrer_id, message.from_user.id)
bot.send_message(
message.chat.id,
f"*{WELCOME_MESSAGE}*",
parse_mode='Markdown',
reply_markup=create_earn_button()
)
@bot.message_handler(func=lambda m: m.text == REFERRAL_BUTTON_TEXT)
def send_referral_message(message):
handle_referral(bot, message)
```
Эти изменения добавят кнопку "👥 Друзья" в меню, с возможностью отправки реферальной ссылки и начислением бонусов за новых приглашенных пользователей.
User: Теперь, добавим под сообщение с профилем инлайн кнопку "📤 Вывести (Число, сколько ирис у игрока, в соответствии с Баланс-ом) Ирис 🍬". При нажатии на инлайн кнопку, проверяеться, есть ли у игрока, чей профиль, 100000 Ирис 🍬, если нет, то мы ничего не делаем при нажатии на кнопку, а просто выводим ему табличку "🚫 Минимальная сумма вывода: 100000 Ирис 🍬". А если у него такое есть, то я расскажу позже.
Assistant: Добавление кнопки "📤 Вывести" в сообщение с профилем предполагает, что когда пользователь достигает нужного баланса, он может перейти к процессу вывода средств. Вот как это можно реализовать в существующем коде.
### 1. Изменения в `profile_button.py`
Добавьте функцию для создания кнопки и обработку нажатия на неё:
```python
from telebot.types import InlineKeyboardMarkup, InlineKeyboardButton
def create_withdraw_button(balance):
markup = InlineKeyboardMarkup()
withdraw_button = InlineKeyboardButton(f"📤 Вывести {balance} Ирис 🍬", callback_data='withdraw')
markup.add(withdraw_button)
return markup
def handle_profile(bot, message):
user_id = message.from_user.id
simple_clicks, super_clicks, balance, _ = get_user_data(user_id)
profile_message = (
f"*Профиль пользователя:*\n"
f"🆔 ID: {user_id}\n"
f"🟢 Простые клики: {simple_clicks}\n"
f"🔴 Супер клики: {super_clicks}\n"
f"💰 Баланс: {balance} Ирис 🍬"
)
bot.send_message(
message.chat.id,
profile_message,
parse_mode='Markdown',
reply_markup=create_withdraw_button(balance)
)
def handle_withdraw(bot, call):
user_id = call.from_user.id
_, _, balance, _ = get_user_data(user_id)
if balance < 100000:
bot.answer_callback_query(call.id, "🚫 Минимальная сумма вывода: 100000 Ирис 🍬", show_alert=True)
else:
# Здесь можно будет реализовать логику вывода, о чем вы упомянули, что расскажите позже
bot.answer_callback_query(call.id, "✅ Вы можете вывести средства.", show_alert=True)
```
### 2. Изменения в `main.py`
Добавьте обработку нового callback_query для "withdraw":
```python
@bot.callback_query_handler(func=lambda call: call.data == 'withdraw')
def callback_withdraw(call):
handle_withdraw(bot, call)
```
### Объяснение
1. **Создание инлайн-кнопки**: Функция `create_withdraw_button` создает инлайн-кнопку с текстом, отображающим текущий баланс Ирис у пользователя.
2. **Проверка баланса**: В функции `handle_withdraw` проверяется, достаточно ли у пользователя средств для вывода. Если нет, показывается предупреждение о минимальной сумме для вывода.
3. **Обработка выведения**: На данный момент, если у пользователя достаточно средств, выводится лишь сообщение об успешной проверке. Ранее вы упомянули, что позже расскажете подробности о дальнейшем процессе вывода средств.
Следующим шагом можно будет интегрировать процесс вывода средств, такой как отправка денег, если условия для вывода выполнены.
User: Добавь, что если бота пытаються добавить в группу, то он сразу выходит из группыю
Assistant: Чтобы бот автоматически покидал группу, в которую его добавили, можно использовать обработчик событий, который реагирует на изменение участников чата. Когда бот обнаруживает, что его добавили в группу, он немедленно покидает её. Вот как это можно реализовать в вашем коде:
### Изменения в `main.py`
Добавьте обработчик для события, когда бот добавляется в новую группу:
```python
@bot.message_handler(content_types=['new_chat_members'])
def handle_new_chat_members(message):
# Проверяем, если бот находится среди новых участников
for member in message.new_chat_members:
if member.id == bot.get_me().id:
bot.leave_chat(message.chat.id)
break
```
### Объяснение
1. **Обработчик событий нового участника**: `@bot.message_handler(content_types=['new_chat_members'])` — этот обработчик срабатывает каждый раз, когда в чат добавляются новые участники.
2. **Проверка среди новых участников**: В этой функции мы перебираем всех новых участников чата и проверяем, если ID бота совпадает с ID одного из новых членов чата.
3. **Покинуть группу**: Если бот обнаруживает, что его добавили в группу, он вызывает метод `leave_chat`, чтобы немедленно покинуть её.
Эта реализация гарантирует, что бот больше не будет состоять в группе сразу после добавления.
User: Баланс должен парситься так, например у человека 200.02 ириса, и так-же мы и пишем, 200.02
def create_withdraw_button(balance):
markup = InlineKeyboardMarkup()
withdraw_button = InlineKeyboardButton(f"📤 Вывести {balance} Ирис 🍬", callback_data='withdraw')
markup.add(withdraw_button)
return markup
Assistant: Чтобы обеспечить правильное отображение баланса, включая две десятичные позиции, вы можете форматировать число с использованием Python-оператора форматирования строк. Это гарантирует, что количество Ирис будет отображаться с точностью до двух знаков после запятой, даже если в памяти число хранится как целое число или с другим числом знаков после запятой. Вот как можно изменить функцию `create_withdraw_button`:
```python
def create_withdraw_button(balance):
markup = InlineKeyboardMarkup()
# Форматируем баланс с двумя знаками после запятой
formatted_balance = f"{balance:.2f}"
withdraw_button = InlineKeyboardButton(f"📤 Вывести {formatted_balance} Ирис 🍬", callback_data='withdraw')
markup.add(withdraw_button)
return markup
```
### Объяснение
1. **Форматирование баланса**: Используя строковое форматирование `f"{balance:.2f}"`, вы гарантируете, что баланс будет всегда отображаться с двумя десятичными знаками (например, `200.02`).
2. **Инлайн-кнопка**: Остальная часть кода в функции создает инлайн-кнопку так же, как и ранее, но теперь баланс выводится точно в формате, который вы ожидаете.
Это поможет избежать ошибок отображения и обеспечит единообразие в формате представления начисленного баланса для пользователей.
User: Добавь в профиль показывание того, сколько ты пригласил пользователей (Друзья).
from telebot.types import Message
from database import get_user_data
from telebot.types import InlineKeyboardMarkup, InlineKeyboardButton
from message import PROFILE_BUTTON_TEXT
def handle_profile(bot, message: Message):
user_id = message.from_user.id
simple_clicks, super_clicks, balance, _ = get_user_data(user_id)
profile_message = (
f"*🆔 ID:* `{user_id}`\n"
f"*💰 Баланс:* {balance:.2f} Ирис 🍬\n\n"
f"*🟢 Обычных кликов:* {simple_clicks}\n"
f"*🔴 Супер кликов:* {super_clicks}\n"
f"*👥 Друзей:* 0"
)
bot.send_message(
message.chat.id,
profile_message,
parse_mode='Markdown',
reply_markup=create_withdraw_button(balance)
)
def create_withdraw_button(balance):
markup = InlineKeyboardMarkup()
formatted_balance = f"{balance:.2f}"
withdraw_button = InlineKeyboardButton(f"📤 Вывести {formatted_balance} Ирис 🍬", callback_data='withdraw')
markup.add(withdraw_button)
return markup
def handle_withdraw(bot, call):
user_id = call.from_user.id
_, _, balance, _ = get_user_data(user_id)
if balance < 100000:
bot.answer_callback_query(call.id, "🚫 Минимальная сумма вывода: 100000 Ирис 🍬", show_alert=True)
else:
bot.answer_callback_query(call.id, "✅ Вы можете вывести средства.", show_alert=True)
Assistant: Чтобы добавить в профиль отображение количества приглашенных пользователей (друзей), вам нужно изменить функцию `get_user_data` в `database.py`, чтобы она также возвращала количество друзей. После этого вы сможете отобразить это значение в профиле. Вот как это можно сделать:
### Изменения в `database.py`
Обновите функцию `get_user_data`, чтобы она возвращала количество приглашенных друзей:
```python
def get_user_data(user_id):
conn = sqlite3.connect('clicker.db')
cursor = conn.cursor()
cursor.execute('SELECT simple_clicks, super_clicks, balance, last_super_click, invited_friends FROM users WHERE user_id = ?', (user_id,))
result = cursor.fetchone()
if not result:
cursor.execute('INSERT INTO users (user_id) VALUES (?)', (user_id,))
conn.commit()
return (0, 0, 0.0, None, 0) # И параллельно возвращаем количество друзей
else:
return result
```
### Изменения в вашем коде `profile_button.py`
Теперь вы можете обновить вашу функцию `handle_profile`, чтобы она отображала количество друзей в профиле:
```python
from telebot.types import Message
from database import get_user_data
from telebot.types import InlineKeyboardMarkup, InlineKeyboardButton
from message import PROFILE_BUTTON_TEXT
def handle_profile(bot, message: Message):
user_id = message.from_user.id
simple_clicks, super_clicks, balance, _, invited_friends = get_user_data(user_id) # Получаем количество друзей
profile_message = (
f"*🆔 ID:* `{user_id}`\n"
f"*💰 Баланс:* {balance:.2f} Ирис 🍬\n\n"
f"*🟢 Обычных кликов:* {simple_clicks}\n"
f"*🔴 Супер кликов:* {super_clicks}\n"
f"*👥 Друзей:* {invited_friends}" # Показываем количество друзей
)
bot.send_message(
message.chat.id,
profile_message,
parse_mode='Markdown',
reply_markup=create_withdraw_button(balance)
)
def create_withdraw_button(balance):
markup = InlineKeyboardMarkup()
formatted_balance = f"{balance:.2f}"
withdraw_button = InlineKeyboardButton(f"📤 Вывести {formatted_balance} Ирис 🍬", callback_data='withdraw')
markup.add(withdraw_button)
return markup
def handle_withdraw(bot, call):
user_id = call.from_user.id
_, _, balance, _, _ = get_user_data(user_id)
if balance < 100000:
bot.answer_callback_query(call.id, "🚫 Минимальная сумма вывода: 100000 Ирис 🍬", show_alert=True)
else:
bot.answer_callback_query(call.id, "✅ Вы можете вывести средства.", show_alert=True)
```
### Объяснение
1. **Изменения в `get_user_data`**: Теперь эта функция возвращает дополнительное поле — количество приглашенных друзей (`invited_friends`).
2. **Обновление профиля**: В профиле отображается новый параметр, который указывает, сколько друзей было приглашено пользователем.
Эти изменения позволят отображать пользователям информацию о количестве приглашенных друзей на их странице профиля.
User: cursor.execute('SELECT simple_clicks, super_clicks, balance, last_super_click, invited_friends FROM users WHERE user_id = ?', (user_id,))
sqlite3.OperationalError: no such column: invited_friends
Assistant: Ошибка `sqlite3.OperationalError: no such column: invited_friends` указывает на то, что в базе данных отсутствует столбец `invited_friends`. Это может произойти, если структура базы данных была изменена после её первоначального создания и новый столбец не был добавлен.
Для решения этой проблемы вам необходимо обновить структуру базы данных, добавив новый столбец. Это можно сделать с помощью команды SQL `ALTER TABLE`. Вам нужно добавить эту команду в вашу функцию инициализации базы данных `init_database`. Вот как это можно сделать:
### Изменения в `database.py`
Добавьте команду для изменения таблицы в функции `init_database`:
```python
def init_database():
conn = sqlite3.connect('clicker.db')
cursor = conn.cursor()
cursor.execute('''
CREATE TABLE IF NOT EXISTS users (
user_id INTEGER PRIMARY KEY,
simple_clicks INTEGER DEFAULT 0,
super_clicks INTEGER DEFAULT 0,
balance REAL DEFAULT 0.0,
last_super_click TEXT
)''')
# Проверяем и добавляем столбец `invited_friends`, если его нет
cursor.execute("PRAGMA table_info(users)")
columns = [column[1] for column in cursor.fetchall()]
if 'invited_friends' not in columns:
cursor.execute('ALTER TABLE users ADD COLUMN invited_friends INTEGER DEFAULT 0')
cursor.execute('''
CREATE TABLE IF NOT EXISTS referrals (
referrer_id INTEGER,
referee_id INTEGER,
UNIQUE(referrer_id, referee_id)
)''')
conn.commit()
conn.close()
```
### Объяснение
1. **Проверка наличия столбца**: SQL-запрос `PRAGMA table_info(users)` используется для получения информации о структуре таблицы `users`. Мы проверяем, содержится ли в этой информации столбец `invited_friends`.
2. **Добавление столбца**: Если столбец отсутствует, происходит его добавление командой `ALTER TABLE`.
После выполнения этих изменений ваша база данных будет включать новый столбец `invited_friends`, и ошибка должна быть устранена. Не забудьте, что для того чтобы изменения структуры таблицы вступили в силу, может потребоваться удалить существующую базу данных или перенести данные в новую структуру.
User: И как это значение будет увеличиваться?
Assistant: Чтобы поле `invited_friends` корректно увеличивалось, необходимо обеспечить его обновление каждый раз, когда регистрация нового реферала проходит успешно. Это достигается с помощью добавления соответствующего кода в функцию `add_referral` в вашем модуле `database.py`.
### Обновление функции `add_referral`
Здесь происходит добавление нового реферала и обновление счетчика приглашенных друзей:
```python
def add_referral(referrer_id, referee_id):
conn = sqlite3.connect('clicker.db')
cursor = conn.cursor()
try:
# Вставляем запись о реферале
cursor.execute('INSERT INTO referrals (referrer_id, referee_id) VALUES (?, ?)', (referrer_id, referee_id))
# Обновляем счетчик приглашенных друзей и баланс
cursor.execute('UPDATE users SET invited_friends = invited_friends + 1, balance = balance + 3.0 WHERE user_id = ?', (referrer_id,))
conn.commit()
except sqlite3.IntegrityError:
pass # Этот реферал уже существует
conn.close()
```
### Как это работает
1. **Добавление нового реферала**: Каждый раз, когда пользователь успешно приглашает нового друга (и тот регистрируется через реферальную ссылку), в таблице `referrals` создается новая запись, связывающая ID пригласившего с ID приглашенного.
2. **Обновление счетчика друзей**: Если новый реферал успешно добавлен, выполняется обновление в таблице `users`, увеличивая значение `invited_friends` на 1 и добавляя 3.0 к балансу пригласившего.
3. **Обработка ошибок**: Использование `sqlite3.IntegrityError` предотвращает повторное добавление одной и той же пары "пригласитель-приглашенный", если такая запись уже существует, обеспечивая уникальность каждой реферальной пары.
С данным кодом, когда пользователь приглашает друзей и они регистрируются через его ссылку, его количество приглашенных друзей будет автоматически увеличиваться.
User: ", line 8822, in _run_middlewares_and_handler
result = handler['function'](message)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\agrif\Desktop\Заказ\main.py", line 61, in send_earn_message
simple_clicks, super_clicks, balance, _ = get_user_data(user_id)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ValueError: too many values to unpack (expected 4)
Assistant: Ошибка `ValueError: too many values to unpack (expected 4)` указывает на то, что функция `get_user_data(user_id)` возвращает больше значений, чем ожидалось. Ранее, когда мы модифицировали функцию `get_user_data`, она начала возвращать 5 значений вместо 4. Это было сделано для добавления поля `invited_friends`.
Вам нужно обновить код, где вызывается `get_user_data`, так, чтобы он корректно принимал все возвращаемые значения. Исправьте строку, которая вызывает функцию, в соответствии с этой новой сигнатурой:
### Измените вашу функцию `send_earn_message` в `main.py` так:
```python
@bot.message_handler(func=lambda m: m.text == EARN_BUTTON_TEXT)
def send_earn_message(message):
user_id = message.from_user.id
# Разбиваем прием значений на 5 переменных
simple_clicks, super_clicks, balance, _, _ = get_user_data(user_id)
earn_message = (
f"За каждый простой клик вы получите: 0.005 Ирис 🍬 🟢\n"
f"За каждый супер клик вы получите: 0.2 Ирис 🍬 🔴\n\n"
f"*Всего простых кликов: {simple_clicks} 🟢*\n"
f"*Всего супер кликов: {super_clicks} 🔴*"
)
bot.send_message(
message.chat.id,
earn_message,
parse_mode='Markdown',
reply_markup=create_inline_buttons()
)
```
### Пояснение
- Функция `get_user_data(user_id)` теперь возвращает 5 значений: `simple_clicks, super_clicks, balance, last_super_click, invited_friends`.
- Чтобы обработать это, я заменил `_` на дополнительные `_`, игнорируя те значения, которые нам не нужны в текущей функции. Это позволяет вам удобно обрабатывать или игнорировать любые из возвращаемых значений.
Эта корректировка в вашем коде должна устранить `ValueError`, связанный с неправильным количеством ожидаемых значений.
User: При клике на простой и супер клик вылезает ошибка:
result = handler['function'](message)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\agrif\Desktop\Заказ\main.py", line 88, in callback_inline
simple_clicks, super_clicks, balance, last_super_click = get_user_data(user_id)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ValueError: too many values to unpack (expected 4)
Assistant: Эта ошибка возникает по той же причине, что и предыдущая: функция `get_user_data` теперь возвращает 5 значений, но вы пытаетесь распаковать только 4. Чтобы исправить это, вам необходимо изменить строку в функции `callback_inline`, чтобы она принимала все возвращаемые значения. Вот как это можно сделать:
### Изменение в `callback_inline`
Измените строку, где вызывается `get_user_data`, чтобы корректно принимать все 5 значений:
```python
@bot.callback_query_handler(func=lambda call: True)
def callback_inline(call):
user_id = call.from_user.id
# Разбиваем прием значений на 5 переменных, игнорируя `invited_friends`
simple_clicks, super_clicks, balance, last_super_click, _ = get_user_data(user_id)
if call.data == 'normal_click':
update_click_data(user_id, 'simple', 0.005)
bot.answer_callback_query(call.id, "Добавлено 0.005 Ирис")
elif call.data == 'super_click':
if super_clicks >= 5 and not can_super_click(last_super_click):
bot.answer_callback_query(call.id, MAX_SUPER_CLICKS_MESSAGE, show_alert=True)
else:
update_click_data(user_id, 'super', 0.2)
bot.answer_callback_query(call.id, "Добавлено 0.2 Ирис")
elif call.data == 'refresh':
simple_clicks, super_clicks, balance, _, _ = get_user_data(user_id)
new_earn_message = (
f"За каждый простой клик вы получите: 0.005 Ирис 🍬 🟢\n"
f"За каждый супер клик вы получите: 0.2 Ирис 🍬 🔴\n\n"
f"*Всего простых кликов: {simple_clicks} 🟢*\n"
f"*Всего супер кликов: {super_clicks} 🔴*"
)
try:
bot.edit_message_text(
chat_id=call.message.chat.id,
message_id=call.message.message_id,
text=new_earn_message,
parse_mode='Markdown',
reply_markup=create_inline_buttons()
)
except telebot.apihelper.ApiTelegramException as e:
if "message is not modified" in str(e):
pass
else:
raise
```
### Пояснение
1. **Исправление распаковки**: Теперь мы распаковываем все 5 значений, возвращаемых `get_user_data(user_id)`, и игнорируем `invited_friends`, так как оно не используется в этой функции.
2. **Корректное использование переменных**: Для поддержки новых изменений нужно убедиться, что объявленные переменные соответствуют данным, которые вам действительно нужны.
Эти изменения должны устранить ошибку `ValueError`, связанную с неправильным количеством ожидаемых значений при вызове `get_user_data`.
User: Сделай проверку, что если айди игрока 6545459678, то в профиле у него будет еще одна инлайн кнопка, а именно "🔴 Заблокировать пользователя". Если он нажмет на инлайн кнопку, то, бот напишет ему сообщение "🔴 | Напишите айди игрока, которого вы хотите заблокировать. Или напишите cancel для отмены." Дальше, если игрок напишет cancel, то мы отменяем блокировку, а если напишет айди игрока, который зарегестрирован в боте, то, мы пишем "🟢 | Успешная блокировка!". А тому игроку, которого заблокировали, мы напишем: "⚠️🔴| Вы были заблокированы администратором. Теперь вы не имеете доступа к боту." Под данным сообщением будет инлайн кнопка "🍬 | Администратор". Если нажать на кнопку, то откроеться профиль игрока с айди 6545459678. А, что же значит блокировка? Если игрок захочет написать какую либо команду, даже кнопку из меню, или инлайн кнопку, мы отменяем действие выводим ему табличку: "⚠️🔴| Вы были заблокированы администратором. Теперь вы не имеете доступа к боту."
Assistant:ASSISTANT
Чтобы реализовать функциональность блокировки, описанную вами, нам понадобится несколько изменений в вашем Telegram боте. Мы добавим функционал для администратора, чтобы он мог блокировать других пользователей бота. Так как вы хотите, чтобы ID `6545459678` имел особые привилегии, мы будем использовать этот ID для проверки и управления блокировкой.
### Изменения в `database.py`
Добавьте новый столбец `is_blocked` для таблицы `users`, чтобы отслеживать, заблокирован ли пользователь:
```python
def init_database():
conn = sqlite3.connect('clicker.db')
cursor = conn.cursor()
cursor.execute('''
CREATE TABLE IF NOT EXISTS users (
user_id INTEGER PRIMARY KEY,
simple_clicks INTEGER DEFAULT 0,
super_clicks INTEGER DEFAULT 0,
balance REAL DEFAULT 0.0,
last_super_click TEXT,
invited_friends INTEGER DEFAULT 0,
is_blocked INTEGER DEFAULT 0
)''')
conn.commit()
conn.close()
```
Добавьте функции для блокировки и проверки состояния блокировки:
```python
def block_user(user_id):
conn = sqlite3.connect('clicker.db')
cursor = conn.cursor()
cursor.execute('UPDATE users SET is_blocked = 1 WHERE user_id = ?', (user_id,))
conn.commit()
conn.close()
def is_user_blocked(user_id):
conn = sqlite3.connect('clicker.db')
cursor = conn.cursor()
cursor.execute('SELECT is_blocked FROM users WHERE user_id = ?', (user_id,))
result = cursor.fetchone()
conn.close()
return result is not None and result[0] == 1
```
### Изменения в `profile_button.py`
Добавьте создание инлайн кнопки для блокировки в профиле администратора:
```python
from telebot.types import InlineKeyboardMarkup, InlineKeyboardButton
ADMIN_ID = 6545459678
def create_profile_buttons(user_id, balance):
markup = InlineKeyboardMarkup()
formatted_balance = f"{balance:.2f}"
withdraw_button = InlineKeyboardButton(f"📤 Вывести {formatted_balance} Ирис 🍬", callback_data='withdraw')
markup.add(withdraw_button)
if user_id == ADMIN_ID:
block_button = InlineKeyboardButton("🔴 Заблокировать пользователя", callback_data='block_user')
markup.add(block_button)
return markup
```
Обновите обработчик профиля, чтобы использовать новый метод кнопок и полностью обрабатывать блокировку:
```python
def handle_profile(bot, message):
user_id = message.from_user.id
simple_clicks, super_clicks, balance, _, invited_friends = get_user_data(user_id)
profile_message = (
f"*🆔 ID:* `{user_id}`\n"
f"*💰 Баланс:* {balance:.2f} Ирис 🍬\n\n"
f"*🟢 Обычных кликов:* {simple_clicks}\n"
f"*🔴 Супер кликов:* {super_clicks}\n"
f"*👥 Друзей:* {invited_friends}"
)
bot.send_message(
message.chat.id,
profile_message,
parse_mode='Markdown',
reply_markup=create_profile_buttons(user_id, balance)
)
```
### Изменения в `main.py`
Внедрите логику блокировки пользователя:
```python
from database import block_user, is_user_blocked
@bot.message_handler(func=lambda message: is_user_blocked(message.from_user.id))
def block_warning(message):
bot.send_message(message.chat.id, "⚠️🔴| Вы были заблокированы администратором. Теперь вы не имеете доступа к боту.")
@bot.callback_query_handler(func=lambda call: call.data == 'block_user')
def callback_block_user(call):
if call.from_user.id == ADMIN_ID:
bot.send_message(call.message.chat.id, "🔴 | Напишите айди игрока, которого вы хотите заблокировать. Или напишите cancel для отмены.")
bot.register_next_step_handler(call.message, process_block_user)
def process_block_user(message):
if message.text.lower() == 'cancel':
bot.send_message(message.chat.id, "✅ Блокировка отменена.")
return
try:
user_id = int(message.text)
if get_user_data(user_id):
block_user(user_id)
bot.send_message(message.chat.id, "🟢 | Успешная блокировка!")
bot.send_message(user_id, "⚠️🔴| Вы были заблокированы администратором. Теперь вы не имеете доступа к боту.", reply_markup=create_blocked_user_button())
else:
bot.send_message(message.chat.id, "🚫 Пользователь с таким ID не найден.")
except ValueError:
bot.send_message(message.chat.id, "🚫 Неправильный ID. Попробуйте ещё раз.")
bot.register_next_step_handler(message, process_block_user)
def create_blocked_user_button():
markup = InlineKeyboardMarkup()
admin_button = InlineKeyboardButton("🍬 | Администратор", callback_data='show_admin_profile')
markup.add(admin_button)
return markup
@bot.callback_query_handler(func=lambda call: call.data == 'show_admin_profile')
def callback_show_admin_profile(call):
user_id = ADMIN_ID
simple_clicks, super_clicks, balance, _, invited_friends = get_user_data(user_id)
profile_message = (
f"*🆔 ID:* `{user_id}`\n"
f"*💰 Баланс:* {balance:.2f} Ирис 🍬\n\n"
f"*🟢 Обычных кликов:* {simple_clicks}\n"
f"*🔴 Супер кликов:* {super_clicks}\n"
f"*👥 Друзей:* {invited_friends}"
)
bot.send_message(
call.message.chat.id,
profile_message,
parse_mode='Markdown',
reply_markup=create_profile_buttons(user_id, balance)
)
```
### Объяснение
1. **Проверка на блокировку**: `is_user_blocked` функция проверяет, заблокирован ли пользователь. Если пользователь заблокирован, они не смогут взаимодействовать с ботом и получат предупреждение.
2. **Создание кнопки блокировки**: Добавляем кнопку блокировки в профиле администратора.
3. **Обработка блокировки**: Администратор может вводить ID игрока для блокировки. Если вместо этого введено `cancel`, процесс блокировки отменяется.
4. **Создание кнопки "Администратор"**: Заблокированный пользователь после блокировки получает кнопку, которая показывает профиль администратора.
Эти изменения позволят администратору (ID 6545459678) управлять доступом пользователей к боту. Заблокированные пользователи будут получать уведомления и не смогут взаимодействовать с ботом.