Hopiu/linkchecker
1
0
Fork 0
mirror of https://github.com/Hopiu/linkchecker.git synced 2026-05-13 09:03:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix blacklist file writing

Browse source 
git-svn-id: https://linkchecker.svn.sourceforge.net/svnroot/linkchecker/trunk/linkchecker@3356 e7d03fd6-7b0d-0410-9947-9c21f3af8025
This commit is contained in:
calvin 2006-06-05 20:21:01 +00:00
parent af255b67e0
commit b9150f1bf6
1 changed files with 14 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
linkcheck/logger/blacklist.py
View file

@ -38,9 +38,8 @@ class BlacklistLogger (linkcheck.logger.Logger):
super(BlacklistLogger, self).__init__(**args)
self.init_fileoutput(args)
self.blacklist = {}
if self.filename is not None:
if os.path.exists(self.filename):
self.read_blacklist(self.filename)
if self.filename is not None and os.path.exists(self.filename):
self.read_blacklist()
def comment (self, s, **args):
"""
@ -69,18 +68,20 @@ class BlacklistLogger (linkcheck.logger.Logger):
"""
self.write_blacklist()
def read_blacklist (self, filename):
def read_blacklist (self):
"""
Read a previously stored blacklist from file fd.
"""
fd = file(filename)
for line in fd:
line = line.rstrip()
if line.startswith('#') or not line:
continue
value, key = line.split(None, 1)
self.blacklist[key] = int(value)
fd.close()
fd = open(self.filename)
try:
for line in fd:
line = self.decode(line.rstrip())
if line.startswith('#') or not line:
continue
value, key = line.split(None, 1)
self.blacklist[key] = int(value)
finally:
fd.close()
def write_blacklist (self):
"""
@ -88,8 +89,7 @@ class BlacklistLogger (linkcheck.logger.Logger):
"""
oldmask = os.umask(0077)
for key, value in self.blacklist.iteritems():
self.fd.write("%d %s" % (value, key))
self.fd.write(os.linesep)
self.write(u"%d %s%s" % (value, key, os.linesep))
self.close_fileoutput()
# restore umask
os.umask(oldmask)