作为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"