fix: wip: add function to check if docstring is at end of file

Author: finswimmer

src/docformatter/classify.py

@@ -471,3 +471,18 @@ def is_string_variable(
         return True
 
     return False
+
+
+def is_docstring_at_end_of_file(tokens: list[tokenize.TokenInfo], index: int) -> bool:
+    """Determine if the docstring is at the end of the file."""
+    for i in range(index + 1, len(tokens)):
+        tok = tokens[i]
+        if tok.type not in (
+            tokenize.NL,
+            tokenize.NEWLINE,
+            tokenize.DEDENT,
+            tokenize.ENDMARKER,
+        ):
+            return False
+
+    return True

src/docformatter/format.py

@@ -379,7 +379,10 @@ def _get_newlines_by_type(
     int
         The number of newlines to insert after the docstring.
     """
-    if _classify.is_module_docstring(tokens, index):
+    if _classify.is_docstring_at_end_of_file(tokens, index):
+        # print("End of file")
+        return 0
+    elif _classify.is_module_docstring(tokens, index):
         # print("Module")
         return _get_module_docstring_newlines(black)
     elif _classify.is_class_docstring(tokens, index):