zfmt

brimdata · May 14, 2024 · eb58396 · eb58396
1 parent 9696c22
commit eb58396
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/zfmt/ast.go b/zfmt/ast.go
@@ -207,6 +207,19 @@ func (c *canon) expr(e ast.Expr, parent string) {
 		c.ret()
 		c.flush()
 		c.write(")")
+	case *ast.FString:
+		c.write("f\"")
+		for _, elem := range e.Elems {
+			switch elem := elem.(type) {
+			case *ast.FStringExpr:
+				c.write("{")
+				c.expr(elem.Expr, "")
+				c.write("}")
+			case *ast.FStringText:
+				c.write(elem.Text)
+			}
+		}
+		c.write("\"")
 	default:
 		c.write("(unknown expr %T)", e)
 	}

diff --git a/zfmt/ztests/f-string.yaml b/zfmt/ztests/f-string.yaml
@@ -0,0 +1,7 @@
+script: |
+  zc -C 'yield f"hello {f"world"}"'
+
+outputs:
+  - name: stdout
+    data: |
+      yield f"hello {f"world"}"