python讀取一個文件的每一行判斷是否為素數，並把結果寫到另一個文件中

剛剛學習python的菜鳥，這道題包括：文件的讀寫，python的參數調用，異常的使用，函數的使用

創建一個文本文件inti_prime.txt

執行命令：python Prime.py init_prime.txt result_prime.txt

會生成一個result_prime.txt文件

#-*- coding:UTF-8 -*-
#讀取一個文件的每一行，每一行為一個數字
#判斷數字是不是素數
#並打印結果到另外一個文件
#輸入文件名和輸出文件名用參數的形式
import sys
import math
#輸入文件
str1=sys.argv[1]
#輸出文件
str2=sys.argv[2]
#打開文件
try:
    file_object=open(str1,"r+")
except IOError:
    print "找不到這個文件".decode('utf8').encode('gbk')
else:
    file_result=open(str2,"w+")
#判斷是否為素數
    def IsPrime(number):
        if number<2:
            return False
        else:
            for i in range(2,int(math.ceil(math.sqrt(number))+1)):
                if number==2:
                    return True
                else:
                    if number%i==0:
                        return False
                    else:
                        continue
            return True
#讀取文件
    for line in file_object:
        line = line.strip()
        if IsPrime(int(line)):
            file_result.write(line+" is a Prime"+"\n")
        else:
            continue
    file_object.close()
    file_result.close()