整数各位数字求和

【问题描述】
编写函数sum(x)，求整数x的各位数字之和。
在主程序中调用该函数：从键盘输入一非负整数，然后调用sum函数计算各位数字之和并输出结果。
【输入形式】
输入一个正整数。
【输出形式】

输出该整数各位数字之和。
【样例输入】

58

【样例输出】

13

【样例说明】

输入整数58，其各位数字之和为：5+8 = 13。

def sum(n):
    m=str(n)
    sum=0
    for i in range (0,len(m)):
        sum+=int(m[i])
    return sum
a=int(input())
print(sum(a))