Move subclass grabbing code into a shared location

Summary: This just moves `all_subclasses` into the more appropriate `subclass_generator` file, and renames it `get_all_subclasses_from_environment` to disambiguate from the pyre related functions. This is being done to support use of this function in IG

Reviewed By: dkgi

Differential Revision: D30134858

fbshipit-source-id: 405523b599fc1c583859f686c1ab840a384fa42b

Author: gbleaney

tools/generate_taint_models/constructor_generator.py

@@ -9,17 +9,12 @@
 from importlib import import_module
 from typing import Callable, Iterable, List, Optional, Type, TypeVar
 
+from .subclass_generator import get_all_subclasses_from_environment as all_subclasses
 
 LOG: logging.Logger = logging.getLogger(__name__)
 T = TypeVar("T")
 
 
-def all_subclasses(parent_class: Type[T]) -> Iterable[Type[T]]:
-    return set(parent_class.__subclasses__()).union(
-        [s for c in parent_class.__subclasses__() for s in all_subclasses(c)]
-    )
-
-
 def gather_all_constructors_in_hierarchy(
     classes_to_taint: List[str],
     filter_classes_by: Optional[Callable[[Type[T]], bool]] = None,

tools/generate_taint_models/subclass_generator.py

@@ -6,7 +6,7 @@
 # pyre-strict
 
 import logging
-from typing import Dict, Iterable, List, Optional, Set
+from typing import TypeVar, Type, Dict, Iterable, List, Optional, Set
 
 from ...api import query
 from ...api.connection import PyreConnection
@@ -58,3 +58,16 @@ def get_all_subclass_defines_from_pyre(
         }
     else:
         return None
+
+
+T = TypeVar("T")
+
+
+def get_all_subclasses_from_environment(parent_class: Type[T]) -> Iterable[Type[T]]:
+    return set(parent_class.__subclasses__()).union(
+        [
+            grandchild
+            for child in parent_class.__subclasses__()
+            for grandchild in get_all_subclasses_from_environment(child)
+        ]
+    )