编写一段Python代码来判断一个数是否为质数。

请输入一个整数：17
```python
num = 17

if num > 1:
    for i in range(2, num):
        if (num % i) == 0:
            print(num, "不是质数")
            break
    else:
        print(num, "是质数")
else:
    print(num, "不是质数")
```
输出：

```
17 是质数
```