From 4f4e87729b61da591bcf9e7666e0e7cb54d90bb0 Mon Sep 17 00:00:00 2001
From: Jeong Yunwon <jeong@youknowone.org>
Date: Sun, 17 Apr 2022 02:18:42 +0900
Subject: [PATCH] Fix ast/asdl_rs.py

---
 ast/asdl_rs.py           | 27 +++++++++++++++------------
 ast/src/ast_gen.rs       |  1 -
 scripts/update_asdl.sh   |  3 +--
 vm/src/stdlib/ast.rs     |  7 ++-----
 vm/src/stdlib/ast/gen.rs |  2 ++
 5 files changed, 20 insertions(+), 20 deletions(-)

diff --git a/ast/asdl_rs.py b/ast/asdl_rs.py
index b12652a97..af50e5242 100755
--- a/ast/asdl_rs.py
+++ b/ast/asdl_rs.py
@@ -1,11 +1,9 @@
 #! /usr/bin/env python
 """Generate Rust code from an ASDL description."""
 
-import os
 import sys
-import textwrap
-
 import json
+import textwrap
 
 from argparse import ArgumentParser
 from pathlib import Path
@@ -13,7 +11,7 @@ from pathlib import Path
 import asdl
 
 TABSIZE = 4
-AUTOGEN_MESSAGE = "// File automatically generated by {}.\n\n"
+AUTOGEN_MESSAGE = "// File automatically generated by {}.\n"
 
 builtin_type_mapping = {
     'identifier': 'Ident',
@@ -390,9 +388,9 @@ class ClassDefVisitor(EmitVisitor):
         self.emit(f"impl {structname} {{", depth)
         self.emit(f"#[extend_class]", depth + 1)
         self.emit("fn extend_class_with_fields(ctx: &PyContext, class: &PyTypeRef) {", depth + 1)
-        fields = ",".join(f"ctx.new_str({json.dumps(f.name)})" for f in fields)
+        fields = ",".join(f"ctx.new_str(ascii!({json.dumps(f.name)})).into()" for f in fields)
         self.emit(f'class.set_str_attr("_fields", ctx.new_list(vec![{fields}]));', depth + 2)
-        attrs = ",".join(f"ctx.new_str({json.dumps(attr.name)})" for attr in attrs)
+        attrs = ",".join(f"ctx.new_str(ascii!({json.dumps(attr.name)})).into()" for attr in attrs)
         self.emit(f'class.set_str_attr("_attributes", ctx.new_list(vec![{attrs}]));', depth + 2)
         self.emit("}", depth + 1)
         self.emit("}", depth)
@@ -401,7 +399,7 @@ class ExtendModuleVisitor(EmitVisitor):
 
     def visitModule(self, mod):
         depth = 0
-        self.emit("pub fn extend_module_nodes(vm: &VirtualMachine, module: &crate::PyObj) {", depth)
+        self.emit("pub fn extend_module_nodes(vm: &VirtualMachine, module: &PyObject) {", depth)
         self.emit("extend_module!(vm, module, {", depth + 1)
         for dfn in mod.dfns:
             self.visit(dfn, depth + 2)
@@ -488,7 +486,7 @@ class TraitImplVisitor(EmitVisitor):
             self.emit("let _dict = _node.as_object().dict().unwrap();", depth)
         for f in fields:
             self.emit(f"_dict.set_item({json.dumps(f.name)}, {rust_field(f.name)}.ast_to_object(_vm), _vm).unwrap();", depth)
-        self.emit("_node.into_object()", depth)
+        self.emit("_node.into()", depth)
 
     def make_pattern(self, fields):
         return ",".join(rust_field(f.name) for f in fields)
@@ -505,7 +503,7 @@ class TraitImplVisitor(EmitVisitor):
             self.emit("} else", depth)
 
         self.emit("{", depth)
-        msg = f'format!("expected some sort of {sumname}, but got {{}}",_vm.to_repr(&_object)?)'
+        msg = f'format!("expected some sort of {sumname}, but got {{}}",_object.repr(_vm)?)'
         self.emit(f"return Err(_vm.new_type_error({msg}));", depth + 1)
         self.emit("})", depth)
 
@@ -549,8 +547,8 @@ class ChainOfVisitors:
 
 
 def write_ast_def(mod, typeinfo, f):
-    f.write('pub use crate::location::Location;\n')
     f.write('pub use crate::constant::*;\n')
+    f.write('pub use crate::location::Location;\n')
     f.write('\n')
     f.write('type Ident = String;\n')
     f.write('\n')
@@ -574,8 +572,13 @@ def write_ast_def(mod, typeinfo, f):
 
 
 def write_ast_mod(mod, f):
-    f.write('use super::*;\n')
-    f.write('\n')
+    f.write(textwrap.dedent("""
+        #![allow(clippy::all)]
+
+        use super::*;
+        use crate::common::ascii;
+
+    """))
 
     c = ChainOfVisitors(ClassDefVisitor(f),
                         TraitImplVisitor(f),
diff --git a/ast/src/ast_gen.rs b/ast/src/ast_gen.rs
index 31380e087..0e40dc4a1 100644
--- a/ast/src/ast_gen.rs
+++ b/ast/src/ast_gen.rs
@@ -1,5 +1,4 @@
 // File automatically generated by ast/asdl_rs.py.
-
 pub use crate::constant::*;
 pub use crate::location::Location;
 
diff --git a/scripts/update_asdl.sh b/scripts/update_asdl.sh
index 148e5ca60..4373b9b62 100755
--- a/scripts/update_asdl.sh
+++ b/scripts/update_asdl.sh
@@ -4,5 +4,4 @@ set -e
 cd "$(dirname "$(dirname "$0")")"
 
 python ast/asdl_rs.py -D ast/src/ast_gen.rs -M vm/src/stdlib/ast/gen.rs ast/Python.asdl
-
-cargo fmt
+rustfmt ast/src/ast_gen.rs vm/src/stdlib/ast/gen.rs
diff --git a/vm/src/stdlib/ast.rs b/vm/src/stdlib/ast.rs
index 72dfd0dc9..817857edb 100644
--- a/vm/src/stdlib/ast.rs
+++ b/vm/src/stdlib/ast.rs
@@ -3,6 +3,8 @@
 //! This module makes use of the parser logic, and translates all ast nodes
 //! into python ast.AST objects.
 
+mod gen;
+
 use crate::{
     builtins::{self, PyStrRef, PyTypeRef},
     pyclass::{PyClassImpl, StaticType},
@@ -17,11 +19,6 @@ use rustpython_compiler as compile;
 #[cfg(feature = "rustpython-parser")]
 use rustpython_parser::parser;
 
-#[rustfmt::skip]
-#[allow(clippy::all)]
-mod gen;
-
-
 #[pymodule]
 mod _ast {
     use crate::{
diff --git a/vm/src/stdlib/ast/gen.rs b/vm/src/stdlib/ast/gen.rs
index f7149e178..f34f2d974 100644
--- a/vm/src/stdlib/ast/gen.rs
+++ b/vm/src/stdlib/ast/gen.rs
@@ -1,5 +1,7 @@
 // File automatically generated by ast/asdl_rs.py.
 
+#![allow(clippy::all)]
+
 use super::*;
 use crate::common::ascii;