import logging
import sys

import gym

logger = logging.getLogger(__name__)

root_logger = logging.getLogger()

# Should be "gym", but we'll support people doing somewhat crazy
# things.
package_name = '.'.join(__name__.split('.')[:-1])
gym_logger = logging.getLogger(package_name)

# Should be modified only by official Gym plugins. This is an
# unsupported API and may be removed in future versions.
_extra_loggers = [gym_logger]

# Set up the default handler
formatter = logging.Formatter('[%(asctime)s] %(message)s')
handler = logging.StreamHandler(sys.stderr)
handler.setFormatter(formatter)

# We need to take in the gym logger explicitly since this is called
# at initialization time.
def logger_setup(_=None):
    # This used to take in an argument; we still take an (ignored)
    # argument for compatibility.
    root_logger.addHandler(handler)
    for logger in _extra_loggers:
        logger.setLevel(logging.INFO)

def undo_logger_setup():
    """Undoes the automatic logging setup done by OpenAI Gym. You should call
    this function if you want to manually configure logging
    yourself. Typical usage would involve putting something like the
    following at the top of your script:

    gym.undo_logger_setup()
    logger = logging.getLogger()
    logger.addHandler(logging.StreamHandler(sys.stderr))
    """
    root_logger.removeHandler(handler)
    for logger in _extra_loggers:
        logger.setLevel(logging.NOTSET)
