Implement Display for all Templates

1 files changed, 10 insertions, 0 deletions

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<Node>) -> 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<Parent>` 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");