db2pickle.py

#!/usr/bin/env python

"""
Synopsis: %(prog)s [-h|-g|-b|-r|-a] dbfile [ picklefile ]

Convert the database file given on the command line to a pickle
representation. The optional flags indicate the type of the database:

 -a - open using anydbm
 -b - open as bsddb btree file
 -d - open as dbm file
 -g - open as gdbm file
 -h - open as bsddb hash file
 -r - open as bsddb recno file

The default is hash. If a pickle file is named it is opened for write
access (deleting any existing data). If no pickle file is named, the pickle
output is written to standard output.

"""

importgetopt
try:
importbsddb
exceptImportError:
bsddb=None
try:
importdbm
exceptImportError:
dbm=None
try:
importgdbm
exceptImportError:
gdbm=None
try:
importanydbm
exceptImportError:
anydbm=None
importsys
try:
importcPickleaspickle
exceptImportError:
importpickle

prog=sys.argv[0]

defusage():
sys.stderr.write(__doc__%globals())

defmain(args):
try:
opts, args=getopt.getopt(args, "hbrdag",
 ["hash", "btree", "recno", "dbm",
"gdbm", "anydbm"])
exceptgetopt.error:
usage()
return1

iflen(args) ==0orlen(args) >2:
usage()
return1
eliflen(args) ==1:
dbfile=args[0]
pfile=sys.stdout
else:
dbfile=args[0]
try:
pfile=open(args[1], 'wb')
exceptIOError:
sys.stderr.write("Unable to open %s\n"%args[1])
return1

dbopen=None
foropt, arginopts:
ifoptin ("-h", "--hash"):
try:
dbopen=bsddb.hashopen
exceptAttributeError:
sys.stderr.write("bsddb module unavailable.\n")
return1
elifoptin ("-b", "--btree"):
try:
dbopen=bsddb.btopen
exceptAttributeError:
sys.stderr.write("bsddb module unavailable.\n")
return1
elifoptin ("-r", "--recno"):
try:
dbopen=bsddb.rnopen
exceptAttributeError:
sys.stderr.write("bsddb module unavailable.\n")
return1
elifoptin ("-a", "--anydbm"):
try:
dbopen=anydbm.open
exceptAttributeError:
sys.stderr.write("anydbm module unavailable.\n")
return1
elifoptin ("-g", "--gdbm"):
try:
dbopen=gdbm.open
exceptAttributeError:
sys.stderr.write("gdbm module unavailable.\n")
return1
elifoptin ("-d", "--dbm"):
try:
dbopen=dbm.open
exceptAttributeError:
sys.stderr.write("dbm module unavailable.\n")
return1
ifdbopenisNone:
ifbsddbisNone:
sys.stderr.write("bsddb module unavailable - ")
sys.stderr.write("must specify dbtype.\n")
return1
else:
dbopen=bsddb.hashopen

try:
db=dbopen(dbfile, 'r')
exceptbsddb.error:
sys.stderr.write("Unable to open %s. "%dbfile)
sys.stderr.write("Check for format or version mismatch.\n")
return1

forkindb.keys():
pickle.dump((k, db[k]), pfile, 1==1)

db.close()
pfile.close()

return0

if__name__=="__main__":
sys.exit(main(sys.argv[1:]))