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

Fix requirements-txt-fixer for comments at end of file

Browse source
This commit is contained in:
Anthony Sottile 2017-10-09 10:59:17 -07:00
parent 5d08e77ad5
commit 86691eda47
2 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
pre_commit_hooks/requirements_txt_fixer.py
View file

@ -63,9 +63,16 @@ def fix_requirements(f):
else:
requirement.value = line
# if a file ends in a comment, preserve it at the end
if requirements[-1].value is None:
rest = requirements.pop().comments
else:
rest = []
for requirement in sorted(requirements):
after.extend(requirement.comments)
after.append(requirement.value)
after.extend(rest)
after_string = b''.join(after)