Skip to content

Commit 406afd5

Browse files
[primer] Better signature and naming in CompareCommand
1 parent 593a860 commit 406afd5

File tree

1 file changed

+16
-12
lines changed

1 file changed

+16
-12
lines changed

pylint/testutils/_primer/primer_compare_command.py

Lines changed: 16 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -13,27 +13,31 @@
1313

1414
class CompareCommand(PrimerCommand):
1515
def run(self) -> None:
16-
main_dict = self._load_json(self.config.base_file)
17-
new_dict = self._load_json(self.config.new_file)
18-
final_main_dict = self._cross_reference(main_dict, new_dict)
19-
comment = self._create_comment(final_main_dict, new_dict)
16+
main_messages = self._load_json(self.config.base_file)
17+
pr_messages = self._load_json(self.config.new_file)
18+
missing_messages, new_messages = self._cross_reference(
19+
main_messages, pr_messages
20+
)
21+
comment = self._create_comment(missing_messages, new_messages)
2022
with open(self.primer_directory / "comment.txt", "w", encoding="utf-8") as f:
2123
f.write(comment)
2224

25+
@staticmethod
2326
def _cross_reference(
24-
self, main_dict: PackageMessages, new_dict: PackageMessages
25-
) -> PackageMessages:
26-
final_main_dict: PackageMessages = {}
27+
main_dict: PackageMessages, pr_messages: PackageMessages
28+
) -> tuple[PackageMessages, PackageMessages]:
29+
missing_messages: PackageMessages = {}
2730
for package, messages in main_dict.items():
28-
final_main_dict[package] = []
31+
missing_messages[package] = []
2932
for message in messages:
3033
try:
31-
new_dict[package].remove(message)
34+
pr_messages[package].remove(message)
3235
except ValueError:
33-
final_main_dict[package].append(message)
34-
return final_main_dict
36+
missing_messages[package].append(message)
37+
return missing_messages, pr_messages
3538

36-
def _load_json(self, file_path: Path | str) -> PackageMessages:
39+
@staticmethod
40+
def _load_json(file_path: Path | str) -> PackageMessages:
3741
with open(file_path, encoding="utf-8") as f:
3842
result: PackageMessages = json.load(f)
3943
return result

0 commit comments

Comments
 (0)