From cece25b0be1f09f6ab9d53d099bb244858928f57 Mon Sep 17 00:00:00 2001 From: Dirkjan Ochtman Date: Thu, 10 Aug 2017 07:38:00 +0200 Subject: Implement Display for all Templates --- askama_derive/src/generator.rs | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'askama_derive/src') diff --git a/askama_derive/src/generator.rs b/askama_derive/src/generator.rs index b412c9d..713d9b6 100644 --- a/askama_derive/src/generator.rs +++ b/askama_derive/src/generator.rs @@ -47,6 +47,7 @@ pub fn generate(ast: &syn::DeriveInput, path: &str, mut nodes: Vec) -> Str } else { gen.impl_template(ast, &content); } + gen.impl_display(ast); gen.result() } @@ -490,6 +491,15 @@ impl<'a> Generator<'a> { self.writeln("}"); } + // Implement `Display` for the given context struct. + fn impl_display(&mut self, ast: &syn::DeriveInput) { + self.write_header(ast, "::std::fmt::Display"); + self.writeln("fn fmt(&self, f: &mut ::std::fmt::Formatter) -> Result<(), ::std::fmt::Error> {"); + self.writeln("self.render_into(f)"); + self.writeln("}"); + self.writeln("}"); + } + // Implement `Deref` for an inheriting context struct. fn deref_to_parent(&mut self, ast: &syn::DeriveInput, parent_type: &syn::Ty) { self.write_header(ast, "::std::ops::Deref"); -- cgit