naturalday-filter added.

README.md

@@ -30,6 +30,7 @@ All additional filters/tags will be registered automatically.
   - Humanize
      - **[intcomma](https://docs.djangoproject.com/en/dev/ref/contrib/humanize/#intcomma)** (put decimal marks into the number)
      - **[ordinal](https://docs.djangoproject.com/en/dev/ref/contrib/humanize/#ordinal)** (convert integer to its ordinal as string)
+     - **[naturalday](https://docs.djangoproject.com/en/dev/ref/contrib/humanize/#naturalday)** (converts `time.Time`-object into today/yesterday/tomorrow if possible; otherwise it will use `naturaltime`)
      - **[timesince](https://docs.djangoproject.com/en/dev/ref/templates/builtins/#timesince)/[timeuntil](https://docs.djangoproject.com/en/1.6/ref/templates/builtins/#timeuntil)/[naturaltime](https://docs.djangoproject.com/en/dev/ref/contrib/humanize/#naturaltime)** (human-readable time [duration] indicator)
 
 ### Tags

filters.go

@@ -7,8 +7,8 @@ import (
 	"github.com/flosch/pongo2"
 
 	"github.com/extemporalgenome/slug"
-	"github.com/russross/blackfriday"
 	"github.com/flosch/go-humanize"
+	"github.com/russross/blackfriday"
 )
 
 func init() {
@@ -18,6 +18,7 @@ func init() {
 	pongo2.RegisterFilter("timeuntil", filterTimeuntilTimesince)
 	pongo2.RegisterFilter("timesince", filterTimeuntilTimesince)
 	pongo2.RegisterFilter("naturaltime", filterTimeuntilTimesince)
+	pongo2.RegisterFilter("naturalday", filterNaturalday)
 	pongo2.RegisterFilter("intcomma", filterIntcomma)
 	pongo2.RegisterFilter("ordinal", filterOrdinal)
 }
@@ -37,13 +38,13 @@ func filterFilesizeformat(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value,
 func filterTimeuntilTimesince(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, error) {
 	basetime, is_time := in.Interface().(time.Time)
 	if !is_time {
-		return nil, errors.New("timeuntil-value is not a time.Time-instance.")
+		return nil, errors.New("time-value is not a time.Time-instance.")
 	}
 	var paramtime time.Time
 	if !param.IsNil() {
 		paramtime, is_time = param.Interface().(time.Time)
 		if !is_time {
-			return nil, errors.New("timeuntil-parameter is not a time.Time-instance.")
+			return nil, errors.New("time-parameter is not a time.Time-instance.")
 		}
 	} else {
 		paramtime = time.Now()
@@ -59,3 +60,35 @@ func filterIntcomma(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, error
 func filterOrdinal(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, error) {
 	return pongo2.AsValue(humanize.Ordinal(in.Integer())), nil
 }
+
+func filterNaturalday(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, error) {
+	basetime, is_time := in.Interface().(time.Time)
+	if !is_time {
+		return nil, errors.New("naturalday-value is not a time.Time-instance.")
+	}
+
+	var reference_time time.Time
+	if !param.IsNil() {
+		reference_time, is_time = param.Interface().(time.Time)
+		if !is_time {
+			return nil, errors.New("naturalday-parameter is not a time.Time-instance.")
+		}
+	} else {
+		reference_time = time.Now()
+	}
+
+	d := reference_time.Sub(basetime) / time.Hour
+
+	switch {
+	case d >= 0 && d < 24:
+		// Today
+		return pongo2.AsValue("today"), nil
+	case d >= 24:
+		return pongo2.AsValue("yesterday"), nil
+	case d < 0 && d >= -24:
+		return pongo2.AsValue("tomorrow"), nil
+	}
+
+	// Default behaviour
+	return pongo2.ApplyFilter("naturaltime", in, param)
+}

filters_test.go

@@ -1,8 +1,8 @@
 package pongo2addons
 
 import (
-	"time"
 	"testing"
+	"time"
 
 	. "gopkg.in/check.v1"
 
@@ -30,7 +30,7 @@ func (s *TestSuite1) TestFilters(c *C) {
 
 	// Timesince/timeuntil
 	base_date := time.Date(2014, time.February, 1, 8, 30, 00, 00, time.UTC)
-	future_date := base_date.Add(24 * 7 * 4 * time.Hour + 2 * time.Hour)
+	future_date := base_date.Add(24*7*4*time.Hour + 2*time.Hour)
 	c.Assert(pongo2.RenderTemplateString("{{ future_date|timeuntil:base_date }}",
 		pongo2.Context{"base_date": base_date, "future_date": future_date}), Equals, "4 weeks from now")
 
@@ -50,6 +50,20 @@ func (s *TestSuite1) TestFilters(c *C) {
 	c.Assert(pongo2.RenderTemplateString("{{ base_date|naturaltime:future_date }}",
 		pongo2.Context{"base_date": base_date, "future_date": future_date}), Equals, "4 seconds ago")
 
+	// Naturalday
+	today := time.Date(2014, time.February, 1, 8, 30, 00, 00, time.UTC)
+	yesterday := today.Add(-24 * time.Hour)
+	tomorrow := today.Add(24 * time.Hour)
+	today_plus_3 := today.Add(3 * 24 * time.Hour)
+	c.Assert(pongo2.RenderTemplateString("{{ date|naturalday:today }}",
+		pongo2.Context{"date": today, "today": today}), Equals, "today")
+	c.Assert(pongo2.RenderTemplateString("{{ date|naturalday:today }}",
+		pongo2.Context{"date": yesterday, "today": today}), Equals, "yesterday")
+	c.Assert(pongo2.RenderTemplateString("{{ date|naturalday:today }}",
+		pongo2.Context{"date": tomorrow, "today": today}), Equals, "tomorrow")
+	c.Assert(pongo2.RenderTemplateString("{{ date|naturalday:today }}",
+		pongo2.Context{"date": today_plus_3, "today": today}), Equals, "3 days from now")
+
 	// Intcomma
 	c.Assert(pongo2.RenderTemplateString("{{ 123456789|intcomma }}", nil), Equals, "123,456,789")