Python: 複雑なif文を消したい

Question

下記のような複雑に入り組んだif文をコード上から消して、設定をクラスとしてまとめたい

def test():
    if "条件1":
        if "条件2":
            """処理1"""
        """処理2"""
    elif "条件3":
        """処理3"""
    else:
        """処理4"""

#試したこと
enumとdictでif文に相当するところをクラスにまとめた。
見た目上、ifは消えたがそれに相当するkansu()を読む際、下から処理を追っていく必要があり読みづらい。
直観的に処理追えるようにしたい（もしくは別の方法で解決したい）

from enum import Enum

class Matome(Enum):
    one = "処理1()"
    two = "処理2()"
    three = "処理3()"
    four = "処理4()"

    @classmethod
    def kansu(cls, res1: bool, res2: bool, res3: bool):
        third = {
            True: A.three,
            False: A.four
        }
        second = {
            True: A.one,
            False: A.two
        }
        first = {
            True: second.get(res2),
            False: third.get(res3)
        }
        return first.get(res1)

def test():
    Matome.kansu(
        res1=bool("条件1"),
        res2=bool("条件2"),
        res3=bool("条件3")
    )

Comments

[quickquip]

> 複雑に入り組んだif文
めちゃくちゃ素直で普通な形だと思うんですが、実際のコードは10個以上になっているとか、質問では2階層しかないのがもっと深いとかでしょうか?

手段をより複雑かつ難しくしているように見えるので、何が問題で何を求めているのか判りづらかったです。

[pwq]

実際にはif文が5～10個、ネストが3つくらいのものを想定していました。仰る通り、自分で複雑化してしまっていたので、if文そのものを全部消すなどではなく、早期リターンなどで対処しようと思います

Answer 1

python3.7以降で辞書の順番が保持されるようになった仕様を悪用すれば、現行のコードは kansu1 のようにまとめて書くことはできます。しかしこの場合、たとえば res1 == res2 == res3 == True の時、
0. 最初に True: Matome.four が成立する
0. 次に res3: Matome.three が成立して結果が上書きされる
0. さらに見ていくと res1: Matome.two も成立する
0. 最終的には (res1 and res2): Matome.one も成立するので、結果は Matome.one になる
のように順番に追いかけていく必要があります。(key: valueの定義順を変えるとうまく動きません)

今回のケースでは、条件がbool型で3つしかないので、全部で8パタンしかないことを利用して kansu2 のように処理内容をタプルの形で用意しておいて、テーブル検索する方法が簡単だと思います。

ただ個人的には、kansu1, kansu2どちらの方法もオススメはしません。
(メンテナンス性が悪すぎるし、不具合があっても気付きにくいため)
普通のif/elseで書いて、必要があれば 処理1 とか 処理2 の部分を関数として切り出すだけで十分だと思います。

from enum import Enum

class Matome(Enum):
    one = "処理1()"
    two = "処理2()"
    three = "処理3()"
    four = "処理4()"

    @classmethod
    def kansu1(cls, res1: bool, res2: bool, res3: bool):
        return {True: Matome.four,
                res3: Matome.three,
                res1: Matome.two,
                (res1 and res2): Matome.one}[True]

    @classmethod
    def kansu2(cls, res1: bool, res2: bool, res3: bool):
        x = (res1 << 2) + (res2 << 1) + res3
        return (Matome.four, Matome.three, Matome.four, Matome.three,
                Matome.two, Matome.two, Matome.one, Matome.one)[x]


def tst(c1, c2, c3):
    return Matome.kansu2(c1, c2, c3)

print(tst(True, True, True))
print(tst(True, False, True))
print(tst(False, True, True))
print(tst(False, False, False))

Comments

[pwq]

ありがとうございます。無理にif文を消そうとして余計に可読性を低下させていました。おとなしく普通のif文で書こうと思います。