Support more requirements things (#116)

builder/pip.py

@@ -76,9 +76,18 @@ def parse_requirements(requirement: Path) -> List[str]:
     with requirement.open("r") as data:
         for line in data:
             line = line.strip()
-            if not line or line.startswith("#"):
+
+            # Ignore comments or constraint files
+            if not line or line.startswith(("#", "-c")):
+                continue
+
+            if not line.startswith("-r"):
+                requirement_list.add(line.split(" ")[-1])
                 continue
-            requirement_list.add(line.split(" ")[-1])
+
+            # References new requirements file
+            requirement_list.update(parse_requirements(requirement.parent / line[3:]))
+
     return list(requirement_list)
 
 

requirements_tests.txt

@@ -1,3 +1,4 @@
 flake8==3.8.3
 pylint==2.5.3
 black==19.10b0
+pytest==5.4.3

tests/__init__.py

@@ -0,0 +1 @@
+"""Tests for wheel builder."""

tests/requirements/constraints.txt

@@ -0,0 +1,2 @@
+# Constraint file
+typing==1000000000000.0.0

tests/requirements/requirements.txt

@@ -0,0 +1,3 @@
+# Core requirements
+-c constraints.txt
+aiohttp==1.2.3

tests/requirements/requirements_all.txt

@@ -0,0 +1,6 @@
+# All requirements
+-c constraints.txt
+
+-r requirements.txt
+
+aiohue==5.6.7

tests/test_pip.py

@@ -0,0 +1,11 @@
+"""Tests for pip module."""
+from pathlib import Path
+from builder import pip
+
+
+def test_parse_requirements():
+    assert sorted(
+        pip.parse_requirements(
+            Path(__file__).parent / "requirements/requirements_all.txt"
+        )
+    ) == ["aiohttp==1.2.3", "aiohue==5.6.7"]

tox.ini

@@ -1,5 +1,5 @@
 [tox]
-envlist = lint, black
+envlist = lint, black, test
 
 [testenv]
 deps =
@@ -10,9 +10,13 @@ deps =
 basepython = python3
 ignore_errors = True
 commands =
-     flake8 builder
+     flake8 builder tests
      pylint --rcfile pylintrc builder
 
 [testenv:black]
 commands =
-    black --target-version py37 --check builder setup.py
+    black --target-version py37 --check builder tests setup.py
+
+[testenv:test]
+commands =
+    pytest tests