pre-commit/pre-commit-hooks
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit-hooks.git synced 2026-04-07 04:26:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Automatically fix package=version in requirements.txt

Browse source
This commit is contained in:
Paul Hooijenga 2018-10-23 21:26:37 +02:00
parent 86bfe8c475
commit f8b8997c07
2 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
pre_commit_hooks/requirements_txt_fixer.py
View file

@ -61,6 +61,9 @@ def fix_requirements(f):
elif line.startswith(b'#') or line.strip() == b'': elif line.startswith(b'#') or line.strip() == b'':
requirement.comments.append(line) requirement.comments.append(line)
else: else:
# Automatically fix 'package=version' to 'package==version'
if b'=' in line and b'==' not in line and b'egg=' not in line:
line = line.replace(b'=', b'==')
requirement.value = line requirement.value = line
# if a file ends in a comment, preserve it at the end # if a file ends in a comment, preserve it at the end

1
tests/requirements_txt_fixer_test.py
View file

@ -30,6 +30,7 @@ from pre_commit_hooks.requirements_txt_fixer import Requirement
), ),
(b'bar\npkg-resources==0.0.0\nfoo\n', FAIL, b'bar\nfoo\n'), (b'bar\npkg-resources==0.0.0\nfoo\n', FAIL, b'bar\nfoo\n'),
(b'foo\npkg-resources==0.0.0\nbar\n', FAIL, b'bar\nfoo\n'), (b'foo\npkg-resources==0.0.0\nbar\n', FAIL, b'bar\nfoo\n'),
(b'foo=1\n', FAIL, b'foo==1\n'),
), ),
) )
def test_integration(input_s, expected_retval, output, tmpdir): def test_integration(input_s, expected_retval, output, tmpdir):