Python code coverage for Tools/scripts/pickle2db.py

#	count	content
1	n/a	#!/usr/bin/env python3
2	n/a
3	n/a	"""
4	n/a	Synopsis: %(prog)s [-h\|-b\|-g\|-r\|-a\|-d] [ picklefile ] dbfile
5	n/a
6	n/a	Read the given picklefile as a series of key/value pairs and write to a new
7	n/a	database. If the database already exists, any contents are deleted. The
8	n/a	optional flags indicate the type of the output database:
9	n/a
10	n/a	-a - open using dbm (open any supported format)
11	n/a	-b - open as bsddb btree file
12	n/a	-d - open as dbm.ndbm file
13	n/a	-g - open as dbm.gnu file
14	n/a	-h - open as bsddb hash file
15	n/a	-r - open as bsddb recno file
16	n/a
17	n/a	The default is hash. If a pickle file is named it is opened for read
18	n/a	access. If no pickle file is named, the pickle input is read from standard
19	n/a	input.
20	n/a
21	n/a	Note that recno databases can only contain integer keys, so you can't dump a
22	n/a	hash or btree database using db2pickle.py and reconstitute it to a recno
23	n/a	database with %(prog)s unless your keys are integers.
24	n/a
25	n/a	"""
26	n/a
27	n/a	import getopt
28	n/a	try:
29	n/a	import bsddb
30	n/a	except ImportError:
31	n/a	bsddb = None
32	n/a	try:
33	n/a	import dbm.ndbm as dbm
34	n/a	except ImportError:
35	n/a	dbm = None
36	n/a	try:
37	n/a	import dbm.gnu as gdbm
38	n/a	except ImportError:
39	n/a	gdbm = None
40	n/a	try:
41	n/a	import dbm.ndbm as anydbm
42	n/a	except ImportError:
43	n/a	anydbm = None
44	n/a	import sys
45	n/a	try:
46	n/a	import pickle as pickle
47	n/a	except ImportError:
48	n/a	import pickle
49	n/a
50	n/a	prog = sys.argv[0]
51	n/a
52	n/a	def usage():
53	n/a	sys.stderr.write(__doc__ % globals())
54	n/a
55	n/a	def main(args):
56	n/a	try:
57	n/a	opts, args = getopt.getopt(args, "hbrdag",
58	n/a	["hash", "btree", "recno", "dbm", "anydbm",
59	n/a	"gdbm"])
60	n/a	except getopt.error:
61	n/a	usage()
62	n/a	return 1
63	n/a
64	n/a	if len(args) == 0 or len(args) > 2:
65	n/a	usage()
66	n/a	return 1
67	n/a	elif len(args) == 1:
68	n/a	pfile = sys.stdin
69	n/a	dbfile = args[0]
70	n/a	else:
71	n/a	try:
72	n/a	pfile = open(args[0], 'rb')
73	n/a	except IOError:
74	n/a	sys.stderr.write("Unable to open %s\n" % args[0])
75	n/a	return 1
76	n/a	dbfile = args[1]
77	n/a
78	n/a	dbopen = None
79	n/a	for opt, arg in opts:
80	n/a	if opt in ("-h", "--hash"):
81	n/a	try:
82	n/a	dbopen = bsddb.hashopen
83	n/a	except AttributeError:
84	n/a	sys.stderr.write("bsddb module unavailable.\n")
85	n/a	return 1
86	n/a	elif opt in ("-b", "--btree"):
87	n/a	try:
88	n/a	dbopen = bsddb.btopen
89	n/a	except AttributeError:
90	n/a	sys.stderr.write("bsddb module unavailable.\n")
91	n/a	return 1
92	n/a	elif opt in ("-r", "--recno"):
93	n/a	try:
94	n/a	dbopen = bsddb.rnopen
95	n/a	except AttributeError:
96	n/a	sys.stderr.write("bsddb module unavailable.\n")
97	n/a	return 1
98	n/a	elif opt in ("-a", "--anydbm"):
99	n/a	try:
100	n/a	dbopen = anydbm.open
101	n/a	except AttributeError:
102	n/a	sys.stderr.write("dbm module unavailable.\n")
103	n/a	return 1
104	n/a	elif opt in ("-g", "--gdbm"):
105	n/a	try:
106	n/a	dbopen = gdbm.open
107	n/a	except AttributeError:
108	n/a	sys.stderr.write("dbm.gnu module unavailable.\n")
109	n/a	return 1
110	n/a	elif opt in ("-d", "--dbm"):
111	n/a	try:
112	n/a	dbopen = dbm.open
113	n/a	except AttributeError:
114	n/a	sys.stderr.write("dbm.ndbm module unavailable.\n")
115	n/a	return 1
116	n/a	if dbopen is None:
117	n/a	if bsddb is None:
118	n/a	sys.stderr.write("bsddb module unavailable - ")
119	n/a	sys.stderr.write("must specify dbtype.\n")
120	n/a	return 1
121	n/a	else:
122	n/a	dbopen = bsddb.hashopen
123	n/a
124	n/a	try:
125	n/a	db = dbopen(dbfile, 'c')
126	n/a	except bsddb.error:
127	n/a	sys.stderr.write("Unable to open %s. " % dbfile)
128	n/a	sys.stderr.write("Check for format or version mismatch.\n")
129	n/a	return 1
130	n/a	else:
131	n/a	for k in list(db.keys()):
132	n/a	del db[k]
133	n/a
134	n/a	while 1:
135	n/a	try:
136	n/a	(key, val) = pickle.load(pfile)
137	n/a	except EOFError:
138	n/a	break
139	n/a	db[key] = val
140	n/a
141	n/a	db.close()
142	n/a	pfile.close()
143	n/a
144	n/a	return 0
145	n/a
146	n/a	if __name__ == "__main__":
147	n/a	sys.exit(main(sys.argv[1:]))