mirror of
https://github.com/python/cpython.git
synced 2025-08-31 05:58:33 +00:00
bpo-45401: Change shouldRollover() methods to only rollover regular f… (GH-28822) (#28866)
…iles.
Also changed some historical return values from 1 -> True and 0 -> False.
(cherry picked from commit 62a667784b
)
Co-authored-by: Vinay Sajip <vinay_sajip@yahoo.co.uk>
Co-authored-by: Vinay Sajip <vinay_sajip@yahoo.co.uk>
This commit is contained in:
parent
d57d33c234
commit
ac421c348b
2 changed files with 25 additions and 4 deletions
|
@ -185,14 +185,17 @@ class RotatingFileHandler(BaseRotatingHandler):
|
|||
Basically, see if the supplied record would cause the file to exceed
|
||||
the size limit we have.
|
||||
"""
|
||||
# See bpo-45401: Never rollover anything other than regular files
|
||||
if os.path.exists(self.baseFilename) and not os.path.isfile(self.baseFilename):
|
||||
return False
|
||||
if self.stream is None: # delay was set...
|
||||
self.stream = self._open()
|
||||
if self.maxBytes > 0: # are we rolling over?
|
||||
msg = "%s\n" % self.format(record)
|
||||
self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
|
||||
if self.stream.tell() + len(msg) >= self.maxBytes:
|
||||
return 1
|
||||
return 0
|
||||
return True
|
||||
return False
|
||||
|
||||
class TimedRotatingFileHandler(BaseRotatingHandler):
|
||||
"""
|
||||
|
@ -342,10 +345,13 @@ class TimedRotatingFileHandler(BaseRotatingHandler):
|
|||
record is not used, as we are just comparing times, but it is needed so
|
||||
the method signatures are the same
|
||||
"""
|
||||
# See bpo-45401: Never rollover anything other than regular files
|
||||
if os.path.exists(self.baseFilename) and not os.path.isfile(self.baseFilename):
|
||||
return False
|
||||
t = int(time.time())
|
||||
if t >= self.rolloverAt:
|
||||
return 1
|
||||
return 0
|
||||
return True
|
||||
return False
|
||||
|
||||
def getFilesToDelete(self):
|
||||
"""
|
||||
|
|
|
@ -5066,6 +5066,13 @@ class RotatingFileHandlerTest(BaseFileTest):
|
|||
rh = logging.handlers.RotatingFileHandler(self.fn, maxBytes=0)
|
||||
self.assertFalse(rh.shouldRollover(None))
|
||||
rh.close()
|
||||
# bpo-45401 - test with special file
|
||||
# We set maxBytes to 1 so that rollover would normally happen, except
|
||||
# for the check for regular files
|
||||
rh = logging.handlers.RotatingFileHandler(
|
||||
os.devnull, encoding="utf-8", maxBytes=1)
|
||||
self.assertFalse(rh.shouldRollover(self.next_rec()))
|
||||
rh.close()
|
||||
|
||||
def test_should_rollover(self):
|
||||
rh = logging.handlers.RotatingFileHandler(self.fn, maxBytes=1)
|
||||
|
@ -5160,6 +5167,14 @@ class RotatingFileHandlerTest(BaseFileTest):
|
|||
rh.close()
|
||||
|
||||
class TimedRotatingFileHandlerTest(BaseFileTest):
|
||||
def test_should_not_rollover(self):
|
||||
# See bpo-45401. Should only ever rollover regular files
|
||||
fh = logging.handlers.TimedRotatingFileHandler(
|
||||
os.devnull, 'S', encoding="utf-8", backupCount=1)
|
||||
time.sleep(1.1) # a little over a second ...
|
||||
r = logging.makeLogRecord({'msg': 'testing - device file'})
|
||||
self.assertFalse(fh.shouldRollover(r))
|
||||
|
||||
# other test methods added below
|
||||
def test_rollover(self):
|
||||
fh = logging.handlers.TimedRotatingFileHandler(self.fn, 'S',
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue