python輸入整數

#!/usr/bin/env python
#ecoding=utf-8
'''
Created on 2017年11月2日

@author: James zhan
'''

def fun(n):
if n==1 or n==0:
return 1
else:
return fun(n-1)*n
s = input('Input a init:')
print fun(s)