/Tools/scripts/db2pickle.py

  1#!/usr/bin/env python
  2
  3"""
  4Synopsis: %(prog)s [-h|-g|-b|-r|-a] dbfile [ picklefile ]
  5
  6Convert the database file given on the command line to a pickle
  7representation.  The optional flags indicate the type of the database:
  8
  9    -a - open using anydbm
 10    -b - open as bsddb btree file
 11    -d - open as dbm file
 12    -g - open as gdbm file
 13    -h - open as bsddb hash file
 14    -r - open as bsddb recno file
 15
 16The default is hash.  If a pickle file is named it is opened for write
 17access (deleting any existing data).  If no pickle file is named, the pickle
 18output is written to standard output.
 19
 20"""
 21
 22import getopt
 23try:
 24    import bsddb
 25except ImportError:
 26    bsddb = None
 27try:
 28    import dbm
 29except ImportError:
 30    dbm = None
 31try:
 32    import gdbm
 33except ImportError:
 34    gdbm = None
 35try:
 36    import anydbm
 37except ImportError:
 38    anydbm = None
 39import sys
 40try:
 41    import cPickle as pickle
 42except ImportError:
 43    import pickle
 44
 45prog = sys.argv[0]
 46
 47def usage():
 48    sys.stderr.write(__doc__ % globals())
 49
 50def main(args):
 51    try:
 52        opts, args = getopt.getopt(args, "hbrdag",
 53                                   ["hash", "btree", "recno", "dbm",
 54                                    "gdbm", "anydbm"])
 55    except getopt.error:
 56        usage()
 57        return 1
 58
 59    if len(args) == 0 or len(args) > 2:
 60        usage()
 61        return 1
 62    elif len(args) == 1:
 63        dbfile = args[0]
 64        pfile = sys.stdout
 65    else:
 66        dbfile = args[0]
 67        try:
 68            pfile = open(args[1], 'wb')
 69        except IOError:
 70            sys.stderr.write("Unable to open %s\n" % args[1])
 71            return 1
 72
 73    dbopen = None
 74    for opt, arg in opts:
 75        if opt in ("-h", "--hash"):
 76            try:
 77                dbopen = bsddb.hashopen
 78            except AttributeError:
 79                sys.stderr.write("bsddb module unavailable.\n")
 80                return 1
 81        elif opt in ("-b", "--btree"):
 82            try:
 83                dbopen = bsddb.btopen
 84            except AttributeError:
 85                sys.stderr.write("bsddb module unavailable.\n")
 86                return 1
 87        elif opt in ("-r", "--recno"):
 88            try:
 89                dbopen = bsddb.rnopen
 90            except AttributeError:
 91                sys.stderr.write("bsddb module unavailable.\n")
 92                return 1
 93        elif opt in ("-a", "--anydbm"):
 94            try:
 95                dbopen = anydbm.open
 96            except AttributeError:
 97                sys.stderr.write("anydbm module unavailable.\n")
 98                return 1
 99        elif opt in ("-g", "--gdbm"):
100            try:
101                dbopen = gdbm.open
102            except AttributeError:
103                sys.stderr.write("gdbm module unavailable.\n")
104                return 1
105        elif opt in ("-d", "--dbm"):
106            try:
107                dbopen = dbm.open
108            except AttributeError:
109                sys.stderr.write("dbm module unavailable.\n")
110                return 1
111    if dbopen is None:
112        if bsddb is None:
113            sys.stderr.write("bsddb module unavailable - ")
114            sys.stderr.write("must specify dbtype.\n")
115            return 1
116        else:
117            dbopen = bsddb.hashopen
118
119    try:
120        db = dbopen(dbfile, 'r')
121    except bsddb.error:
122        sys.stderr.write("Unable to open %s.  " % dbfile)
123        sys.stderr.write("Check for format or version mismatch.\n")
124        return 1
125
126    for k in db.keys():
127        pickle.dump((k, db[k]), pfile, 1==1)
128
129    db.close()
130    pfile.close()
131
132    return 0
133
134if __name__ == "__main__":
135    sys.exit(main(sys.argv[1:]))