github/pipreqs
1
1
Fork 0
mirror of https://github.com/bndr/pipreqs.git synced 2025-06-06 03:25:21 +00:00
Code Issues 3 Packages Projects Releases Wiki Activity

Merge pull request #41 from dmand/fix-duplicate-dependencies

Browse Source 
Make pipreqs not write diplicate lines to requirements.txt file
This commit is contained in:
Vadim Kravcenko 2016-02-10 19:07:19 +01:00
commit bd3f208c59
3 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pipreqs/pipreqs.py
View File

@ -182,7 +182,8 @@ def get_pkg_names(pkgs):
if item[0] == pkg:
toappend = item[1]
break
result.append(toappend)
if toappend not in result:
result.append(toappend)
return result

6
tests/_data_duplicated_deps/db.py Normal file
View File

@ -0,0 +1,6 @@
import pymongo
from bson.objectid import ObjectId
# 'bson' package is mapped to 'pymongo'.
# But running pipreqs should not result in two duplicated
# lines 'pymongo==x.x.x'.

7
tests/test_pipreqs.py
View File

@ -26,6 +26,7 @@ class TestPipreqs(unittest.TestCase):
self.project = os.path.join(os.path.dirname(__file__), "_data")
self.project_invalid = os.path.join(os.path.dirname(__file__), "_invalid_data")
self.project_with_ignore_directory = os.path.join(os.path.dirname(__file__), "_data_ignore")
self.project_with_duplicated_deps = os.path.join(os.path.dirname(__file__), "_data_duplicated_deps")
self.requirements_path = os.path.join(self.project, "requirements.txt")
self.alt_requirement_path = os.path.join(
self.project, "requirements2.txt")
@ -43,6 +44,12 @@ class TestPipreqs(unittest.TestCase):
self.assertFalse("django" in imports)
self.assertFalse("models" in imports)
def test_deduplicate_dependencies(self):
imports = pipreqs.get_all_imports(self.project_with_duplicated_deps)
pkgs = pipreqs.get_pkg_names(imports)
self.assertEqual(len(pkgs), 1)
self.assertTrue("pymongo" in pkgs)
def test_invalid_python(self):
"""
Test that invalid python files cannot be imported.