Árvore de natal

import sys

def desenhar_arvore(n, msg="Feliz Natal!!!"):
    msg = ' '*(n-6) + msg
    z = n - 1
    x, t = 1, 0
    print(msg)
    for i in range(0,n):
        for i in range(0,z):
            print(' ',end='')
        for i in range(0,x):
            print('+',end='')
        x=x+2
        z=z-1
        t = x
        print()
    t = t // 3
    for x in range(t):
        if n < 5:
            for x in range(t-1):
                print(' ', end='')
        else:
            for x in range(t):
                print(' ', end='')
        for x in range(t//2+2):
            print('|', end='')
        print()
if len(sys.argv) > 1:
    if sys.argv[1] == '-n':
        desenhar_arvore(int(sys.argv[2]))
else:
    n = int(input("Digite o total de ramos: "))
    desenhar_arvore(n)