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

requirements fixer output is similar to pip freeze

Browse source
This commit is contained in:
Vinay Karanam 2016-07-03 04:10:20 +05:30
parent 493665a5fc
commit c58ae08fed
2 changed files with 15 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

18
pre_commit_hooks/requirements_txt_fixer.py
View file

@ -10,6 +10,19 @@ class Requirement(object):
self.value = None
self.comments = []
@property
def name(self):
if self.value is None:
return
if self.value == b'\n':
return
if self.value.startswith(b'-e '):
return self.value.lower().partition(b'=')[-1]
return self.value.lower().partition(b'==')[0]
def __lt__(self, requirement):
# \n means top of file comment, so always return True,
# otherwise just do a string comparison with value.
@ -18,10 +31,7 @@ class Requirement(object):
elif requirement.value == b'\n':
return False
else:
return (
self.value.lower().partition(b'==') <
requirement.value.lower().partition(b'==')
)
return self.name < requirement.name
def fix_requirements(f):