From 701a0ac09f8e8fc0d337bb64d10027dd168e3b54 Mon Sep 17 00:00:00 2001
From: Andreas Arvidsson <andreas.arvidsson87@gmail.com>
Date: Tue, 5 Dec 2023 13:48:08 +0100
Subject: [PATCH] Added python import to collection item

---
 .../cursorless-engine/src/languages/python.ts |  4 ++++
 .../recorded/languages/python/chuckItem.yml   | 23 +++++++++++++++++++
 2 files changed, 27 insertions(+)
 create mode 100644 packages/cursorless-vscode-e2e/src/suite/fixtures/recorded/languages/python/chuckItem.yml

diff --git a/packages/cursorless-engine/src/languages/python.ts b/packages/cursorless-engine/src/languages/python.ts
index 309d5b79fa..041d265170 100644
--- a/packages/cursorless-engine/src/languages/python.ts
+++ b/packages/cursorless-engine/src/languages/python.ts
@@ -43,6 +43,10 @@ const nodeMatchers: Partial<
       itemNodeFinder("import_from_statement", "dotted_name", true),
       argumentSelectionExtractor(),
     ),
+    matcher(
+      itemNodeFinder("import_statement", "dotted_name", false),
+      argumentSelectionExtractor(),
+    ),
     matcher(
       itemNodeFinder("global_statement", "identifier"),
       argumentSelectionExtractor(),
diff --git a/packages/cursorless-vscode-e2e/src/suite/fixtures/recorded/languages/python/chuckItem.yml b/packages/cursorless-vscode-e2e/src/suite/fixtures/recorded/languages/python/chuckItem.yml
new file mode 100644
index 0000000000..7863ac5bc1
--- /dev/null
+++ b/packages/cursorless-vscode-e2e/src/suite/fixtures/recorded/languages/python/chuckItem.yml
@@ -0,0 +1,23 @@
+languageId: python
+command:
+  version: 6
+  spokenForm: chuck item
+  action:
+    name: remove
+    target:
+      type: primitive
+      modifiers:
+        - type: containingScope
+          scopeType: {type: collectionItem}
+  usePrePhraseSnapshot: true
+initialState:
+  documentContents: import foo, baz
+  selections:
+    - anchor: {line: 0, character: 7}
+      active: {line: 0, character: 7}
+  marks: {}
+finalState:
+  documentContents: import baz
+  selections:
+    - anchor: {line: 0, character: 7}
+      active: {line: 0, character: 7}