fix(requirements-txt-fixer): handle -i and --index-url= forms

Author: Louis Shawn

pre_commit_hooks/requirements_txt_fixer.py

@@ -34,6 +34,10 @@ def name(self) -> bytes:
         assert m is not None
 
         name = m.group()
+        if name == b'-i' or name.startswith(b'--index-url='):
+            return b'--index-url'
+        elif name.startswith(b'--extra-index-url='):
+            return b'--extra-index-url'
         m = self.UNTIL_COMPARISON.search(name)
         if not m:
             return name

tests/requirements_txt_fixer_test.py

@@ -116,6 +116,24 @@
             b'--extra-index-url https://example.com/simple\n'
             b'requests\n',
         ),
+        (
+            b'--extra-index-url https://example.com/simple\n'
+            b'-i https://pypi.org/simple\n'
+            b'requests\n',
+            FAIL,
+            b'-i https://pypi.org/simple\n'
+            b'--extra-index-url https://example.com/simple\n'
+            b'requests\n',
+        ),
+        (
+            b'--extra-index-url=https://example.com/simple\n'
+            b'--index-url=https://pypi.org/simple\n'
+            b'requests\n',
+            FAIL,
+            b'--index-url=https://pypi.org/simple\n'
+            b'--extra-index-url=https://example.com/simple\n'
+            b'requests\n',
+        ),
     ),
 )
 def test_integration(input_s, expected_retval, output, tmpdir):