Python code coverage for Lib/plat-irix5/cddb.py

#	count	content
1	n/a	# This file implements a class which forms an interface to the .cddb
2	n/a	# directory that is maintained by SGI's cdman program.
3	n/a	#
4	n/a	# Usage is as follows:
5	n/a	#
6	n/a	# import readcd
7	n/a	# r = readcd.Readcd()
8	n/a	# c = Cddb(r.gettrackinfo())
9	n/a	#
10	n/a	# Now you can use c.artist, c.title and c.track[trackno] (where trackno
11	n/a	# starts at 1). When the CD is not recognized, all values will be the empty
12	n/a	# string.
13	n/a	# It is also possible to set the above mentioned variables to new values.
14	n/a	# You can then use c.write() to write out the changed values to the
15	n/a	# .cdplayerrc file.
16	n/a	from warnings import warnpy3k
17	n/a	warnpy3k("the cddb module has been removed in Python 3.0", stacklevel=2)
18	n/a	del warnpy3k
19	n/a
20	n/a	import string, posix, os
21	n/a
22	n/a	_cddbrc = '.cddb'
23	n/a	_DB_ID_NTRACKS = 5
24	n/a	_dbid_map = '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ@_=+abcdefghijklmnopqrstuvwxyz'
25	n/a	def _dbid(v):
26	n/a	if v >= len(_dbid_map):
27	n/a	return string.zfill(v, 2)
28	n/a	else:
29	n/a	return _dbid_map[v]
30	n/a
31	n/a	def tochash(toc):
32	n/a	if type(toc) == type(''):
33	n/a	tracklist = []
34	n/a	for i in range(2, len(toc), 4):
35	n/a	tracklist.append((None,
36	n/a	(int(toc[i:i+2]),
37	n/a	int(toc[i+2:i+4]))))
38	n/a	else:
39	n/a	tracklist = toc
40	n/a	ntracks = len(tracklist)
41	n/a	hash = _dbid((ntracks >> 4) & 0xF) + _dbid(ntracks & 0xF)
42	n/a	if ntracks <= _DB_ID_NTRACKS:
43	n/a	nidtracks = ntracks
44	n/a	else:
45	n/a	nidtracks = _DB_ID_NTRACKS - 1
46	n/a	min = 0
47	n/a	sec = 0
48	n/a	for track in tracklist:
49	n/a	start, length = track
50	n/a	min = min + length[0]
51	n/a	sec = sec + length[1]
52	n/a	min = min + sec / 60
53	n/a	sec = sec % 60
54	n/a	hash = hash + _dbid(min) + _dbid(sec)
55	n/a	for i in range(nidtracks):
56	n/a	start, length = tracklist[i]
57	n/a	hash = hash + _dbid(length[0]) + _dbid(length[1])
58	n/a	return hash
59	n/a
60	n/a	class Cddb:
61	n/a	def __init__(self, tracklist):
62	n/a	if os.environ.has_key('CDDB_PATH'):
63	n/a	path = os.environ['CDDB_PATH']
64	n/a	cddb_path = path.split(',')
65	n/a	else:
66	n/a	home = os.environ['HOME']
67	n/a	cddb_path = [home + '/' + _cddbrc]
68	n/a
69	n/a	self._get_id(tracklist)
70	n/a
71	n/a	for dir in cddb_path:
72	n/a	file = dir + '/' + self.id + '.rdb'
73	n/a	try:
74	n/a	f = open(file, 'r')
75	n/a	self.file = file
76	n/a	break
77	n/a	except IOError:
78	n/a	pass
79	n/a	ntracks = int(self.id[:2], 16)
80	n/a	self.artist = ''
81	n/a	self.title = ''
82	n/a	self.track = [None] + [''] * ntracks
83	n/a	self.trackartist = [None] + [''] * ntracks
84	n/a	self.notes = []
85	n/a	if not hasattr(self, 'file'):
86	n/a	return
87	n/a	import re
88	n/a	reg = re.compile(r'^([^.])\.([^:]):[\t ]+(.*)')
89	n/a	while 1:
90	n/a	line = f.readline()
91	n/a	if not line:
92	n/a	break
93	n/a	match = reg.match(line)
94	n/a	if not match:
95	n/a	print 'syntax error in ' + file
96	n/a	continue
97	n/a	name1, name2, value = match.group(1, 2, 3)
98	n/a	if name1 == 'album':
99	n/a	if name2 == 'artist':
100	n/a	self.artist = value
101	n/a	elif name2 == 'title':
102	n/a	self.title = value
103	n/a	elif name2 == 'toc':
104	n/a	if not self.toc:
105	n/a	self.toc = value
106	n/a	if self.toc != value:
107	n/a	print 'toc\'s don\'t match'
108	n/a	elif name2 == 'notes':
109	n/a	self.notes.append(value)
110	n/a	elif name1[:5] == 'track':
111	n/a	try:
112	n/a	trackno = int(name1[5:])
113	n/a	except strings.atoi_error:
114	n/a	print 'syntax error in ' + file
115	n/a	continue
116	n/a	if trackno > ntracks:
117	n/a	print 'track number %r in file %r out of range' % (trackno, file)
118	n/a	continue
119	n/a	if name2 == 'title':
120	n/a	self.track[trackno] = value
121	n/a	elif name2 == 'artist':
122	n/a	self.trackartist[trackno] = value
123	n/a	f.close()
124	n/a	for i in range(2, len(self.track)):
125	n/a	track = self.track[i]
126	n/a	# if track title starts with `,', use initial part
127	n/a	# of previous track's title
128	n/a	if track and track[0] == ',':
129	n/a	try:
130	n/a	off = self.track[i - 1].index(',')
131	n/a	except ValueError:
132	n/a	pass
133	n/a	else:
134	n/a	self.track[i] = self.track[i-1][:off] \
135	n/a	+ track
136	n/a
137	n/a	def _get_id(self, tracklist):
138	n/a	# fill in self.id and self.toc.
139	n/a	# if the argument is a string ending in .rdb, the part
140	n/a	# upto the suffix is taken as the id.
141	n/a	if type(tracklist) == type(''):
142	n/a	if tracklist[-4:] == '.rdb':
143	n/a	self.id = tracklist[:-4]
144	n/a	self.toc = ''
145	n/a	return
146	n/a	t = []
147	n/a	for i in range(2, len(tracklist), 4):
148	n/a	t.append((None, \
149	n/a	(int(tracklist[i:i+2]), \
150	n/a	int(tracklist[i+2:i+4]))))
151	n/a	tracklist = t
152	n/a	ntracks = len(tracklist)
153	n/a	self.id = _dbid((ntracks >> 4) & 0xF) + _dbid(ntracks & 0xF)
154	n/a	if ntracks <= _DB_ID_NTRACKS:
155	n/a	nidtracks = ntracks
156	n/a	else:
157	n/a	nidtracks = _DB_ID_NTRACKS - 1
158	n/a	min = 0
159	n/a	sec = 0
160	n/a	for track in tracklist:
161	n/a	start, length = track
162	n/a	min = min + length[0]
163	n/a	sec = sec + length[1]
164	n/a	min = min + sec / 60
165	n/a	sec = sec % 60
166	n/a	self.id = self.id + _dbid(min) + _dbid(sec)
167	n/a	for i in range(nidtracks):
168	n/a	start, length = tracklist[i]
169	n/a	self.id = self.id + _dbid(length[0]) + _dbid(length[1])
170	n/a	self.toc = string.zfill(ntracks, 2)
171	n/a	for track in tracklist:
172	n/a	start, length = track
173	n/a	self.toc = self.toc + string.zfill(length[0], 2) + \
174	n/a	string.zfill(length[1], 2)
175	n/a
176	n/a	def write(self):
177	n/a	import posixpath
178	n/a	if os.environ.has_key('CDDB_WRITE_DIR'):
179	n/a	dir = os.environ['CDDB_WRITE_DIR']
180	n/a	else:
181	n/a	dir = os.environ['HOME'] + '/' + _cddbrc
182	n/a	file = dir + '/' + self.id + '.rdb'
183	n/a	if posixpath.exists(file):
184	n/a	# make backup copy
185	n/a	posix.rename(file, file + '~')
186	n/a	f = open(file, 'w')
187	n/a	f.write('album.title:\t' + self.title + '\n')
188	n/a	f.write('album.artist:\t' + self.artist + '\n')
189	n/a	f.write('album.toc:\t' + self.toc + '\n')
190	n/a	for note in self.notes:
191	n/a	f.write('album.notes:\t' + note + '\n')
192	n/a	prevpref = None
193	n/a	for i in range(1, len(self.track)):
194	n/a	if self.trackartist[i]:
195	n/a	f.write('track%r.artist:\t%s\n' % (i, self.trackartist[i]))
196	n/a	track = self.track[i]
197	n/a	try:
198	n/a	off = track.index(',')
199	n/a	except ValuError:
200	n/a	prevpref = None
201	n/a	else:
202	n/a	if prevpref and track[:off] == prevpref:
203	n/a	track = track[off:]
204	n/a	else:
205	n/a	prevpref = track[:off]
206	n/a	f.write('track%r.title:\t%s\n' % (i, track))
207	n/a	f.close()