From 3d87f2139258bf555e230a6bb04392f926e6ab71 Mon Sep 17 00:00:00 2001 From: kxrd Date: Fri, 2 Jun 2017 19:59:59 +0200 Subject: [PATCH] Add `file_` parameter to function `parse_requirements`, implement a try-except block in function `parse_requirements` --- pipreqs/pipreqs.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/pipreqs/pipreqs.py b/pipreqs/pipreqs.py index 4c3042e..9e9c312 100755 --- a/pipreqs/pipreqs.py +++ b/pipreqs/pipreqs.py @@ -222,12 +222,16 @@ def get_name_without_alias(name): def join(f): return os.path.join(os.path.dirname(__file__), f) -def parse_requirements(): +def parse_requirements(file_): modules = [] delim = ["<", ">", "=", "!", "~"] # https://www.python.org/dev/peps/pep-0508/#complete-grammar - with open("requirements.txt", "r") as f: - data = [x.strip() for x in f.readlines() if x != "\n"] + try: + with open(file_, "r") as f: + data = [x.strip() for x in f.readlines() if x != "\n"] + except OSError as e: + print(e) + sys.exit(0) parameters = [x for x in data if x.startswith("-")] data = [x for x in data if x[0].isalpha()]