fix a bug of file leak

This commit is contained in:
Sun Junyi 2013-05-02 11:24:22 +08:00
parent 1275b3679f
commit ff4ea5d882

View File

@ -81,7 +81,8 @@ def initialize(*args):
min_freq = min(FREQ.itervalues())
print >> sys.stderr, "dumping model to file cache " + cache_file
tmp_suffix = "."+str(random.random())
marshal.dump((trie,FREQ,total,min_freq),open(cache_file+tmp_suffix,'wb'))
with open(cache_file+tmp_suffix,'wb') as temp_cache_file:
marshal.dump((trie,FREQ,total,min_freq),temp_cache_file)
if os.name=='nt':
import shutil
replace_file = shutil.move