관리-도구

편집 파일: pylama_isort.py

import os
import sys
from contextlib import contextmanager
from typing import Any, Dict, Iterator, List, Optional

from pylama.lint import Linter as BaseLinter  # type: ignore

from isort.exceptions import FileSkipped

from . import api

@contextmanager
def suppress_stdout() -> Iterator[None]:
    stdout = sys.stdout
    with open(os.devnull, "w") as devnull:
        sys.stdout = devnull
        yield
        sys.stdout = stdout

class Linter(BaseLinter):  # type: ignore
    def allow(self, path: str) -> bool:
        """Determine if this path should be linted."""
        return path.endswith(".py")

def run(
        self, path: str, params: Optional[Dict[str, Any]] = None, **meta: Any
    ) -> List[Dict[str, Any]]:
        """Lint the file. Return an array of error dicts if appropriate."""
        with suppress_stdout():
            try:
                if not api.check_file(path, disregard_skip=False, **params or {}):
                    return [
                        {
                            "lnum": 0,
                            "col": 0,
                            "text": "Incorrectly sorted imports.",
                            "type": "ISORT",
                        }
                    ]
            except FileSkipped:
                pass

return []