上面的代码帮我重写一下

以下是重写后的代码：

def is_prime(n):
    """
    判断一个数是否为质数
    """
    if n < 2:
        return False
    for i in range(2, int(n ** 0.5) + 1):
        if n % i == 0:
            return False
    return True

def get_primes(n):
    """
    返回小于等于n的所有质数
    """
    primes = []
    for i in range(2, n + 1):
        if is_prime(i):
            primes.append(i)
    return primes

if __name__ == '__main__':
    n = int(input('请输入一个正整数：'))
    primes = get_primes(n)
    print('小于等于{}的所有质数为：{}'.format(n, primes))

重写后的代码更加简洁，使用了函数来实现判断质数和获取质数列表的功能，提高了代码的可读性和可维护性。同时，加入了输入提示，使代码更加友好。