Comment prendre le prochain message de l'utilisateur dans aiogram

Je ferais comme ceci :

def prime_f(n):    if n%2 == 0: return 2    i = 3    while n%i != 0 and i*i <= n:        i+= 2    if i*i <= n: return i    return nN = int(input())print(prime_f(N))

On vérifie 2 séparément, puis seulement les nombres impairs, et jusqu'à la racine de N - sinon N est lui-même premier.