编写了一个collatz函数

def collatz(num):     if num % 2 == 0:         print(num // 2)     else:         print(3 * num + 1)         

后面怎么编写，可以让用户输入一个整数，并不断的对这个数进行调用collatz(),直到返回值1.
输出结果大概是这样的：

Enter number: 3 10 5 16 8 4 2 1

我自己觉得应该是用while循环，但是编写的总是错误，不知道该怎么写，求教。