From 4c1e9a68d8eabde2684b0f7973964106692d95d3 Mon Sep 17 00:00:00 2001 From: likang Date: Wed, 8 Oct 2014 15:16:05 +0800 Subject: [PATCH] fix requirements.txt parsing --- dev_server/bundle_local.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/dev_server/bundle_local.py b/dev_server/bundle_local.py index d5f1d5d..d3d9972 100755 --- a/dev_server/bundle_local.py +++ b/dev_server/bundle_local.py @@ -16,6 +16,7 @@ import sys import pip.util import shutil +import re TMP = 'virtualenv.bundle' @@ -44,10 +45,11 @@ def main(): # Get the dists in requirements.txt dists = [] - for line in open(options.requirements, 'r').readlines(): - if line.strip() or line.startswith('#'): - pass - pkg = line.split('==')[0] + for line in open(options.requirements, 'r'): + line = line.strip() + if not line or line.startswith('#'): + continue + pkg = re.split(r'=|>|<', line)[0] if pkg not in installed_dists: raise Exception('%s not installed' % pkg)