Added workaround and early returning if file name matches excluded arg

bundled/tool/lsp_server.py

@@ -394,6 +394,19 @@ def _run_tool_on_document(
     argv += TOOL_ARGS + settings["args"] + extra_args
 
     if use_stdin:
+        arg = _parse_autopep_exclude_arg(argv)
+
+        if arg.exclude is not None:
+            exclude_patterns = _split_comma_separated(arg.exclude)
+
+            import fnmatch
+            for pattern in exclude_patterns:
+                if fnmatch.fnmatch(document.path, pattern):
+                    log_to_output(f"Excluded file: {document.path} because it matches patterh: {pattern}")
+                    return None
+
+
+
         argv += ["-"]
 
     if use_path:
@@ -545,6 +558,29 @@ def _to_run_result_with_logging(rpc_result: jsonrpc.RpcRunResult) -> utils.RunRe
     return utils.RunResult(rpc_result.stdout, error)
 
 
+def _parse_autopep_exclude_arg(
+    argv: list(str)
+):
+    import argparse
+    parser = argparse.ArgumentParser(
+        description="Exclude Argument Parser"
+    )
+
+    parser.add_argument(
+        "--exclude",
+        metavar='globs',
+        required=False
+    )
+
+    exclude_argument, _ = parser.parse_known_args(argv)
+
+    return exclude_argument
+
+def _split_comma_separated(string: str):
+    """Return a set of strings."""
+    return {text.strip() for text in string.split(',') if text.strip()}
+
+
 # *****************************************************
 # Logging and notification.
 # *****************************************************

src/test/python_tests/test_data/sample3/sample.unformatted

@@ -0,0 +1,2 @@
+import sys;print(sys.executable)
+# There is intentionally no last empty line.

src/test/python_tests/test_data/sample4/sample.excluded.unformatted.

@@ -0,0 +1 @@
+

src/test/python_tests/test_data/sample4/sample.included.unformatted

@@ -0,0 +1,2 @@
+import sys;print(sys.executable)
+# There is intentionally no last empty line.

src/test/python_tests/test_data/sample4/sample_formatted.py

@@ -0,0 +1,3 @@
+import sys
+print(sys.executable)
+# There is intentionally no last empty line.

src/test/python_tests/test_formatting.py

@@ -7,7 +7,7 @@
 
 from hamcrest import assert_that, is_
 
-from .lsp_test_client import constants, session, utils
+from .lsp_test_client import constants, defaults, session, utils
 
 FORMATTER = utils.get_server_info_defaults()
 TIMEOUT = 10000  # 10 seconds
@@ -138,3 +138,127 @@ def test_skipping_site_packages_files():
 
     expected = None
     assert_that(actual, is_(expected))
+
+
+def test_skipping_excluded_files():
+    """Test skipping formatting when the file is in excluded pattern"""
+
+    UNFORMATTED_EXCLUDED_FILE_PATH = constants.TEST_DATA / "sample3" / "sample.unformatted"
+    with session.LspSession() as ls_session:
+        # Use any stdlib path here
+        uri = utils.as_uri(pathlib.__file__)
+
+        init_options = defaults.VSCODE_DEFAULT_INITIALIZE["initializationOptions"]
+        init_options["settings"][0]["args"] = ["--exclude='**/*.py'"]
+        ls_session.initialize(defaults.VSCODE_DEFAULT_INITIALIZE)
+        ls_session.initialize()
+
+        ls_session.notify_did_open(
+            {
+                "textDocument": {
+                    "uri": uri,
+                    "languageId": "python",
+                    "version": 1,
+                    "text": UNFORMATTED_EXCLUDED_FILE_PATH.read_text(encoding="utf-8"),
+                }
+            }
+        )
+
+        actual = ls_session.text_document_formatting(
+            {
+                "textDocument": {"uri": uri},
+                # `options` is not used by black
+                "options": {"tabSize": 4, "insertSpaces": True},
+            }
+        )
+
+    expected = None
+    assert_that(actual, is_(expected))
+
+    def test_formatting_file_not_in_excluded_files():
+        """Test formatting when the file is not in excluded pattern"""
+        FORMATTED_TEST_FILE_PATH = constants.TEST_DATA / "sample4" / "sample.py"
+        UNFORMATTED_INCLUDED_FILE_PATH = constants.TEST_DATA / "sample4" / "sample.included.unformatted"
+
+        with session.LspSession() as ls_session:
+            # Use any stdlib path here
+            uri = utils.as_uri(pathlib.__file__)
+
+            init_options = defaults.VSCODE_DEFAULT_INITIALIZE["initializationOptions"]
+            init_options["settings"][0]["args"] = ["--exclude='**/*exclude'"]
+            ls_session.initialize(defaults.VSCODE_DEFAULT_INITIALIZE)
+            ls_session.initialize()
+
+            ls_session.notify_did_open(
+                {
+                    "textDocument": {
+                        "uri": uri,
+                        "languageId": "python",
+                        "version": 1,
+                        "text": UNFORMATTED_INCLUDED_FILE_PATH.read_text(encoding="utf-8"),
+                    }
+                }
+            )
+
+            actual = ls_session.text_document_formatting(
+                {
+                    "textDocument": {"uri": uri},
+                    # `options` is not used by black
+                    "options": {"tabSize": 4, "insertSpaces": True},
+                }
+            )
+
+        expected = [
+            {
+                "range": {
+                    "start": {"line": 0, "character": 0},
+                    "end": {"line": len(lines), "character": 0},
+                    },
+                "newText": FORMATTED_TEST_FILE_PATH.read_text(),
+                }
+            ]
+        assert_that(actual, is_(expected))
+
+    def test_formatting_file_with_excluded_and_other_args():
+        """Test formatting when we have more arguments specified"""
+        FORMATTED_TEST_FILE_PATH = constants.TEST_DATA / "sample4" / "sample.py"
+        UNFORMATTED_INCLUDED_FILE_PATH = constants.TEST_DATA / "sample4" / "sample.included.unformatted"
+
+        with session.LspSession() as ls_session:
+            # Use any stdlib path here
+            uri = utils.as_uri(pathlib.__file__)
+
+            init_options = defaults.VSCODE_DEFAULT_INITIALIZE["initializationOptions"]
+            init_options["settings"][0]["args"] = ["--exclude='**/*exclude'", "--aggressive"]
+            ls_session.initialize(defaults.VSCODE_DEFAULT_INITIALIZE)
+            ls_session.initialize()
+
+            ls_session.notify_did_open(
+                {
+                    "textDocument": {
+                        "uri": uri,
+                        "languageId": "python",
+                        "version": 1,
+                        "text": UNFORMATTED_INCLUDED_FILE_PATH.read_text(encoding="utf-8"),
+                    }
+                }
+            )
+
+            actual = ls_session.text_document_formatting(
+                {
+                    "textDocument": {"uri": uri},
+                    # `options` is not used by black
+                    "options": {"tabSize": 4, "insertSpaces": True},
+                }
+            )
+
+        expected = [
+            {
+                "range": {
+                    "start": {"line": 0, "character": 0},
+                    "end": {"line": len(lines), "character": 0},
+                    },
+                "newText": FORMATTED_TEST_FILE_PATH.read_text(),
+                }
+            ]
+        assert_that(actual, is_(expected))