Add initial json.dumps support

README.md

@@ -4008,6 +4008,65 @@ func main() {
 	fmt.Println("The sum of", num1, "and", num2, "is", sum)
 }
 ```
+### jsondump
+#### Python
+```python
+import json
+import random
+
+
+def main():
+    print(json.dumps(1))
+    print(json.dumps("hello"))
+    c = json.dumps({"hello": 1, "how": "are you"})
+    print(c + c)
+    print(json.dumps([1, 2, 3]))
+
+
+if __name__ == '__main__':
+    main()
+```
+#### Go
+```go
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+)
+
+func main() {
+	fmt.Println(func() string {
+		b, err := json.Marshal(1)
+		if err != nil {
+			panic(err)
+		}
+		return string(b)
+	}())
+	fmt.Println(func() string {
+		b, err := json.Marshal("hello")
+		if err != nil {
+			panic(err)
+		}
+		return string(b)
+	}())
+	c := func() string {
+		b, err := json.Marshal(map[string]interface{}{"hello": 1, "how": "are you"})
+		if err != nil {
+			panic(err)
+		}
+		return string(b)
+	}()
+	fmt.Println(c + c)
+	fmt.Println(func() string {
+		b, err := json.Marshal([]int{1, 2, 3})
+		if err != nil {
+			panic(err)
+		}
+		return string(b)
+	}())
+}
+```
 ### algomajorityelement
 #### Python
 ```python

examples/jsondump.go

@@ -0,0 +1,38 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+)
+
+func main() {
+	fmt.Println(func() string {
+		b, err := json.Marshal(1)
+		if err != nil {
+			panic(err)
+		}
+		return string(b)
+	}())
+	fmt.Println(func() string {
+		b, err := json.Marshal("hello")
+		if err != nil {
+			panic(err)
+		}
+		return string(b)
+	}())
+	c := func() string {
+		b, err := json.Marshal(map[string]interface{}{"hello": 1, "how": "are you"})
+		if err != nil {
+			panic(err)
+		}
+		return string(b)
+	}()
+	fmt.Println(c + c)
+	fmt.Println(func() string {
+		b, err := json.Marshal([]int{1, 2, 3})
+		if err != nil {
+			panic(err)
+		}
+		return string(b)
+	}())
+}

examples/jsondump.py

@@ -0,0 +1,14 @@
+import json
+import random
+
+
+def main():
+    print(json.dumps(1))
+    print(json.dumps("hello"))
+    c = json.dumps({"hello": 1, "how": "are you"})
+    print(c + c)
+    print(json.dumps([1, 2, 3]))
+
+
+if __name__ == '__main__':
+    main()

pytago/go_ast/py_snippets.py

@@ -780,6 +780,15 @@ def go_json_dumps(m):  # pragma: no cover
     return string(b)
 
 
+obj = TypeVar("obj")
+@Bindable.add(r"json\.loads", bind_type=BindType.FUNC_LIT, results=['obj'])
+def go_json_loads(s: str) -> PytagoInterfaceType[obj]:
+    err = json.Unmarshal(bytes(s), '&' @ obj)
+    if err != nil:
+        panic(err)
+    return obj
+
+
 # Logging methods
 
 @Bindable.add(r"logging\.info", bind_type=BindType.EXPR)

pytago/go_ast/transformers.py

@@ -1800,6 +1800,17 @@ def visit_TypeSwitchStmt(self, node: TypeSwitchStmt):
         return node
 
 
+class RemoveConflictingImports(BaseTransformer):
+    REPEATABLE = False
+    def visit_GenDecl(self, node: GenDecl):
+        if len(node.Specs) == 1:
+            match node.Specs:
+                # If this is imported, encoding/json won't auto-import
+                case [ImportSpec(Path=BasicLit(Value='"json"'))]:
+                    return
+        return node
+
+
 ALL_TRANSFORMS = [
     #### STAGE 0 ####
     InsertUniqueInitializers,
@@ -1844,6 +1855,7 @@ def visit_TypeSwitchStmt(self, node: TypeSwitchStmt):
     NodeTransformerWithInterfaceTypes,
     TypeSwitchStatementsRedeclareWithType,
     RemoveUnnecessaryFunctionLiterals,
+    RemoveConflictingImports,
     RemoveGoCallReturns,  # Needs to be below scoping functions or they'll just get added back
     RemoveBadStmt,  # Should be last as these are used for scoping
     MergeAdjacentInits,

pytago/tests/test_core.py

@@ -211,6 +211,9 @@ def test_unpacking(self):
     def test_cast_to_float(self):
         self.assert_examples_match("cast_to_float")
 
+    def test_jsondump(self):
+        self.assert_examples_match("jsondump")
+
     # Algorithms
     def test_algomajorityelement(self):
         self.assert_examples_match("algomajorityelement")