liniting

Author: loechel

src/RestrictedPython/compile.py

@@ -1,5 +1,9 @@
 import ast
 import warnings
+from ast import Expression
+from ast import Interactive
+from ast import Module
+from ast import NodeTransformer
 from collections import namedtuple
 from os import PathLike
 from typing import Any
@@ -26,12 +30,13 @@
 
 
 def _compile_restricted_mode(
-        source: str | ReadableBuffer | ast.Module | ast.Expression | ast.Interactive,
+        source: str | ReadableBuffer | Module | Expression | Interactive,
         filename: str | ReadableBuffer | PathLike[Any] = '<string>',
         mode: Literal["exec", "eval", "single"] = "exec",
         flags: int = 0,
         dont_inherit: bool = False,
-        policy: ast.NodeTransformer = RestrictingNodeTransformer) -> CompileResult:
+        policy: NodeTransformer = RestrictingNodeTransformer,
+) -> CompileResult:
 
     if not IS_CPYTHON:
         warnings.warn_explicit(
@@ -47,13 +52,13 @@ def _compile_restricted_mode(
                             dont_inherit=dont_inherit)
     elif issubclass(policy, RestrictingNodeTransformer):
         c_ast = None
-        allowed_source_types = [str, ast.Module]
+        allowed_source_types = [str, Module]
         if not issubclass(type(source), tuple(allowed_source_types)):
             raise TypeError('Not allowed source type: '
                             '"{0.__class__.__name__}".'.format(source))
         c_ast = None
         # workaround for pypy issue https://bitbucket.org/pypy/pypy/issues/2552
-        if isinstance(source, ast.Module):
+        if isinstance(source, Module):
             c_ast = source
         else:
             try:
@@ -86,11 +91,12 @@ def _compile_restricted_mode(
 
 
 def compile_restricted_exec(
-        source: str | ReadableBuffer | ast.Module | ast.Expression | ast.Interactive,
+        source: str | ReadableBuffer | Module | Expression | Interactive,
         filename: str | ReadableBuffer | PathLike[Any] = '<string>',
         flags: int = 0,
         dont_inherit: bool = False,
-        policy: ast.NodeTransformer = RestrictingNodeTransformer) -> CompileResult:
+        policy: NodeTransformer = RestrictingNodeTransformer,
+) -> CompileResult:
     """Compile restricted for the mode `exec`."""
     return _compile_restricted_mode(
         source,
@@ -102,11 +108,12 @@ def compile_restricted_exec(
 
 
 def compile_restricted_eval(
-        source: str | ReadableBuffer | ast.Module | ast.Expression | ast.Interactive,
+        source: str | ReadableBuffer | Module | Expression | Interactive,
         filename: str | ReadableBuffer | PathLike[Any] = '<string>',
         flags: int = 0,
         dont_inherit: bool = False,
-        policy: ast.NodeTransformer = RestrictingNodeTransformer) -> CompileResult:
+        policy: NodeTransformer = RestrictingNodeTransformer,
+) -> CompileResult:
     """Compile restricted for the mode `eval`."""
     return _compile_restricted_mode(
         source,
@@ -118,11 +125,12 @@ def compile_restricted_eval(
 
 
 def compile_restricted_single(
-        source: str | ReadableBuffer | ast.Module | ast.Expression | ast.Interactive,
+        source: str | ReadableBuffer | Module | Expression | Interactive,
         filename: str | ReadableBuffer | PathLike[Any] = '<string>',
         flags: int = 0,
         dont_inherit: bool = False,
-        policy: ast.NodeTransformer = RestrictingNodeTransformer) -> CompileResult:
+        policy: NodeTransformer = RestrictingNodeTransformer,
+) -> CompileResult:
     """Compile restricted for the mode `single`."""
     return _compile_restricted_mode(
         source,
@@ -141,7 +149,8 @@ def compile_restricted_function(
         globalize=None,  # List of globals (e.g. ['here', 'context', ...])
         flags: int = 0,
         dont_inherit: bool = False,
-        policy: ast.NodeTransformer = RestrictingNodeTransformer) -> CompileResult:
+        policy: ast.NodeTransformer = RestrictingNodeTransformer,
+) -> CompileResult:
     """Compile a restricted code object for a function.
 
     Documentation see:
@@ -189,12 +198,13 @@ def compile_restricted_function(
 
 
 def compile_restricted(
-        source: str | ReadableBuffer | ast.Module | ast.Expression | ast.Interactive,
-        filename: str | ReadableBuffer | PathLike[Any] = '<unknown>',
-        mode: str = 'exec',
-        flags: int = 0,
-        dont_inherit: bool = False,
-        policy: ast.NodeTransformer = RestrictingNodeTransformer) -> CompileResult:
+    source: str | ReadableBuffer | Module | Expression | Interactive,
+    filename: str | ReadableBuffer | PathLike[Any] = '<unknown>',
+    mode: str = 'exec',
+    flags: int = 0,
+    dont_inherit: bool = False,
+    policy: NodeTransformer = RestrictingNodeTransformer,
+) -> CompileResult:
     """Replacement for the built-in compile() function.
 
     policy ... `ast.NodeTransformer` class defining the restrictions.