add_digits.py

class Solution:
    def addDigits(self, num: int) -> int:
        
        if num == 0:
            return 0
        
        if num % 9 == 0:
            return 9
        
        return num % 9