Left shift and right shift added to log21.Logger.Logger

Author: MPCodeWriter21

src/log21/Logger.py

@@ -2,9 +2,10 @@
 # CodeWriter21
 
 import re as _re
+import sys as _sys
 import logging as _logging
 from types import MethodType as _MethodType
-from typing import (Union as _Union, Literal as _Literal, Mapping,
+from typing import (Any, List, Union as _Union, Literal as _Literal, Mapping,
                     Callable as _Callable, Optional as _Optional, Sequence as _Sequence)
 from getpass import getpass as _getpass
 from logging import raiseExceptions as _raiseExceptions
@@ -40,7 +41,7 @@ def __init__(
                     handlers = [handlers]
                 else:
                     raise TypeError(
-                        "handlers must be a list of logging.Handler objects"
+                        'handlers must be a list of logging.Handler objects'
                     )
             for handler in handlers:
                 self.addHandler(handler)
@@ -61,7 +62,7 @@ def log(self, level: int, *msg, args: tuple = (), end='\n', **kwargs):
         msg = ' '.join([str(m) for m in msg]) + end
         if not isinstance(level, int):
             if _raiseExceptions:
-                raise TypeError("level must be an integer")
+                raise TypeError('level must be an integer')
             return
         if self.isEnabledFor(level):
             self._log(level, msg, args, **kwargs)
@@ -161,19 +162,22 @@ def input(self, *msg, args: tuple = (), end='', **kwargs):
         """Log 'msg % args'.
 
         To pass exception information, use the keyword argument exc_info with a true
-        value, e.g.
+        value.
 
+        Usage example:
         age = logger.input("Enter your age: ")
         """
         msg = ' '.join([str(m) for m in msg]) + end
         self._log(INPUT, msg, args, **kwargs)
         return input()
 
     def getpass(self, *msg, args: tuple = (), end='', **kwargs):
-        """
-
+        """Takes a password input from the user.
 
-        :return:
+        :param msg: The message to display to the user.
+        :param args: The arguments to pass to the message.
+        :param end: The ending character to append to the message.
+        :return: The password.
         """
         msg = ' '.join([str(m) for m in msg]) + end
         self._log(self.level if self.level >= NOTSET else NOTSET, msg, args, **kwargs)
@@ -213,7 +217,7 @@ def add_level(
         self,
         level: int,
         name: str,
-        errors: _Literal["raise", "ignore", "handle", "force"] = "raise"
+        errors: _Literal['raise', 'ignore', 'handle', 'force'] = 'raise'
     ) -> str:
         """Adds a new method to the logger with a specific level and name.
 
@@ -276,7 +280,7 @@ def log_for_level(self, *msg, args: tuple = (), end='\n', **kwargs):
     def add_levels(
         self,
         level_names: Mapping[int, str],
-        errors: _Literal["raise", "ignore", "handle", "force"] = "raise"
+        errors: _Literal['raise', 'ignore', 'handle', 'force'] = 'raise'
     ) -> None:
         """Adds new methods to the logger with specific levels and names.
 
@@ -287,6 +291,67 @@ def add_levels(
         for level, name in level_names.items():
             self.add_level(level, name, errors)
 
+    def __lshift__(self, obj):
+        """Prints the object to the output stream.
+        This operator is meant to make the Logger object be usable in a
+        std::cout-like way.
+
+        :param obj: The object to print.
+        :return: The Logger object.
+        """
+        logger = self
+        found = 0
+        while logger:
+            for handler in logger.handlers:
+                if (isinstance(handler, _logging.StreamHandler)
+                        and hasattr(handler.stream, 'write')
+                        and hasattr(handler.stream, 'flush')):
+                    found = found + 1
+                    handler.stream.write(str(obj))
+                    handler.stream.flush()
+            if not logger.propagate:
+                break
+            logger = logger.parent
+        if found == 0:
+            _sys.stderr.write(
+                f"No handlers could be found for logger \"{self.name}\"\n"
+            )
+        return self
+
+    def __rshift__(self, obj: List[Any]):
+        """A way of receiving input from the stdin.
+        This operator is meant to make a std::cin-like operation possible in Python.
+
+        :param obj: The object to redirect the output to.
+        :return: The Logger object.
+        """
+        n = len(obj) - 1
+        if n >= 0:
+            data = []
+            while n >= 0:
+                tmp = _sys.stdin.readline()[:-1].split(maxsplit=n)
+                data.extend(tmp)
+                n -= len(tmp)
+                tmp = []
+                for i, item in enumerate(data):
+                    if obj[i] is None:
+                        tmp.append(item)
+                    elif isinstance(obj[i], type):
+                        try:
+                            tmp.append(obj[i](item))
+                        except ValueError:
+                            tmp.append(obj[i]())
+                    else:
+                        try:
+                            tmp.append(obj[i].__class__(item))
+                        except ValueError:
+                            tmp.append(obj[i])
+                obj[:] = tmp
+        else:
+            obj[:] = _sys.stdin.readline()[:-1].split()
+
+        return self
+
 
 def _add_one(name: str) -> str:
     """Add one to the end of a string.

src/log21/__init__.py

@@ -6,24 +6,24 @@
 from typing import (Type as _Type, Tuple as _Tuple, Union as _Union,
                     Mapping as _Mapping, Optional as _Optional)
 
-from log21 import CrashReporter
-from log21.Colors import (Colors, get_color, get_colors, ansi_escape, closest_color,
-                          get_color_name)
-from log21.Levels import (INFO, WARN, DEBUG, ERROR, FATAL, INPUT, NOTSET, WARNING,
-                          CRITICAL)
-from log21.Logger import Logger
-from log21.PPrint import PrettyPrinter, pformat
-from log21.Manager import Manager
-from log21.Argparse import ColorizingArgumentParser
-from log21.TreePrint import TreePrint, tree_format
-from log21.Formatters import ColorizingFormatter, DecolorizingFormatter, _Formatter
-from log21.Argumentify import argumentify
-from log21.FileHandler import FileHandler, DecolorizingFileHandler
-from log21.ProgressBar import ProgressBar
-from log21.LoggingWindow import LoggingWindow, LoggingWindowHandler
-from log21.StreamHandler import StreamHandler, ColorizingStreamHandler
-
-__author__ =         'CodeWriter21 (Mehrad Pooryoussof)'
+from . import CrashReporter
+from .Colors import (Colors, get_color, get_colors, ansi_escape, closest_color,
+                     get_color_name)
+from .Levels import INFO, WARN, DEBUG, ERROR, FATAL, INPUT, NOTSET, WARNING, CRITICAL
+from .Logger import Logger
+from .PPrint import PrettyPrinter, pformat
+from .Manager import Manager
+from .Argparse import ColorizingArgumentParser
+from .TreePrint import TreePrint, tree_format
+from .Formatters import ColorizingFormatter, DecolorizingFormatter, _Formatter
+from .Argumentify import argumentify
+from .FileHandler import FileHandler, DecolorizingFileHandler
+from .ProgressBar import ProgressBar
+from .LoggingWindow import LoggingWindow, LoggingWindowHandler
+from .StreamHandler import StreamHandler, ColorizingStreamHandler
+
+__author__ = 'CodeWriter21 (Mehrad Pooryoussof)'
+__version__ = '2.9.0'
 __github__ = 'Https://GitHub.com/MPCodeWriter21/log21'
 __all__ = [
     'ColorizingStreamHandler', 'DecolorizingFileHandler', 'ColorizingFormatter',