파이썬 약수, 소수 코드

# 약수 구하기
inputNumber = int(input('0보다 큰 정수 입력: '))

divisor = []
for number in range(1, (inputNumber + 1)):
    if inputNumber % number == 0:
        divisor.append(number)

print(divisor)

# 소수 구하기

inputNumber = int(input('0보다 큰 정수 입력: '))

prime_number = []
for number in range(2, (inputNumber + 1)):
    flag = True
    for n in range(2, number):
        if number % n  == 0:
            flag = False
            break

    if flag:
        prime_number.append(number)

print(prime_number)