diff options
| -rw-r--r-- | askama/src/generator.rs | 14 |
1 files changed, 6 insertions, 8 deletions
diff --git a/askama/src/generator.rs b/askama/src/generator.rs index 3c5e2a1..f262f7c 100644 --- a/askama/src/generator.rs +++ b/askama/src/generator.rs @@ -173,21 +173,21 @@ impl Generator { res } - fn init(&mut self, name: &str, generics: &syn::Generics) { + fn template_impl(&mut self, ast: &syn::DeriveInput, tokens: &Vec<Node>) { self.write("impl"); - let anno = self.annotations(generics); + let anno = self.annotations(&ast.generics); self.write(&anno); self.write(" askama::Template for "); - self.write(name); + self.write(ast.ident.as_ref()); self.write(&anno); self.writeln(" {"); self.indent(); self.writeln("fn render_into(&self, writer: &mut std::fmt::Write) {"); self.indent(); - } - fn finalize(&mut self) { + self.handle(tokens); + self.dedent(); self.writeln("}"); self.dedent(); @@ -202,8 +202,6 @@ impl Generator { pub fn generate(ast: &syn::DeriveInput, tokens: &Vec<Node>) -> String { let mut gen = Generator::new(); - gen.init(ast.ident.as_ref(), &ast.generics); - gen.handle(tokens); - gen.finalize(); + gen.template_impl(ast, tokens); gen.result() } |