Sistema de referência em bots do Telegram

Olá a todos! Certamente você já viu em vários bots um link de referência como https://t.me/ <username_bot>? Start = <number>. Normalmente, o ID do telegrama do referenciador é indicado como um número. Neste artigo, explicarei como lidar com esses links em meu bot.

Para desenvolver bots, uso a biblioteca pytelegrambotapi Python.

Vou omitir o processo de criação de um bot via @BotFather , vou prosseguir diretamente para o código. Existe um arquivo config.py



com a variável TOKEN, que armazena o token do bot. bot.py



Vamos "quimizar" o arquivo . Vamos escrever um manipulador para o comando / start e exibir tudo o que está armazenado no objeto de mensagem.

import telebot

import config

bot = telebot.TeleBot(config.TOKEN)


@bot.message_handler(commands=["start"])
def start_command_handler(msg):
    print(msg)


if __name__ == '__main__':
    bot.polling(none_stop=True)

      
      

        
        
        
      

    
        
        
        
      
      

        
        
        
      

    
    

'text': '/start'.



https://t.me/<_>?start=test 10- print(msg.text)

/start test



. , msg.text



, .

. , /start , /start



. ( /start test



). ID .

@bot.message_handler(commands=["start"])
def start_command_handler(msg):
    user_id = msg.from_user.id
    referrer = None

    if " " in msg.text:
        referrer_candidate = msg.text.split()[1]
				#  
      
      

        
        
        
      

    
        
        
        
      
      

        
        
        
      

    
    

https://t.me/<_>?start=test test2, test2 msg.text, , , test2 . msg.text



. , . (msg.text.split()



), ( , , ?).

, . , , . - "" , ID - . .

@bot.message_handler(commands=["start"])
def start_command_handler(msg):
    user_id = msg.from_user.id
    referrer = None

    #    -    
    if " " in msg.text:
        referrer_candidate = msg.text.split()[1]

        #     
        try:
            referrer_candidate = int(referrer_candidate)
						#  
        except ValueError:
            pass
      
      

        
        
        
      

    
        
        
        
      
      

        
        
        
      

    
    

, try... except



int()



. referer



, int()



ValueError

referer . : int()



"-101" -101. referer . , .

TG ID. , . . ID , user_id = msg.from_user.id

@bot.message_handler(commands=["start"])
def start_command_handler(msg):
    user_id = msg.from_user.id
    referrer = None

    #    -    
    if " " in msg.text:
        referrer_candidate = msg.text.split()[1]

        #     
        try:
            referrer_candidate = int(referrer_candidate)

            #    TG ID  TG ID 
            if user_id != referrer_candidate:
              #  
							pass

        except ValueError:
            pass
      
      

        
        
        
      

    
        
        
        
      
      

        
        
        
      

    
    

, get_all_users(), . , , .

@bot.message_handler(commands=["start"])
def start_command_handler(msg):
    user_id = msg.from_user.id
    referrer = None

    #    -    
    if " " in msg.text:
        referrer_candidate = msg.text.split()[1]

        #     
        try:
            referrer_candidate = int(referrer_candidate)

            #    TG ID  TG ID 
            #  ,       
            if user_id != referrer_candidate and referrer_candidate in get_all_users():
                referer = referrer_candidate

        except ValueError:
            pass
      
      

        
        
        
      

    
        
        
        
      
      

        
        
        
      

    
    

O que deve ser feito se o usuário já tiver um referenciador? Afinal, não daremos bônus pela indicação indicada cada vez que você usar um link de indicação, isso simplesmente nos arruinará. Portanto, é necessário fazer um cheque. Vamos escrever uma função has_referrer()



que retornará True se o usuário tiver um referenciador e False se não tiver. É lógico escrevê-lo no início para otimização.

@bot.message_handler(commands=["start"])
def start_command_handler(msg):
    user_id = msg.from_user.id
    #      
    if not has_referrer():
        referrer = None

        #    -    
        if " " in msg.text:
            referrer_candidate = msg.text.split()[1]

            #     
            try:
                referrer_candidate = int(referrer_candidate)

                #    TG ID  TG ID 
                #  ,       
                if user_id != referrer_candidate and referrer_candidate in get_all_users():
                    referer = referrer_candidate

            except ValueError:
                pass
      
      

        
        
        
      

    
        
        
        
      
      

        
        
        
      

    
    

Espero que você ache este artigo útil!