diff options
Diffstat (limited to 'askama_shared/src')
-rw-r--r-- | askama_shared/src/filters/mod.rs | 19 | ||||
-rw-r--r-- | askama_shared/src/lib.rs | 1 |
2 files changed, 19 insertions, 1 deletions
diff --git a/askama_shared/src/filters/mod.rs b/askama_shared/src/filters/mod.rs index 35166ff..672c5b3 100644 --- a/askama_shared/src/filters/mod.rs +++ b/askama_shared/src/filters/mod.rs @@ -12,6 +12,7 @@ pub use self::json::json; use askama_escape::MarkupDisplay; use error::Error::Fmt; +use humansize::{file_size_opts, FileSize}; use num_traits::cast::NumCast; use num_traits::Signed; use std::fmt; @@ -22,12 +23,13 @@ use super::Result; // Askama or should refer to a local `filters` module. It should contain all the // filters shipped with Askama, even the optional ones (since optional inclusion // in the const vector based on features seems impossible right now). -pub const BUILT_IN_FILTERS: [&str; 21] = [ +pub const BUILT_IN_FILTERS: [&str; 22] = [ "abs", "capitalize", "center", "e", "escape", + "filesizeformat", "format", "indent", "into_f64", @@ -77,6 +79,12 @@ where escape(i) } +/// Returns adequate string representation (in KB, ..) of number of bytes +pub fn filesizeformat(b: usize) -> Result<String> { + b.file_size(file_size_opts::DECIMAL) + .map_err(|_| Fmt(fmt::Error)) +} + /// Formats arguments according to the specified format /// /// The first argument to this filter must be a string literal (as in normal @@ -268,6 +276,15 @@ mod tests { use std::f64::INFINITY; #[test] + fn test_filesizeformat() { + assert_eq!(filesizeformat(0).unwrap(), "0 B"); + assert_eq!(filesizeformat(999).unwrap(), "999 B"); + assert_eq!(filesizeformat(1000).unwrap(), "1 KB"); + assert_eq!(filesizeformat(1023).unwrap(), "1.02 KB"); + assert_eq!(filesizeformat(1024).unwrap(), "1.02 KB"); + } + + #[test] fn test_linebreaks() { assert_eq!( linebreaks(&"Foo\nBar Baz").unwrap(), diff --git a/askama_shared/src/lib.rs b/askama_shared/src/lib.rs index 9e2ba6c..a8839fa 100644 --- a/askama_shared/src/lib.rs +++ b/askama_shared/src/lib.rs @@ -1,6 +1,7 @@ #![cfg_attr(feature = "cargo-clippy", allow(unused_parens))] extern crate askama_escape; +extern crate humansize; extern crate num_traits; extern crate serde; #[macro_use] |