From 6490d95835d8b7b43a6db65c9a1bb82a08bb2eed Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <kloeckner@haamster.(none)>
Date: Mon, 1 Dec 2008 17:46:58 -0600
Subject: [PATCH] DependencyMapper: Add option to only gather from function
 *arguments*.

---
 src/mapper/dependency.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/mapper/dependency.py b/src/mapper/dependency.py
index 389ddda..aea10c4 100644
--- a/src/mapper/dependency.py
+++ b/src/mapper/dependency.py
@@ -44,7 +44,10 @@ class DependencyMapper(CombineMapper):
         return set([expr])
 
     def map_call(self, expr):
-        if self.include_calls:
+        if self.include_calls == "descend_args":
+                return self.combine(
+                        [self.rec(child) for child in expr.parameters])
+        elif self.include_calls:
             return set([expr])
         else:
             return CombineMapper.map_call(self, expr)
-- 
GitLab