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

lint: fix double quotes, fix long lines, add return type hint

Browse source
This commit is contained in:
Roger Billerey-Mosier 2023-06-27 15:58:13 -07:00
parent 947f47a221
commit 3f29b57820
1 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

15
pre_commit_hooks/requirements_txt_fixer.py
View file

@ -12,7 +12,7 @@ FAIL = 1
class Requirement: class Requirement:
UNTIL_COMPARISON = re.compile(b'={2,3}|!=|~=|>=?|<=?') UNTIL_COMPARISON = re.compile(b'={2,3}|!=|~=|>=?|<=?')
VERSION_MATCHER = re.compile(b'(?:={2,3}|!=|~=|>=?|<=?|@)\s*(?P<version>[A-Za-z0-9./:]+)$') VERSION_MATCHER = re.compile(br'(?:={2,3}|!=|~=|>=?|<=?|@)\s*(?P<version>[A-Za-z0-9./:]+)$') # noqa: E501
UNTIL_SEP = re.compile(rb'[^;\s]+') UNTIL_SEP = re.compile(rb'[^;\s]+')
def __init__(self) -> None: def __init__(self) -> None:
@ -41,7 +41,9 @@ class Requirement:
def has_version(self) -> bool: def has_version(self) -> bool:
return self.extract_version() is not None return self.extract_version() is not None
def extract_version(self): def extract_version(self) -> str | None:
if not self.value:
return None
matches = self.VERSION_MATCHER.search(self.value) matches = self.VERSION_MATCHER.search(self.value)
if matches: if matches:
self.version = matches.groups()[0].decode() self.version = matches.groups()[0].decode()
@ -145,7 +147,7 @@ def fix_requirements(f: IO[bytes], require_version: bool = False) -> int:
outcome = PASS outcome = PASS
if len(missing_versions) > 0: if len(missing_versions) > 0:
print("Missing versions in:", ", ".join(missing_versions)) print('Missing versions in:', ', '.join(missing_versions))
outcome = FAIL outcome = FAIL
if before_string != after_string: if before_string != after_string:
@ -160,11 +162,12 @@ def fix_requirements(f: IO[bytes], require_version: bool = False) -> int:
def main(argv: Sequence[str] | None = None) -> int: def main(argv: Sequence[str] | None = None) -> int:
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('filenames', nargs='*', help='Filenames to fix') parser.add_argument('filenames', nargs='*', help='Filenames to fix')
parser.add_argument('-r', '--require_version', parser.add_argument(
'-r', '--require_version',
required=False, required=False,
help='Use this to require each requirement to include a version number', help='Each requirement must include a version number',
action='store_true', action='store_true',
default=False default=False,
) )
args = parser.parse_args(argv) args = parser.parse_args(argv)