FM / extensions

/opt/imh-python/lib/python3.9/site-packages/pylint/extensions/

/opt/imh-python/lib/python3.9/site-packages/pylint/extensions

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
bad_builtin.py	2268	0644	edit dl rm
broad_try_clause.py	2268	0644	edit dl rm
check_elif.py	2149	0644	edit dl rm
code_style.py	13675	0644	edit dl rm
comparison_placement.py	2362	0644	edit dl rm
confusing_elif.py	2049	0644	edit dl rm
consider_refactoring_into_while_condition.py	3322	0644	edit dl rm
consider_ternary_expression.py	1708	0644	edit dl rm
dict_init_mutate.py	2121	0644	edit dl rm
docparams.py	25716	0644	edit dl rm
docstyle.py	2953	0644	edit dl rm
dunder.py	2378	0644	edit dl rm
empty_comment.py	1963	0644	edit dl rm
eq_without_hash.py	1470	0644	edit dl rm
for_any_all.py	5835	0644	edit dl rm
magic_value.py	4260	0644	edit dl rm
mccabe.py	6916	0644	edit dl rm
no_self_use.py	3710	0644	edit dl rm
overlapping_exceptions.py	3338	0644	edit dl rm
private_import.py	11247	0644	edit dl rm
redefined_loop_name.py	3230	0644	edit dl rm
redefined_variable_type.py	4105	0644	edit dl rm
set_membership.py	1806	0644	edit dl rm
typing.py	21907	0644	edit dl rm
while_used.py	1103	0644	edit dl rm
_check_docs_utils.py	29753	0644	edit dl rm
__init__.py	585	0644	edit dl rm

Edit: /opt/imh-python/lib/python3.9/site-packages/pylint/extensions/set_membership.py (1806B)

from __future__ import annotations

from typing import TYPE_CHECKING

from astroid import nodes

from pylint.checkers import BaseChecker
from pylint.checkers.utils import only_required_for_messages

if TYPE_CHECKING:
    from pylint.lint import PyLinter

class SetMembershipChecker(BaseChecker):
    name = "set_membership"
    msgs = {
        "R6201": (
            "Consider using set for membership test",
            "use-set-for-membership",
            "Membership tests are more efficient when performed on "
            "a lookup optimized datatype like ``sets``.",
        ),
    }

def __init__(self, linter: PyLinter) -> None:
        """Initialize checker instance."""
        super().__init__(linter=linter)

@only_required_for_messages("use-set-for-membership")
    def visit_compare(self, node: nodes.Compare) -> None:
        for op, comparator in node.ops:
            if op == "in":
                self._check_in_comparison(comparator)

def _check_in_comparison(self, comparator: nodes.NodeNG) -> None:
        """Checks for membership comparisons with in-place container objects."""
        if not isinstance(comparator, nodes.BaseContainer) or isinstance(
            comparator, nodes.Set
        ):
            return

# Heuristic - We need to be sure all items in set are hashable
        if all(isinstance(item, nodes.Const) for item in comparator.elts):
            self.add_message("use-set-for-membership", node=comparator)

def register(linter: PyLinter) -> None:
    linter.register_checker(SetMembershipChecker(linter))

# Licensed under the GPL: https://www.gnu.org/licenses/old-licenses/gpl-2.0.html # For details: https://github.com/pylint-dev/pylint/blob/main/LICENSE # Copyright (c) https://github.com/pylint-dev/pylint/blob/main/CONTRIBUTORS.txt from __future__ import annotations from typing import TYPE_CHECKING from astroid import nodes from pylint.checkers import BaseChecker from pylint.checkers.utils import only_required_for_messages if TYPE_CHECKING: from pylint.lint import PyLinter class SetMembershipChecker(BaseChecker): name = "set_membership" msgs = { "R6201": ( "Consider using set for membership test", "use-set-for-membership", "Membership tests are more efficient when performed on " "a lookup optimized datatype like ``sets``.", ), } def __init__(self, linter: PyLinter) -> None: """Initialize checker instance.""" super().__init__(linter=linter) @only_required_for_messages("use-set-for-membership") def visit_compare(self, node: nodes.Compare) -> None: for op, comparator in node.ops: if op == "in": self._check_in_comparison(comparator) def _check_in_comparison(self, comparator: nodes.NodeNG) -> None: """Checks for membership comparisons with in-place container objects.""" if not isinstance(comparator, nodes.BaseContainer) or isinstance( comparator, nodes.Set ): return # Heuristic - We need to be sure all items in set are hashable if all(isinstance(item, nodes.Const) for item in comparator.elts): self.add_message("use-set-for-membership", node=comparator) def register(linter: PyLinter) -> None: linter.register_checker(SetMembershipChecker(linter))