commit b505db6f6df1f67ba437e60d7070a8f6698f6113 · pyrox.dev/nixpkgs

+39 -2
nixos/lib/test-driver/test_driver/logger.py
···

       3
       3
        
       import sys

     

       4
       4
        
       import time

     

       5
       5
        
       import unicodedata

     

       6
       6
       +
       from abc import ABC, abstractmethod

     

       6
       7
        
       from contextlib import contextmanager

     

       7
       8
        
       from queue import Empty, Queue

     

       8
       9
        
       from typing import Any, Dict, Iterator

     
···

       12
       13
        
       from colorama import Fore, Style

     

       13
       14
        
       

     

       14
       15
        
       

     

       15
       15
       -
       class Logger:

     

       16
       16
       +
       class AbstractLogger(ABC):

     

       17
       17
       +
           @abstractmethod

     

       18
       18
       +
           def log(self, message: str, attributes: Dict[str, str] = {}) -> None:

     

       19
       19
       +
               pass

     

       20
       20
       +
       

     

       21
       21
       +
           @abstractmethod

     

       22
       22
       +
           @contextmanager

     

       23
       23
       +
           def subtest(self, name: str, attributes: Dict[str, str] = {}) -> Iterator[None]:

     

       24
       24
       +
               pass

     

       25
       25
       +
       

     

       26
       26
       +
           @abstractmethod

     

       27
       27
       +
           @contextmanager

     

       28
       28
       +
           def nested(self, message: str, attributes: Dict[str, str] = {}) -> Iterator[None]:

     

       29
       29
       +
               pass

     

       30
       30
       +
       

     

       31
       31
       +
           @abstractmethod

     

       32
       32
       +
           def info(self, *args, **kwargs) -> None:  # type: ignore

     

       33
       33
       +
               pass

     

       34
       34
       +
       

     

       35
       35
       +
           @abstractmethod

     

       36
       36
       +
           def warning(self, *args, **kwargs) -> None:  # type: ignore

     

       37
       37
       +
               pass

     

       38
       38
       +
       

     

       39
       39
       +
           @abstractmethod

     

       40
       40
       +
           def error(self, *args, **kwargs) -> None:  # type: ignore

     

       41
       41
       +
               pass

     

       42
       42
       +
       

     

       43
       43
       +
           @abstractmethod

     

       44
       44
       +
           def log_serial(self, message: str, machine: str) -> None:

     

       45
       45
       +
               pass

     

       46
       46
       +
       

     

       47
       47
       +
           @abstractmethod

     

       48
       48
       +
           def print_serial_logs(self, enable: bool) -> None:

     

       49
       49
       +
               pass

     

       50
       50
       +
       

     

       51
       51
       +
       

     

       52
       52
       +
       class Logger(AbstractLogger):

     

       16
       53
        
           def __init__(self) -> None:

     

       17
       54
        
               self.logfile = os.environ.get("LOGFILE", "/dev/null")

     

       18
       55
        
               self.logfile_handle = codecs.open(self.logfile, "wb")

     
···

       110
       147
        
               self.xml.endElement("nest")

     

       111
       148
        
       

     

       112
       149
        
       

     

       113
       113
       -
       rootlog = Logger()

     

       150
       150
       +
       rootlog: AbstractLogger = Logger()