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

Parse more operators in requirements

Browse source
This commit is contained in:
Max Rozentsveyg 2020-05-16 23:58:27 -04:00
parent 3d379a962d
commit 7ebd420417
2 changed files with 37 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

17
pre_commit_hooks/requirements_txt_fixer.py
View file

@ -1,4 +1,5 @@
import argparse
import re
from typing import IO
from typing import List
from typing import Optional
@ -10,6 +11,9 @@ FAIL = 1
class Requirement:
UNTIL_COMPARISON = re.compile(b'={2,3}|!=|~=|>=?|<=?')
UNTIL_SEP = re.compile(rb'[^;\s]+')
def __init__(self) -> None:
self.value: Optional[bytes] = None
self.comments: List[bytes] = []
@ -17,11 +21,20 @@ class Requirement:
@property
def name(self) -> bytes:
assert self.value is not None, self.value
name = self.value.lower()
for egg in (b'#egg=', b'&egg='):
if egg in self.value:
return self.value.lower().partition(egg)[-1]
return name.partition(egg)[-1]
return self.value.lower().partition(b'==')[0]
m = self.UNTIL_SEP.match(name)
assert m is not None
name = m.group()
m = self.UNTIL_COMPARISON.search(name)
if not m:
return name
return name[:m.start()]
def __lt__(self, requirement: 'Requirement') -> int:
# \n means top of file comment, so always return True,