diff --git a/tailbone/grids/core.py b/tailbone/grids/core.py index 8a3393c8..dce57d73 100644 --- a/tailbone/grids/core.py +++ b/tailbone/grids/core.py @@ -225,6 +225,8 @@ class Grid(object): self.set_renderer(key, self.render_quantity) elif type_ == 'duration': self.set_renderer(key, self.render_duration) + elif type_ == 'duration_hours': + self.set_renderer(key, self.render_duration_hours) else: raise ValueError("Unsupported type for column '{}': {}".format(key, type_)) @@ -302,6 +304,12 @@ class Grid(object): return "" return pretty_hours(datetime.timedelta(seconds=value)) + def render_duration_hours(self, obj, field): + value = self.obtain_value(obj, field) + if value is None: + return "" + return pretty_hours(hours=value) + def set_url(self, url): self.url = url