From 151d485a81f3ee00f5943b84165ca9b157f97d73 Mon Sep 17 00:00:00 2001 From: Anthony Sottile Date: Thu, 12 Feb 2015 07:51:34 -0800 Subject: [PATCH] More naturally sort requirements. --- pre_commit_hooks/requirements_txt_fixer.py | 5 ++++- tests/requirements_txt_fixer_test.py | 1 + 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/pre_commit_hooks/requirements_txt_fixer.py b/pre_commit_hooks/requirements_txt_fixer.py index 2dec6bf..cc09242 100644 --- a/pre_commit_hooks/requirements_txt_fixer.py +++ b/pre_commit_hooks/requirements_txt_fixer.py @@ -18,7 +18,10 @@ class Requirement(object): elif requirement.value == b'\n': return False else: - return self.value < requirement.value + return ( + self.value.partition(b'==') < + requirement.value.partition(b'==') + ) def fix_requirements(f): diff --git a/tests/requirements_txt_fixer_test.py b/tests/requirements_txt_fixer_test.py index b135fdf..3eac637 100644 --- a/tests/requirements_txt_fixer_test.py +++ b/tests/requirements_txt_fixer_test.py @@ -15,6 +15,7 @@ TESTS = ( (b'#comment\n\nbar\nfoo\n', 0, b'#comment\n\nbar\nfoo\n'), (b'\nfoo\nbar\n', 1, b'bar\n\nfoo\n'), (b'\nbar\nfoo\n', 0, b'\nbar\nfoo\n'), + (b'pyramid==1\npyramid-foo==2\n', 0, b'pyramid==1\npyramid-foo==2\n'), )