Write a Python function that calculates the factorial of a number.
==============
def factorial(n):
    if n == 0:
        return 1
    return n * factorial(n - 1)