python: fix generated contract methods (#7)

CityOfZion · Apr 21, 2023 · f7bf5ef · f7bf5ef
1 parent 5d23770
commit f7bf5ef
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/generators/python.go b/generators/python.go
@@ -27,14 +27,16 @@ import (
 const pythonSrcTmpl = `
 {{- define "METHOD" }}
     @staticmethod
+    {{- if ne .NameABI .Name}}
+    @display_name("{{.NameABI }}"){{end}}
     def {{.Name}}({{range $index, $arg := .Arguments -}}
        {{- if ne $index 0}}, {{end}}
           {{- .Name}}: {{.Type}}
        {{- end}}) -> {{if .ReturnType }}{{ .ReturnType }}: {{ else }} None: {{ end }}
         pass
 {{- end -}}
 from boa3.builtin.type import UInt160, UInt256, ECPoint
-from boa3.builtin.compile_time import contract
+from boa3.builtin.compile_time import contract, display_name
 from typing import cast, Any