Descendants for categories

gively · Dec 29, 2011 · 5a6c241 · 5a6c241
1 parent af662ec
commit 5a6c241
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/lib/ntee.rb b/lib/ntee.rb
@@ -32,6 +32,10 @@ def ancestors
       end
     end
 
+    def descendants
+      (subcategories + subcategories.map(&:descendants)).flatten
+    end
+
     def add_subcategory!(subcategory)
       subcategories[subcategory.code.to_s] = subcategory
       subcategory.parent = self