Changeset bc798fc for remit/finance_core/models.py

Timestamp:

Apr 18, 2010, 4:58:22 PM (15 years ago)

Author:

Alex Dehnert <adehnert@…>

Branches:

master, client

Children:

074fac7

Parents:

5a72e32

git-author:

Alex Dehnert <adehnert@…> (04/18/10 16:58:22)

git-committer:

Alex Dehnert <adehnert@…> (04/18/10 16:58:22)

Message:

Allow getting nodes by path relative to a base

File:

• remit/finance_core/models.py (modified) (2 diffs)

remit/finance_core/models.py

@@ -56 +56 @@
 
     @classmethod
-    def get_by_path(cls, path):
-        try:
-            root = BudgetArea.objects.get(name=path[0], depth=1)
-        except IndexError, e:
-            raise KeyError(e)
-        node = root
-        for name in path[1:]:
+    def get_by_path(cls, path, base=None, ):
+        if base:
+            node = base
+        else:
+            try:
+                root = BudgetArea.objects.get(name=path[0], depth=1)
+            except IndexError, e:
+                raise KeyError(e)
+            node = root
+            path = path[1:]
+        for name in path:
             try:
                 node = node.get_children().filter(name=name)[0]
@@ -70 +74 @@
 
     @classmethod
-    def get_by_pathstr(cls, path):
+    def get_by_pathstr(cls, path, base=None):
         path = path.split('.')
-        return cls.get_by_path(path)
+        return cls.get_by_path(path, base=base, )
 
     def pathstr(self, skip=0):