Цифровой корень

Помогите найти промах в коде 
Задача :
 
Цифровой корень натурального числа получается следующим образом. Складываются все цифры данного числа. Процесс повторяется, пока в результате не будет получено однозначное число, которое и называется цифровым корнем числа.
 
Пример :
156
Должно вывести 3 т.к. 1 + 5 + 6 = 12, 1 + 2 = 3
Выводит : 1
 
 
 
 
 
n = int(input())
if n // 10 == 0 :
    print(n)
else :
    b = 10
    f = 0
    a = n % 10
    n -= a
    n //= 10
    b = a + b
    while n // 10 != 0 :
        a = n % 10
        n -= a
        n //= 10
        b = a + b
        f += 1
if f == 1 :
     res = b + n - 10
elif f == 0 :
     res = n
else :
    res = b - 10
while res // 10 != 0 :
    b = 10
    f = 0
    a = res % 10
    res -= a
    res //= 10
    b = a + b
print(res)

n = int (input())
while n > 9:
    i = int (n%10)
    n = n//10
    n = n + i
print(n)

n = int(input())
 
s = str(n)
while len(s) > 1:
    n = 0;
    for i in s:
        n += int(i)
    s = str(n)
print(s)

a=356
def f(x):
    if len(str(x))==1:
        return(x)
    else:
        return f(sum([int(i) for i in str(x)]))
print(f(a))