setup.py cleanup

1 files changed, 6 insertions, 5 deletions

diff --git a/setup.py b/setup.py
index 458b5bfe7..f61a0f2a4 100644
--- a/setup.py
+++ b/setup.py
@@ -62,16 +62,17 @@ def get_description(file: Path) -> str:
 def get_requirements(file: Path) -> List[str]:
     requirements = []
     with file.open("r") as fh:
-        for line in fh.readlines():
-            line = line.strip()
+        for line in fh.read().splitlines():
             if line.startswith("#") or line == "":
                 continue
             elif line.startswith("-r"):
-                filename = line[3:].strip()
+                # Remove the first word/argument (-r)
+                filename = " ".join(line.split(" ")[1:])
                 requirements += get_requirements(file.parent / filename)
             elif line.startswith("https"):
-                _splitItems = line.split("#")
-                requirements.append("{} @ {}".format(_splitItems[1], _splitItems[0]))
+                # Convert 'URL#NAME' to 'NAME @ URL'
+                splitItems = line.split("#")
+                requirements.append("{} @ {}".format(splitItems[1], splitItems[0]))
             else:
                 requirements.append(line)
     return requirements