Fix/1423 clear docstring

src/ethereum_spec_tools/new_fork/builder.py

@@ -228,6 +228,25 @@ def _to_args(
 
         return commands
 
+@dataclass
+class ClearDocstring(CodemodArgs):
+    """
+    Describe how to clear the docstring in __init__.py to libcst.tool:main.
+    """
+    @override
+    def _to_args(
+        self, fork_builder: "ForkBuilder", working_directory: Path
+    ) -> list[list[str]]:
+        init_path = working_directory / "ethereum" / fork_builder.new_fork / "__init__.py"
+        return [
+            [
+                "codemod",
+                "remove_docstring.RemoveDocstringCommand",
+                "--no-format",
+                str(init_path),
+            ]
+        ]
+
 
 class ForkBuilder:
     """
@@ -353,6 +372,7 @@ def __init__(
             RenameFork(),
             SetForkCriteria(),
             ReplaceForkName(),
+            ReplaceForkName(),
         ]
 
     def _create_working_directory(self) -> TemporaryDirectory:

src/ethereum_spec_tools/new_fork/codemod/remove_docstring.py

@@ -0,0 +1,32 @@
+"""
+libcst codemod that removes the module docstring.
+"""
+import libcst as cst
+from libcst.codemod import CodemodCommand, CodemodContext
+from typing_extensions import override
+
+class RemoveDocstringCommand(CodemodCommand):
+    """
+    Removes the module docstring if it exists.
+    """
+    DESCRIPTION: str = "Remove the module docstring."
+
+    @override
+    def transform_module_impl(self, tree: cst.Module) -> cst.Module:
+        """
+        Transform the tree by removing the docstring.
+        """
+        if len(tree.body) == 0:
+            return tree
+        first_stmt = tree.body[0]
+        if not isinstance(first_stmt, cst.SimpleStatementLine):
+            return tree
+        if len(first_stmt.body) != 1:
+            return tree
+        expr = first_stmt.body[0]
+        if not isinstance(expr, cst.Expr):
+            return tree
+        if not isinstance(expr.value, cst.SimpleString):
+            return tree
+        new_body = tree.body[1:]
+        return tree.with_changes(body=new_body)

vulture_whitelist.py

@@ -20,7 +20,7 @@
 )
 from ethereum_spec_tools.lint.lints.import_hygiene import ImportHygiene
 from ethereum_spec_tools.new_fork.codemod.constant import SetConstantCommand
-from ethereum_spec_tools.new_fork.codemod.string import StringReplaceCommand
+from ethereum_spec_tools.new_fork.codemod.string_replace import StringReplaceCommand
 from ethereum_spec_tools.new_fork.codemod.comment import CommentReplaceCommand
 from ethereum.trace import EvmTracer