factor out nullcontext

Author: Kareem Zidane

check50/__main__.py

@@ -22,6 +22,7 @@
 import termcolor
 
 from . import _exceptions, internal, renderer, __version__
+from .contextmanagers import nullcontext
 from .runner import CheckRunner
 
 LOGGER = logging.getLogger("check50")
@@ -53,13 +54,6 @@ def format(self, record):
         return msg if not self.use_color else termcolor.colored(msg, getattr(record, "color", self.COLORS.get(record.levelname)))
 
 
-@contextlib.contextmanager
-def nullcontext(entry_result=None):
-    """This is just contextlib.nullcontext but that function is only available in 3.7+."""
-    yield entry_result
-
-
-
 _exceptions.ExceptHook.initialize()
 
 

check50/_exceptions.py

@@ -5,9 +5,8 @@
 import lib50
 import termcolor
 
-from contextlib import nullcontext
-
 from . import internal, __version__
+from .contextmanagers import nullcontext
 
 class Error(Exception):
     """Exception for internal check50 errors."""

check50/contextmanagers.py

@@ -0,0 +1,7 @@
+import contextlib
+
+
+@contextlib.contextmanager
+def nullcontext(entry_result=None):
+    """This is just contextlib.nullcontext but that function is only available in 3.7+."""
+    yield entry_result