Нашёл в хабр ответ, но с кодом что то не так, во первых в ответе написано что нужно создать базу, как это сделать?
Заработал только команда /ref и всё
Код который нашёл:
библиотека pyTelegramBotApi
Файл main.py:
from models import *
from telebot import TeleBot
bot = TeleBot('TOKEN')
ref_link = 'https://telegram.me/{}?start={}'
@bot.message_handler(commands=['start'])
def start(message):
user_id = message.chat.id
splited = message.text.split()
if not Users.user_exists(user_id):
Users.create_user(user_id)
if len(splited) == 2:
Users.increase_ref_count(splited[1])
bot.reply_to(message, text='hello')
@bot.message_handler(commands=['ref'])
def get_my_ref(message):
bot_name = bot.get_me().username
bot.reply_to(message, text=ref_link.format(bot_name, message.chat.id))
@bot.message_handler(commands=['ref_count'])
def get_my_refs(message):
count = Users.get_ref_count(message.chat.id)
bot.reply_to(message, text=f'Count: {count}')
if __name__ == '__main__':
bot.polling(none_stop=True)
Файл models.py:
from peewee import *
db = SqliteDatabase('users.db')
class BaseModel(Model):
class Meta:
database = db
class Users(BaseModel):
user_id = IntegerField(unique=True)
ref = IntegerField(default=0)
@classmethod
def get_user(cls, user_id):
return cls.get(user_id == user_id)
@classmethod
def get_ref_count(cls, user_id):
return cls.get(Users.user_id == user_id).ref
@classmethod
def increase_ref_count(cls, user_id):
user = cls.get_user(user_id)
user.ref += 1
user.save()
@classmethod
def user_exists(cls, user_id):
query = cls().select().where(cls.user_id == user_id)
return query.exists()
@classmethod
def create_user(cls, user_id):
user, created = cls.get_or_create(user_id=user_id)
#db.create_tables([Users]) #Создание таблицы
