fix: indentation of multiline string (#10)

closes: #8

.pre-commit-config.yaml

@@ -14,7 +14,7 @@ repos:
    hooks:
    -   id: black
  - repo: https://github.com/pycqa/isort
-   rev: 5.10.1
+   rev: 5.12.0
    hooks:
      - id: isort
  - repo: https://github.com/pre-commit/mirrors-mypy

.pylintrc

@@ -13,7 +13,7 @@ ignore=CVS
 
 # Add files or directories matching the regex patterns to the ignore-list. The
 # regex matches against paths and can be in Posix or Windows format.
-ignore-paths=
+ignore-paths=tests,docs
 
 # Files or directories matching the regex patterns are skipped. The regex
 # matches against base names, not paths.

README.md

@@ -13,7 +13,7 @@ asort for `__all__` your `__all__` lists
 
 asort is a Python utility / library to sort imports items in an `__all__` list alphabetically
 It provides a command line utility, Python library and pre-commit support to
-quickly sort all your imports. It requires Python 3.7+ to run but supports formatting 
+quickly sort all your imports. It requires Python 3.7+ to run but supports formatting
 any version of Python code.
 
 Works seemlessly with [black](https://github.com/psf/black) and [isort](https://github.com/PyCQA/isort) with no configuration needed

src/asort/api.py

@@ -228,7 +228,7 @@ def write_stream(processed_tokens: Iterable[TokenInfo], output_stream: TextIO) -
             whitespace_len = token.start[1] - prev_token.end[1]
         else:
             whitespace_len = len(token.line) - len(token.line.lstrip())
-            if token.string.isspace() or token.start[0] != token.end[0]:
+            if token.string.isspace():
                 whitespace_len = 0
 
         output_stream.write(" " * whitespace_len)

tests/integration/__snapshots__/test_cli.ambr

@@ -35,6 +35,34 @@
   True
 # ---
 # name: test_cli_output.2
+  '''
+  class SingleLine:
+      """Single line of docstring."""
+
+      var = ["1", "2"]
+
+
+  class Multiline:
+      """Multiple lines of docstring.
+
+      Used to test the docstring with multiple lines.
+      """
+
+      var = ["1", "2"]
+      """Test string in
+      multiple lines."""
+
+      """Test string in a single line."""
+
+      """Test \
+      string in a single line."""
+
+  '''
+# ---
+# name: test_cli_output.3
+  False
+# ---
+# name: test_cli_output.4
   '''
   __all__ = [  # comment
       "Today",  # comment
@@ -44,33 +72,33 @@
 
   '''
 # ---
-# name: test_cli_output.3
+# name: test_cli_output.5
   True
 # ---
-# name: test_cli_output.4
+# name: test_cli_output.6
   '''
   __all__ = ["1", "2", "5", "9"] ; __all__ = ["2", "3", "9"]  # fmt: skip
 
   '''
 # ---
-# name: test_cli_output.5
+# name: test_cli_output.7
   True
 # ---
-# name: test_cli_output.6
+# name: test_cli_output.8
   '''
   __all__ = ["One", "Zed"]
 
   '''
 # ---
-# name: test_cli_output.7
+# name: test_cli_output.9
   False
 # ---
-# name: test_cli_output.8
+# name: test_cli_output.10
   '''
   __all__ = ["Today", "Yesterday", "_Thursday"]  # comment
 
   '''
 # ---
-# name: test_cli_output.9
+# name: test_cli_output.11
   True
 # ---

tests/integration/fixtures/multi_line_string_indentation.py

@@ -0,0 +1,20 @@
+class SingleLine:
+    """Single line of docstring."""
+
+    var = ["1", "2"]
+
+
+class Multiline:
+    """Multiple lines of docstring.
+
+    Used to test the docstring with multiple lines.
+    """
+
+    var = ["1", "2"]
+    """Test string in
+    multiple lines."""
+
+    """Test string in a single line."""
+
+    """Test \
+    string in a single line."""