Add current package selection syntax (#9270)

Co-authored-by: Jeremy Cohen <[email protected]>
Co-authored-by: Kshitij Aranke <[email protected]>

.changes/unreleased/Features-20231212-163409.yaml

@@ -0,0 +1,6 @@
+kind: Features
+body: Package selector syntax for the current package
+time: 2023-12-12T16:34:09.328891Z
+custom:
+  Author: barton996
+  Issue: "6891"

core/dbt/graph/selector_methods.py

@@ -455,6 +455,10 @@ def search(self, included_nodes: Set[UniqueId], selector: str) -> Iterator[Uniqu
 class PackageSelectorMethod(SelectorMethod):
     def search(self, included_nodes: Set[UniqueId], selector: str) -> Iterator[UniqueId]:
         """Yields nodes from included that have the specified package"""
+        # `this` is an alias for the current dbt project name
+        if selector == "this" and self.manifest.metadata.project_name is not None:
+            selector = self.manifest.metadata.project_name
+
         for unique_id, node in self.all_nodes(included_nodes):
             if fnmatch(node.package_name, selector):
                 yield unique_id

tests/unit/test_graph_selector_methods.py

@@ -343,6 +343,33 @@ def test_select_package(manifest):
     }
 
 
+def test_select_package_this(manifest):
+    new_manifest = copy.deepcopy(manifest)
+
+    # change the package name for all nodes except ones where the unique_id contains "table_model"
+    for id, node in new_manifest.nodes.items():
+        if "table_model" not in id:
+            node.package_name = "foo"
+
+    for source in new_manifest.sources.values():
+        if "table_model" not in source.unique_id:
+            source.package_name = "foo"
+
+    methods = MethodManager(new_manifest, None)
+    method = methods.get_method("package", [])
+    assert isinstance(method, PackageSelectorMethod)
+    assert method.arguments == []
+
+    assert search_manifest_using_method(new_manifest, method, "this") == {
+        "not_null_table_model_id",
+        "table_model",
+        "table_model_csv",
+        "table_model_py",
+        "unique_table_model_id",
+        "unit_test_table_model",
+    }
+
+
 def test_select_config_materialized(manifest):
     methods = MethodManager(manifest, None)
     method = methods.get_method("config", ["materialized"])

tests/unit/utils/manifest.py

@@ -1001,6 +1001,6 @@ def manifest(
         disabled={},
         selectors={},
         groups={},
-        metadata=ManifestMetadata(adapter_type="postgres"),
+        metadata=ManifestMetadata(adapter_type="postgres", project_name="pkg"),
     )
     return manifest