Python p2 Sum of Digits in a String

2. Sum of Digits in a String

Write a program that asks the user to enter a series of single-digit numbers with nothing

separating them. The program should display the sum of all the single digit numbers in the

string. For example, if the user enters 2514, the method should return 12, which is the sum

of 2, 5, 1, and 4.

This is what I have but does not work..help

def calculateSum():

data = input('Enter the single number to get sum: ')

dataList = list(str(data))

digitsSum = 0

for item in dataList:

digitsSum += int(item)

return(digitsSum)

print(calculateSum())