Apply annotations from stub.

Basically move the apply_annotations code from Pyre. Make it a ContextAwareTransformer named ApplyTypeAnnotationsVisitor. Use GatherImportsVisitor to collect imports. Add a static method `add_stub_to_context` so that users can schedule a stub whose types are to be applied later. Use `assertCodemod` and `data_provider` in the tests. Add documentation. Remove fixmes.
Instagram · Mar 23, 2020 · b8c49ef · b8c49ef
1 parent ea8619a
commit b8c49ef
Show file tree

Hide file tree

Showing 4 changed files with 1,094 additions and 0 deletions.
diff --git a/docs/source/codemods.rst b/docs/source/codemods.rst
@@ -153,3 +153,5 @@ inside codemods. As of now, the list includes the following helpers.
   :exclude-members: CONTEXT_KEY, visit_Module, leave_ImportFrom, leave_Module
 .. autoclass:: libcst.codemod.visitors.RemoveImportsVisitor
   :exclude-members: CONTEXT_KEY, METADATA_DEPENDENCIES, visit_Module, leave_ImportFrom, leave_Import
+.. autoclass:: libcst.codemod.visitors.ApplyTypeAnnotationsVisitor
+  :exclude-members: CONTEXT_KEY, transform_module_impl, visit_ClassDef, visit_Comment, visit_FunctionDef, leave_Assign, leave_ClassDef, leave_FunctionDef, leave_ImportFrom, leave_Module
diff --git a/libcst/codemod/visitors/__init__.py b/libcst/codemod/visitors/__init__.py
@@ -5,6 +5,7 @@
 #
 # pyre-strict
 from libcst.codemod.visitors._add_imports import AddImportsVisitor
+from libcst.codemod.visitors._apply_type_annotations import ApplyTypeAnnotationsVisitor
 from libcst.codemod.visitors._gather_exports import GatherExportsVisitor
 from libcst.codemod.visitors._gather_imports import GatherImportsVisitor
 from libcst.codemod.visitors._remove_imports import RemoveImportsVisitor
@@ -14,5 +15,6 @@
     "AddImportsVisitor",
     "GatherImportsVisitor",
     "GatherExportsVisitor",
+    "ApplyTypeAnnotationsVisitor",
     "RemoveImportsVisitor",
 ]