作爲python腳本關閉執行一個動作

Question

我想知道是否有可能在任何給定的點在一個基本的Python腳本中執行一個動作，所以說它什麼時候關閉。我有以下代碼來尋找素數（只是爲了好玩）

number = 1 
primelist = [] 
nonprime = [] 

while number < 1000: 

    number += 1 

    for i in range(number): 
     if i != 1 and i != number and i !=0: 
      if number%i == 0: 
       nonprime.append(number) 
      else: 
       primelist.append(number) 

nonprimes = open("nonprimes.txt", "w") 

for nonprime in set(primelist) & set(nonprime): 
    nonprimes.write(str(nonprime) + ", ") 

nonprimes.close() 

所以基本上我想運行最後一部分，因爲腳本停止。如果這是不可能的，有什麼方法可以讓我在程序運行時按「空格」，然後保存列表？

乾杯提前:)

編輯：

我已經修改了代碼，包括atexit對模塊的建議，但它似乎沒有奏效。那就是：

導入時，atexit對

class primes(): 



    def __init__(self): 

     self.work(1) 

    def work(self, number): 

     number = 1 
     self.primelist = [] 
     self.nonprime = [] 

     while number < 20: 
      time.sleep(0.1) 
      print "Done" 

      number += 1 

      for i in range(number): 
       if i != 1 and i != number and i !=0: 
        if number%i == 0: 
         self.nonprime.append(number) 
        else: 
         self.primelist.append(number) 

     nonprimes = open("nonprimes.txt", "w") 

     for nonprime in set(self.primelist) & set(self.nonprime): 
      nonprimes.write(str(nonprime) + ", ") 

     nonprimes.close() 


    def exiting(self, primelist, nonprimelist): 
     primelist = self.primelist 
     nonprimelist = self.nonprime 

     nonprimes = open("nonprimes.txt", "w") 

     for nonprime in set(self.primelist) & set(self.nonprime): 
      nonprimes.write(str(nonprime) + ", ") 

     nonprimes.close() 

atexit.register(exiting) 


if __name__ == "__main__": 
    primes() 

Answer 1

雖然我敢肯定的文件對象不清理和沖洗的東西，當它被回收到文件。最好的方法是使用with語句。

with open("nonprimes.txt", "w") as nonprimes: 
    for nonprime in set(primelist) & set(nonprime): 
     nonprimes.write(str(nonprime) + ", ") 

當聲明結束時，關閉文件等的鍋爐代碼自動執行。

Answer 2

Python有一個atexit對模塊，允許你註冊代碼，當一個腳本退出你想執行：

import atexit, sys 

def doSomethingAtExit(): 
    print "Doing something on exit" 

atexit.register(doSomethingAtExit) 

if __name__ == "__main__": 
    sys.exit(1) 
    print "This won't get called"