o g(@sDdZddlmZddlmZddlmZGdddeZddZd S) zGCheck for if / assign blocks that can be rewritten with if-expressions.)nodes) BaseChecker)IAstroidCheckerc@s2eZdZefZdZddiZdejddfddZ dS) ConsiderTernaryExpressionCheckerconsider_ternary_expressionW0160)z*Consider rewriting as a ternary expressionconsider-ternary-expressionzxMultiple assign statements spread across if/else blocks can be rewritten with a single assingment and ternary expressionnodereturnNcCst|jtjr dSt|jdkst|jdkrdS|jd}|jd}t|tjr/t|tjs1dSt|j |j D]\}}t|tj rHt|tj sKdS|j |j krTdSq8|j d|ddS)Nrr)r ) isinstanceparentrIflenbodyorelseAssignziptargets AssignNamename add_message)selfr bstostbnameonamerk/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/pylint/extensions/consider_ternary_expression.pyvisit_ifs"   z)ConsiderTernaryExpressionChecker.visit_if) __name__ __module__ __qualname__r__implements__rmsgsrrrrrrrr s  rcCs|t|dS)zRequired method to auto register this checker. :param linter: Main interface object for Pylint plugins :type linter: Pylint object N)register_checkerr)linterrrrregister/sr'N) __doc__astroidrpylint.checkersrpylint.interfacesrrr'rrrrs     &