Merge pull request #8 from NarrativeScience/QPT-29657-updates-for-new…

…-deploy

QPT-29657-tweaking pypants for new py2sfn deploy

src/pypants/__init__.py

@@ -1,3 +1,3 @@
 """CLI for working with Python packages and BUILD files in a Pants monorepo"""
 
-__version__ = "0.4.0"
+__version__ = "0.5.0"

src/pypants/build_targets/py2sfn_project.py

@@ -24,7 +24,7 @@ def set_dependencies(self, targets: Dict[str, BuildTarget]) -> None:
 
         """
         logger.debug(f"Gathering dependencies for {self.key}")
-        for task_package_path in Path(self.package_path).glob("tasks/*/src/*"):
+        for task_package_path in Path(self.build_dir).glob("tasks/*/src/*"):
             package_name = task_package_path.name
             if package_name in targets:
                 self.dependencies.add(targets[package_name])

src/pypants/process_packages.py

@@ -5,7 +5,8 @@
 from multiprocessing import Pool
 import os
 from pathlib import Path
-from typing import Dict, List, Tuple
+import re
+from typing import Dict, List, Optional, Tuple
 
 import networkx as nx
 
@@ -101,17 +102,23 @@ def _gather_dependencies(self) -> None:
             for package_name in package_names:
                 target.add_dependency(self._targets, package_name)
 
-    def generate_build_files(self) -> None:
+    def generate_build_files(self, target_pattern: Optional[str] = None) -> None:
         """Generate BUILD files.
 
         You must have already called :py:meth:`.register_packages`.
+
+        Args:
+            target_pattern: If provided, BUILD files are only generated for targets with
+                keys matching the pattern
+
         """
         logger.info("Generating BUILD files")
         trees_to_write = []
-        for target in self._targets.values():
-            tree = target.generate_build_file()
-            if tree is not None:
-                trees_to_write.append((tree, target.build_file))
+        for key, target in self._targets.items():
+            if not target_pattern or re.search(target_pattern, key):
+                tree = target.generate_build_file()
+                if tree is not None:
+                    trees_to_write.append((tree, target.build_file))
 
         # Create a pool of workers that will render, format, and write each AST module
         # tree to disk.