From: ♡༺ErVinTract༻♡

main.py

def extract_unique_code(text):
    # Extracts the unique_code from the sent /start command.
    return text.split()[1] if len(text.split()) > 1 else None

create_tables()

@dp.message_handler(commands=["start"], state="*")
async def user_registration(message: types.Message):
    register_user(message.from_user.id, message.from_user.username, message.from_user.first_name, message.from_user.last_name)
    #Выделение id рефера
    unique_code = extract_unique_code(message.text)
    if unique_code:
        #если есть id рефера
        check_stat = check_status(message.from_user.id)
        check_stat = float(check_stat)
        if not check_stat >= 1:
            refferals(unique_code)
            status(message.from_user.id)
            register_user(message.from_user.id, message.from_user.username, message.from_user.first_name, message.from_user.last_name)
            start_messages = text(
                f"Привествую {message.from_user.first_name} ! \n",
                sep="\n"
            )
            await bot.send_message(message.from_user.id, start_messages, reply_markup=kb.Menu)
        else:
            register_user(message.from_user.id, message.from_user.username,
                  message.from_user.first_name, message.from_user.last_name)
            start_message = text(
                f'Привествую {message.from_user.first_name} ! \n',
                sep="\n"
            )
            await bot.send_message(message.from_user.id, start_message, reply_markup=kb.Menu)

    else:
        #если в ссылке нет id рефера
        status(message.from_user.id)
        register_user(message.from_user.id, message.from_user.username,
                  message.from_user.first_name, message.from_user.last_name)
        start_message = text(
            f'Привествую {message.from_user.first_name} ! \n',
            sep="\n"
        )
        await bot.send_message(message.from_user.id, start_message, reply_markup=kb.Menu)