[1/3][Asset Selection] Remove key_substring + support wildcards in key:

python_modules/dagster/dagster/_core/definitions/antlr_asset_selection/AssetSelection.g4

@@ -27,8 +27,7 @@ functionName: SINKS | ROOTS;
 
 // Attribute expressions for specific attributes
 attributeExpr:
-	KEY COLON value							# KeyExpr
-	| KEY_SUBSTRING COLON value				# KeySubstringExpr
+	KEY COLON keyValue						# KeyExpr
 	| TAG COLON value (EQUAL value)?		# TagAttributeExpr
 	| OWNER COLON value						# OwnerAttributeExpr
 	| GROUP COLON value						# GroupAttributeExpr
@@ -44,6 +43,10 @@ EQUAL: '=';
 
 // Value can be a quoted or unquoted string
 value: QUOTED_STRING | UNQUOTED_STRING;
+keyValue:
+	QUOTED_STRING
+	| UNQUOTED_STRING
+	| UNQUOTED_REGEX_STRING;
 
 // Operators and keywords
 AND: 'and';
@@ -64,7 +67,6 @@ COMMA: ',';
 
 // Attributes
 KEY: 'key';
-KEY_SUBSTRING: 'key_substring';
 OWNER: 'owner';
 GROUP: 'group';
 TAG: 'tag';
@@ -82,6 +84,7 @@ ROOTS: 'roots';
 // String tokens
 QUOTED_STRING: '"' (~["\\\r\n])* '"';
 UNQUOTED_STRING: [a-zA-Z_][a-zA-Z0-9_]*;
+UNQUOTED_REGEX_STRING: [a-zA-Z_*][a-zA-Z0-9_*]*;
 
 // Whitespace
 WS: [ \t\r\n]+ -> skip;

python_modules/dagster/dagster/_core/definitions/antlr_asset_selection/antlr_asset_selection.py

@@ -16,6 +16,7 @@
     CodeLocationAssetSelection,
     ColumnAssetSelection,
     ColumnTagAssetSelection,
+    KeyWildCardAssetSelection,
     TableNameAssetSelection,
 )
 from dagster._core.storage.tags import KIND_PREFIX
@@ -113,12 +114,8 @@ def visitFunctionName(self, ctx: AssetSelectionParser.FunctionNameContext):
             return "roots"
 
     def visitKeyExpr(self, ctx: AssetSelectionParser.KeyExprContext):
-        value = self.visit(ctx.value())
-        return AssetSelection.assets(value)
-
-    def visitKeySubstringExpr(self, ctx: AssetSelectionParser.KeySubstringExprContext):
-        value = self.visit(ctx.value())
-        return AssetSelection.key_substring(value)
+        value = self.visit(ctx.keyValue())
+        return KeyWildCardAssetSelection(selected_key_wildcard=value)
 
     def visitTagAttributeExpr(self, ctx: AssetSelectionParser.TagAttributeExprContext):
         key = self.visit(ctx.value(0))
@@ -145,6 +142,14 @@ def visitCodeLocationAttributeExpr(
         code_location = self.visit(ctx.value())
         return CodeLocationAssetSelection(selected_code_location=code_location)
 
+    def visitKeyValue(self, ctx: AssetSelectionParser.KeyValueContext):
+        if ctx.QUOTED_STRING():
+            return ctx.QUOTED_STRING().getText().strip('"')
+        elif ctx.UNQUOTED_REGEX_STRING():
+            return ctx.UNQUOTED_REGEX_STRING().getText()
+        elif ctx.UNQUOTED_STRING():
+            return ctx.UNQUOTED_STRING().getText()
+
     def visitValue(self, ctx: AssetSelectionParser.ValueContext):
         if ctx.QUOTED_STRING():
             return ctx.QUOTED_STRING().getText().strip('"')