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

Merge pull request #243 from pre-commit/comment_only_requirements_txt

Browse source 
Fix requirements-txt-fixer for comments at end of file
This commit is contained in:
Anthony Sottile 2017-10-09 11:27:52 -07:00 • committed by GitHub
commit 361a6ac1bc
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)