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

Handled multiline dependencies

Browse source
This commit is contained in:
Aniket Bhatnagar 2020-05-07 21:02:12 +01:00 • committed by Anthony Sottile
parent 41e26ab636
commit bbcd31e000
2 changed files with 32 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

16
pre_commit_hooks/requirements_txt_fixer.py
View file

@ -34,6 +34,18 @@ class Requirement:
else:
return self.name < requirement.name
def is_complete(self) -> bool:
return (
self.value is not None and
not self.value.rstrip(b'\r\n').endswith(b'\\')
)
def append_value(self, value: bytes) -> None:
if self.value is not None:
self.value += value
else:
self.value = value
def fix_requirements(f: IO[bytes]) -> int:
requirements: List[Requirement] = []
@ -55,7 +67,7 @@ def fix_requirements(f: IO[bytes]) -> int:
# If the most recent requirement object has a value, then it's
# time to start building the next requirement object.
if not len(requirements) or requirements[-1].value is not None:
if not len(requirements) or requirements[-1].is_complete():
requirements.append(Requirement())
requirement = requirements[-1]
@ -73,7 +85,7 @@ def fix_requirements(f: IO[bytes]) -> int:
elif line.startswith(b'#') or line.strip() == b'':
requirement.comments.append(line)
else:
requirement.value = line
requirement.append_value(line)
# if a file ends in a comment, preserve it at the end
if requirements[-1].value is None: