Sort imports, pass flake8 by adding __all__ to colorlog.wrappers

borntyping · Aug 23, 2021 · feb4652 · feb4652
1 parent 194fd9d
commit feb4652
Show file tree

Hide file tree

Showing 4 changed files with 43 additions and 21 deletions.
diff --git a/colorlog/__init__.py b/colorlog/__init__.py
@@ -32,30 +32,30 @@
 )
 
 __all__ = (
+    "CRITICAL",
+    "DEBUG",
+    "ERROR",
+    "FATAL",
+    "INFO",
+    "NOTSET",
+    "WARN",
+    "WARNING",
     "ColoredFormatter",
-    "default_log_colors",
+    "LevelFormatter",
+    "StreamHandler",
+    "TTYColoredFormatter",
     "basicConfig",
-    "root",
-    "getLogger",
+    "critical",
     "debug",
-    "info",
-    "warning",
+    "default_log_colors",
     "error",
     "exception",
-    "critical",
-    "log",
     "exception",
-    "StreamHandler",
-    "LevelFormatter",
-    "TTYColoredFormatter",
-    "CRITICAL",
-    "FATAL",
-    "ERROR",
-    "WARNING",
-    "WARN",
-    "INFO",
-    "DEBUG",
-    "NOTSET",
+    "getLogger",
+    "info",
+    "log",
+    "root",
+    "warning",
 )
 
 if sys.version_info < (3, 6):

diff --git a/colorlog/tests/test_escape_codes.py b/colorlog/tests/test_escape_codes.py
@@ -1,9 +1,9 @@
 """Test the colorlog.escape_codes module."""
 
-from colorlog.escape_codes import escape_codes, esc, parse_colors
-
 import pytest
 
+from colorlog.escape_codes import esc, escape_codes, parse_colors
+
 
 def test_esc():
     assert esc(1, 2, 3) == "\033[1;2;3m"

diff --git a/colorlog/wrappers.py b/colorlog/wrappers.py
@@ -19,6 +19,28 @@
 
 import colorlog.formatter
 
+__all__ = (
+    "CRITICAL",
+    "DEBUG",
+    "ERROR",
+    "FATAL",
+    "INFO",
+    "NOTSET",
+    "WARN",
+    "WARNING",
+    "StreamHandler",
+    "basicConfig",
+    "critical",
+    "debug",
+    "error",
+    "exception",
+    "getLogger",
+    "info",
+    "log",
+    "root",
+    "warning",
+)
+
 
 def basicConfig(
     style: str = "%",

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name="colorlog",
-    version="6.4.0",
+    version="6.4.1",
     description="Add colours to the output of Python's logging module.",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",