From 6a5cd68a2883aa6f68f4563e51034dc42ff38fa6 Mon Sep 17 00:00:00 2001 From: Dave Schaefer Date: Tue, 10 Oct 2023 20:54:50 -0600 Subject: [PATCH] Move common sorting code to function --- pipenv/project.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/pipenv/project.py b/pipenv/project.py index f9257aa4..512f3644 100644 --- a/pipenv/project.py +++ b/pipenv/project.py @@ -1111,6 +1111,9 @@ class Project: return name return None + def _sort_category(self, category): + return dict(sorted(category.items())) + def remove_package_from_pipfile(self, package_name, category): # Read and append Pipfile. name = self.get_package_name_in_pipfile(package_name, category=category) @@ -1118,7 +1121,7 @@ class Project: if name: del p[category][name] if self.settings.get("sort_pipfile"): - p[category] = dict(sorted(p[category].items())) + p[category] = self._sort_category(p[category]) self.write_toml(p) return True return False @@ -1230,7 +1233,7 @@ class Project: p[category][normalized_name] = entry if self.settings.get("sort_pipfile"): - p[category] = dict(sorted(p[category].items())) + p[category] = self._sort_category(p[category]) # Write Pipfile. self.write_toml(p)