diff options
| author |  Dirkjan Ochtman <dirkjan@ochtman.nl> | 2018-07-22 14:59:56 +0100 |
|---|---|---|
| committer | Dirkjan Ochtman <dirkjan@ochtman.nl> | 2018-07-22 14:59:56 +0100 |
| commit | 8a25a1ee65031a687ad7d208eaa2843aed6c9752 (patch) | |
| tree | 83e16e2662a3182ea94989d436e9752d2ca54bdb /askama_derive/src/generator.rs | |
| parent | ed98793b506c66723c713991f40c0065557a35d5 (diff) | |
| download | askama-8a25a1ee65031a687ad7d208eaa2843aed6c9752.tar.gz askama-8a25a1ee65031a687ad7d208eaa2843aed6c9752.tar.bz2 askama-8a25a1ee65031a687ad7d208eaa2843aed6c9752.zip | |
Add Template method to return template extension
Thanks to Ryan McGrath for the suggestion and initial implementation.
Diffstat (limited to 'askama_derive/src/generator.rs')
| -rw-r--r-- | askama_derive/src/generator.rs | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/askama_derive/src/generator.rs b/askama_derive/src/generator.rs index 5af38c0..db9ac17 100644 --- a/askama_derive/src/generator.rs +++ b/askama_derive/src/generator.rs @@ -117,6 +117,11 @@ impl<'a> Generator<'a> { self.flush_ws(WS(false, false)); self.writeln("Ok(())"); self.writeln("}"); + + self.writeln("fn extension(&self) -> Option<&str> {"); + self.writeln(&format!("{:?}", self.input.path.extension().map(|s| s.to_str().unwrap()))); + self.writeln("}"); + self.writeln("}"); } |