123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226 |
- # -*- test-case-name: twisted.logger.test.test_global -*-
- # Copyright (c) Twisted Matrix Laboratories.
- # See LICENSE for details.
-
- """
- This module includes process-global state associated with the logging system,
- and implementation of logic for managing that global state.
- """
-
- import sys
- import warnings
- from typing import IO, Any, Iterable, Optional, Type
-
- from twisted.python.compat import currentframe
- from twisted.python.reflect import qual
- from ._buffer import LimitedHistoryLogObserver
- from ._file import FileLogObserver
- from ._filter import FilteringLogObserver, LogLevelFilterPredicate
- from ._format import eventAsText
- from ._interfaces import ILogObserver
- from ._io import LoggingFile
- from ._levels import LogLevel
- from ._logger import Logger
- from ._observer import LogPublisher
-
- MORE_THAN_ONCE_WARNING = (
- "Warning: primary log target selected twice at <{fileNow}:{lineNow}> - "
- "previously selected at <{fileThen}:{lineThen}>. Remove one of the calls "
- "to beginLoggingTo."
- )
-
-
- class LogBeginner:
- """
- A L{LogBeginner} holds state related to logging before logging has begun,
- and begins logging when told to do so. Logging "begins" when someone has
- selected a set of observers, like, for example, a L{FileLogObserver} that
- writes to a file on disk, or to standard output.
-
- Applications will not typically need to instantiate this class, except
- those which intend to initialize the global logging system themselves,
- which may wish to instantiate this for testing. The global instance for
- the current process is exposed as
- L{twisted.logger.globalLogBeginner}.
-
- Before logging has begun, a L{LogBeginner} will:
-
- 1. Log any critical messages (e.g.: unhandled exceptions) to the given
- file-like object.
-
- 2. Save (a limited number of) log events in a
- L{LimitedHistoryLogObserver}.
-
- @cvar _DEFAULT_BUFFER_SIZE: The default size for the initial log events
- buffer.
-
- @ivar _initialBuffer: A buffer of messages logged before logging began.
- @ivar _publisher: The log publisher passed in to L{LogBeginner}'s
- constructor.
- @ivar _log: The logger used to log messages about the operation of the
- L{LogBeginner} itself.
- @ivar _stdio: An object with C{stderr} and C{stdout} attributes (like the
- L{sys} module) which will be replaced when redirecting standard I/O.
- @ivar _temporaryObserver: If not L{None}, an L{ILogObserver} that observes
- events on C{_publisher} for this L{LogBeginner}.
- """
-
- _DEFAULT_BUFFER_SIZE = 200
-
- def __init__(
- self,
- publisher: LogPublisher,
- errorStream: IO[Any],
- stdio: object,
- warningsModule: Any,
- initialBufferSize: Optional[int] = None,
- ) -> None:
- """
- Initialize this L{LogBeginner}.
-
- @param initialBufferSize: The size of the event buffer into which
- events are collected until C{beginLoggingTo} is called. Or
- C{None} to use the default size.
- """
- if initialBufferSize is None:
- initialBufferSize = self._DEFAULT_BUFFER_SIZE
- self._initialBuffer = LimitedHistoryLogObserver(size=initialBufferSize)
- self._publisher = publisher
- self._log = Logger(observer=publisher)
- self._stdio = stdio
- self._warningsModule = warningsModule
- self._temporaryObserver: Optional[ILogObserver] = LogPublisher(
- self._initialBuffer,
- FilteringLogObserver(
- FileLogObserver(
- errorStream,
- lambda event: eventAsText(
- event,
- includeTimestamp=False,
- includeSystem=False,
- )
- + "\n",
- ),
- [LogLevelFilterPredicate(defaultLogLevel=LogLevel.critical)],
- ),
- )
- self._previousBegin = ("", 0)
- publisher.addObserver(self._temporaryObserver)
- self._oldshowwarning = warningsModule.showwarning
-
- def beginLoggingTo(
- self,
- observers: Iterable[ILogObserver],
- discardBuffer: bool = False,
- redirectStandardIO: bool = True,
- ) -> None:
- """
- Begin logging to the given set of observers. This will:
-
- 1. Add all the observers given in C{observers} to the
- L{LogPublisher} associated with this L{LogBeginner}.
-
- 2. Optionally re-direct standard output and standard error streams
- to the logging system.
-
- 3. Re-play any messages that were previously logged to that
- publisher to the new observers, if C{discardBuffer} is not set.
-
- 4. Stop logging critical errors from the L{LogPublisher} as strings
- to the C{errorStream} associated with this L{LogBeginner}, and
- allow them to be logged normally.
-
- 5. Re-direct warnings from the L{warnings} module associated with
- this L{LogBeginner} to log messages.
-
- @note: Since a L{LogBeginner} is designed to encapsulate the transition
- between process-startup and log-system-configuration, this method
- is intended to be invoked I{once}.
-
- @param observers: The observers to register.
- @param discardBuffer: Whether to discard the buffer and not re-play it
- to the added observers. (This argument is provided mainly for
- compatibility with legacy concerns.)
- @param redirectStandardIO: If true, redirect standard output and
- standard error to the observers.
- """
- caller = currentframe(1)
- filename = caller.f_code.co_filename
- lineno = caller.f_lineno
-
- for observer in observers:
- self._publisher.addObserver(observer)
-
- if self._temporaryObserver is not None:
- self._publisher.removeObserver(self._temporaryObserver)
- if not discardBuffer:
- self._initialBuffer.replayTo(self._publisher)
- self._temporaryObserver = None
- self._warningsModule.showwarning = self.showwarning
- else:
- previousFile, previousLine = self._previousBegin
- self._log.warn(
- MORE_THAN_ONCE_WARNING,
- fileNow=filename,
- lineNow=lineno,
- fileThen=previousFile,
- lineThen=previousLine,
- )
-
- self._previousBegin = (filename, lineno)
- if redirectStandardIO:
- streams = [("stdout", LogLevel.info), ("stderr", LogLevel.error)]
- else:
- streams = []
-
- for (stream, level) in streams:
- oldStream = getattr(self._stdio, stream)
- loggingFile = LoggingFile(
- logger=Logger(namespace=stream, observer=self._publisher),
- level=level,
- encoding=getattr(oldStream, "encoding", None),
- )
- setattr(self._stdio, stream, loggingFile)
-
- def showwarning(
- self,
- message: str,
- category: Type[Warning],
- filename: str,
- lineno: int,
- file: Optional[IO[Any]] = None,
- line: Optional[str] = None,
- ) -> None:
- """
- Twisted-enabled wrapper around L{warnings.showwarning}.
-
- If C{file} is L{None}, the default behaviour is to emit the warning to
- the log system, otherwise the original L{warnings.showwarning} Python
- function is called.
-
- @param message: A warning message to emit.
- @param category: A warning category to associate with C{message}.
- @param filename: A file name for the source code file issuing the
- warning.
- @param lineno: A line number in the source file where the warning was
- issued.
- @param file: A file to write the warning message to. If L{None},
- write to L{sys.stderr}.
- @param line: A line of source code to include with the warning message.
- If L{None}, attempt to read the line from C{filename} and
- C{lineno}.
- """
- if file is None:
- self._log.warn(
- "{filename}:{lineno}: {category}: {warning}",
- warning=message,
- category=qual(category),
- filename=filename,
- lineno=lineno,
- )
- else:
- self._oldshowwarning(message, category, filename, lineno, file, line)
-
-
- globalLogPublisher = LogPublisher()
- globalLogBeginner = LogBeginner(globalLogPublisher, sys.stderr, sys, warnings)
|