getnames for scopes

pyfuzzyparser.py

@@ -15,7 +15,7 @@ of the python module 'tokenize'.
 This original codebase of this parser, which has been refactored and heavily
 changed, was programmed by Aaron Griffin <aaronmgriffin@gmail.com>.
 
-*The structure of the following script:*
+**The structure of the following script:**
 A Scope has
  - imports (Import)
  - subscopes (Scope, Class, Function, Flow)
@@ -97,6 +97,8 @@ class Scope(object):
 
     def add_docstr(self, str):
         """ Clean up a docstring """
+        # TODO the docstring clean isn't good, because things like
+        # r""" test """ are not being handled
         d = str.replace('\n', ' ')
         d = d.replace('\t', ' ')
         while d.find('  ') > -1:
@@ -139,6 +141,20 @@ class Scope(object):
             string = indent_block(string, indention=indention)
         return string
 
+    def get_names(self):
+        """
+        Get all the names, that are active and accessible in the current
+        scope.
+
+        :return: list of Name
+        :rtype: list
+        """
+        n = []
+        for stmt in self.statements:
+            n += stmt.get_names()
+        n += self.subscopes
+        return n
+
     def is_empty(self):
         """
         :return: True if there are no subscopes, imports and statements.
@@ -207,7 +223,10 @@ class Flow(Scope):
         super(Flow, self).__init__(name, indent, line_nr, '')
         self.command = command
         self.statement = statement
-        self.set_args = set_args
+        if set_args == None:
+            self.set_args = []
+        else:
+            self.set_args = set_args
         self.next = None
 
     def get_code(self, first_indent=False, indention="    "):
@@ -227,6 +246,17 @@ class Flow(Scope):
             str += self.next.get_code()
         return str
 
+    def get_names(self):
+        """
+        Get the names for the flow. This includes also a call to the super
+        class.
+        """
+        n = self.set_args
+        if self.next:
+            n += self.next.get_names()
+        n += super(Flow, self).get_names()
+        return n
+
     def set_next(self, next):
         """ Set the next element in the flow, those are else, except, etc. """
         if self.next:
@@ -342,6 +372,10 @@ class Statement(object):
         else:
             return self.code
 
+    def get_names(self):
+        """ Get the names for the statement. """
+        return self.set_vars
+
 
 class Name(object):
     """