Sum of digits

Write digit_sum(n) that returns the sum of the digits of a non-negative whole number.

digit_sum(1234)  ->  10
digit_sum(0)     ->  0

def digit_sum(n):
    ...