Пожалуйста сделайте 4 задание. Python

Детский вариант:

n = int(input())

if n % 7 == 0:

       print('+'*n)

   else:

       print('-'*n)

Красивый вариант:

def wrighter(n):

   if n % 7 == 0:

       print('+'*n)

   else:

       print('-'*n)

def main():

   print('Введите число:', end = '')

   n = int(input())

   wrighter(n)

main()