Python: Print a diamond shape with as many lines as the input number.

print_diamond.py

# For example, if the input is 5, the shape should look like this:
#   *
#  ***
# *****
#  ***
#   *
import math

lines = int(input('How many lines? '))

half = math.floor(lines / 2)

for current_line in range(lines):
    distance_from_half = abs(half - current_line)
    padding = distance_from_half
    stars = lines - padding * 2
    print(' ' * padding + '*' * stars)