Fix up the cleanup of the temporary DB so it works for BSD DB's

compatibility layer as well as "classic" ndbm.
This commit is contained in:
Fred Drake 2000-09-18 17:56:58 +00:00
parent a12adfe485
commit 57a4e90922
1 changed files with 12 additions and 4 deletions

View File

@ -6,7 +6,7 @@ import dbm
from dbm import error
from test_support import verbose
filename= '/tmp/delete_me'
filename = '/tmp/delete_me'
d = dbm.open(filename, 'c')
d['a'] = 'b'
@ -28,7 +28,15 @@ d.close()
try:
import os
if dbm.library == "ndbm":
# classic dbm
os.unlink(filename + '.dir')
os.unlink(filename + '.pag')
elif dbm.library == "BSD db":
# BSD DB's compatibility layer
os.unlink(filename + '.db')
else:
# GNU gdbm compatibility layer
os.unlink(filename)
except:
pass