[AddImportsVisitor] generate deterministic add import output by sorti…

…ng the sets.
Instagram · Mar 25, 2020 · 67f7d66 · 67f7d66
1 parent 1b9a52b
commit 67f7d66
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/libcst/codemod/visitors/_add_imports.py b/libcst/codemod/visitors/_add_imports.py
@@ -228,17 +228,20 @@ def leave_ImportFrom(
 
         # Now, do the actual update.
         return updated_node.with_changes(
-            names=(
-                *[libcst.ImportAlias(name=libcst.Name(imp)) for imp in imports_to_add],
-                *[
+            names=[
+                *(
+                    libcst.ImportAlias(name=libcst.Name(imp))
+                    for imp in sorted(imports_to_add)
+                ),
+                *(
                     libcst.ImportAlias(
                         name=libcst.Name(imp),
                         asname=libcst.AsName(name=libcst.Name(alias)),
                     )
-                    for (imp, alias) in aliases_to_add
-                ],
+                    for (imp, alias) in sorted(aliases_to_add)
+                ),
                 *updated_node.names,
-            )
+            ]
         )
 
     def _split_module(