parse srcs field in python_binary targets

Summary: It turns out that `python_binary` targets can specify `srcs` -- we should also handle this case.

Reviewed By: grievejia

Differential Revision: D14752547

fbshipit-source-id: 7bb75a11eb02d069f2b1b1e6dd22841ec0646a44

Author: Matt D'Souza

client/buck_project_builder/parser/build_target.py

@@ -77,13 +77,24 @@ def parse(call: ast.Call, build_file_directory: str) -> "NonPythonTarget":
 
 
 class PythonBinary(BuildTarget):
+    def __init__(
+        self,
+        build_file_directory: str,
+        name: str,
+        dependencies: List[str],
+        sources: Optional[List[str]] = None,
+    ) -> None:
+        super(PythonBinary, self).__init__(build_file_directory, name, dependencies)
+        self.sources = sources or []  # type: List[str]
+
     def rule_name(self) -> str:
         return "python_binary"
 
     @staticmethod
     def parse(call: ast.Call, build_file_directory: str) -> "PythonBinary":
         base = BuildTarget.parse_base_information(call, build_file_directory)
-        return PythonBinary(build_file_directory, base.name, base.dependencies)
+        sources = _get_sources(base.keywords.get("srcs"))
+        return PythonBinary(build_file_directory, base.name, base.dependencies, sources)
 
 
 class PythonLibrary(BuildTarget):

client/buck_project_builder/parser/tests/build_target_test.py

@@ -35,7 +35,8 @@ def _get_call(tree: ast.AST) -> ast.Call:
 PYTHON_BINARY_TARGET_2 = """
 python_binary(
     name = "binary_target_2",
-    deps = []
+    deps = [],
+    srcs = ["a.py"],
 )
 """
 
@@ -96,13 +97,15 @@ def test_python_binary(self):
             target.dependencies,
             ["//some/project:another_target", "//some/other:target"],
         )
+        self.assertListEqual(target.sources, [])
 
         tree = ast.parse(PYTHON_BINARY_TARGET_2)
         call = _get_call(tree)
         target = PythonBinary.parse(call, "some/project")
         self.assertEqual(target.target, "//some/project:binary_target_2")
         self.assertEqual(target.name, "binary_target_2")
         self.assertListEqual(target.dependencies, [])
+        self.assertListEqual(target.sources, ["a.py"])
 
         tree = ast.parse(PYTHON_BINARY_TARGET_3)
         call = _get_call(tree)