| #!/usr/bin/env python |
| |
| """ |
| Synopsis: %(prog)s [-h|-b|-r|-a|-d] dbfile [ picklefile ] |
| |
| Read the given picklefile as a series of key/value pairs and write to a new |
| database. If the database already exists, any contents are deleted. The |
| optional flags indicate the type of the database (bsddb hash, bsddb btree, |
| bsddb recno, anydbm, dbm). The default is hash. If a pickle file is named |
| it is opened for read access. If no pickle file is named, the pickle input |
| is read from standard input. |
| |
| Note that recno databases can only contain numeric keys, so you can't dump a |
| hash or btree database using db2pickle.py and reconstitute it to a recno |
| database with %(prog)s. |
| """ |
| |
| import getopt |
| try: |
| import bsddb |
| except ImportError: |
| bsddb = None |
| try: |
| import dbm |
| except ImportError: |
| dbm = None |
| try: |
| import anydbm |
| except ImportError: |
| anydbm = None |
| import sys |
| try: |
| import cPickle as pickle |
| except ImportError: |
| import pickle |
| |
| prog = sys.argv[0] |
| |
| def usage(): |
| sys.stderr.write(__doc__ % globals()) |
| |
| def main(args): |
| try: |
| opts, args = getopt.getopt(args, "hbrda", |
| ["hash", "btree", "recno", "dbm", "anydbm"]) |
| except getopt.error: |
| usage() |
| return 1 |
| |
| if len(args) == 0 or len(args) > 2: |
| usage() |
| return 1 |
| elif len(args) == 1: |
| dbfile = args[0] |
| pfile = sys.stdin |
| else: |
| dbfile = args[0] |
| try: |
| pfile = open(args[1], 'rb') |
| except IOError: |
| sys.stderr.write("Unable to open %s\n" % args[1]) |
| return 1 |
| |
| dbopen = None |
| for opt, arg in opts: |
| if opt in ("-h", "--hash"): |
| try: |
| dbopen = bsddb.hashopen |
| except AttributeError: |
| sys.stderr.write("bsddb module unavailable.\n") |
| return 1 |
| elif opt in ("-b", "--btree"): |
| try: |
| dbopen = bsddb.btopen |
| except AttributeError: |
| sys.stderr.write("bsddb module unavailable.\n") |
| return 1 |
| elif opt in ("-r", "--recno"): |
| try: |
| dbopen = bsddb.rnopen |
| except AttributeError: |
| sys.stderr.write("bsddb module unavailable.\n") |
| return 1 |
| elif opt in ("-a", "--anydbm"): |
| try: |
| dbopen = anydbm.open |
| except AttributeError: |
| sys.stderr.write("anydbm module unavailable.\n") |
| return 1 |
| elif opt in ("-d", "--dbm"): |
| try: |
| dbopen = dbm.open |
| except AttributeError: |
| sys.stderr.write("dbm module unavailable.\n") |
| return 1 |
| if dbopen is None: |
| if bsddb is None: |
| sys.stderr.write("bsddb module unavailable - ") |
| sys.stderr.write("must specify dbtype.\n") |
| return 1 |
| else: |
| dbopen = bsddb.hashopen |
| |
| try: |
| db = dbopen(dbfile, 'c') |
| except bsddb.error: |
| sys.stderr.write("Unable to open %s. " % dbfile) |
| sys.stderr.write("Check for format or version mismatch.\n") |
| return 1 |
| else: |
| for k in db.keys(): |
| del db[k] |
| |
| while 1: |
| try: |
| (key, val) = pickle.load(pfile) |
| except EOFError: |
| break |
| db[key] = val |
| |
| db.close() |
| pfile.close() |
| |
| return 0 |
| |
| if __name__ == "__main__": |
| sys.exit(main(sys.argv[1:])) |