#!/usr/bin/env python2.7

import logging
import os
from logging.handlers import TimedRotatingFileHandler, RotatingFileHandler


for Handler in [TimedRotatingFileHandler, RotatingFileHandler]:
    logger = logging.getLogger('test')
    if os.path.exists('test.log'):
        os.remove('test.log')
    if type(Handler) == TimedRotatingFileHandler:
        handler = Handler('test.log', interval=1440, delay=True)
    else:
        # This bug only manifests itself for RotatingFileHandler
        # if backupCount is set
        handler = Handler('test.log', backupCount=10, delay=True)
    logger.addHandler(handler)

    try:
        logger.handlers[0].doRollover()
    except OSError:
        print str(Handler) + "Failed to rollover"