用 Python 實(shí)現(xiàn)簡(jiǎn)單的 switch/case 語句

王偉廷發(fā)布于2019-07-30 17:50 / 1927人閱讀

摘要：類實(shí)現(xiàn)一通過以上約束，我們可以把當(dāng)成一個(gè)類來實(shí)現(xiàn)，傳入的參數(shù)在構(gòu)造函數(shù)里處理，然后再分別實(shí)現(xiàn)和方法即可。在構(gòu)造函數(shù)中我們記住了和執(zhí)行狀態(tài)，在里如果當(dāng)前的和匹配并且函數(shù)沒有被執(zhí)行過，那么就更新并執(zhí)行對(duì)應(yīng)的方法。

在Python中是沒有Switch / Case語句的，很多人認(rèn)為這種語句不夠優(yōu)雅靈活，在Python中用字典來處理多條件匹配問題字典會(huì)更簡(jiǎn)單高效，對(duì)于有一定經(jīng)驗(yàn)的Python玩家不得不承認(rèn)，的確如此。

但今天我們還是來看看如果一定要用Python來Switch / Case，可以怎么玩。

語法約束

我們先定義一下Switch/Case應(yīng)該怎么表達(dá)，為了簡(jiǎn)單我們可以讓它長(zhǎng)成這樣。

def cn():
    print("cn")

def us():
    print("us")

switch(lang).case("cn",cn)
            .case("us",us)
               .default(us)

類實(shí)現(xiàn)一

通過以上約束，我們可以把switch當(dāng)成一個(gè)類來實(shí)現(xiàn)，傳入的參數(shù)在構(gòu)造函數(shù)里處理，然后再分別實(shí)現(xiàn)case和default方法即可。

class switch(object):
    def __init__(self, case_path):
        self.switch_to = case_path
        self._invoked = False

    def case(self, key, method):
        if self.switch_to == key and not self._invoked:
            self._invoked = True
            method()

        return self

    def default(self, method):
        if not self._invoked:
            self._invoked = True
            method()

在構(gòu)造函數(shù)中我們記住了case_path 和執(zhí)行狀態(tài)_invoked，在case()里如果當(dāng)前的key和switch_to匹配并且函數(shù)沒有被執(zhí)行過，那么就更新_invoked并執(zhí)行對(duì)應(yīng)的方法。在default()里檢查一下_invoked，如果從沒執(zhí)行過，那么就調(diào)用default分支的函數(shù)。

看上去還不錯(cuò)，我們來試用一下。

switch("cn").case("cn",cn).case("us",us).default(fail)
>>> cn
switch("us").case("cn",cn).case("us",us).default(fail)
>>> cn
switch("jp").case("cn",cn).case("us",us).default(fail)
>>> fail
switch("cn").case("cn",cn).case("us",us)
>>> cn

讓我們來看幾個(gè)奇葩一點(diǎn)的case。

# duplicate case
switch("us").case("us",cn).case("us",us).default(fail)
>>> cn

def cn() return "cn"
def us() return "us"

# return value
result = switch("cn").case("cn",cn).case("us",us)
result
>>>

發(fā)現(xiàn)了沒有，上面的實(shí)現(xiàn)不會(huì)處理重復(fù)的case，當(dāng)然你可以加強(qiáng)一下case方法，最好是拋出異常，其他編程語言通常都這樣做。

第二個(gè)問題，你希望從case里拿到返回值，像上面的寫法是沒希望了，因?yàn)槿拥袅恕Ｎ覀兛梢钥紤]在switch類里加一個(gè)result的變量來保存執(zhí)行結(jié)果。

class switch(object):
    def __init__(self, case_path):
        ...
        self.result = None

    def case(self, key, method):
        ...
        self.result = method()
    ...

在調(diào)用結(jié)束后，就可以通過result拿到結(jié)果了。

_ = switch("cn").case("cn",cn).case("us",us)
_.result
>>> cn

類實(shí)現(xiàn)二

我大概在網(wǎng)上搜了一下，你還可以參考Brian Beck通過類來實(shí)現(xiàn)Swich/Case。

class switch(object):
    def __init__(self, value):
        self.value = value
        self.fall = False

    def __iter__(self):
        """Return the match method once, then stop"""
        yield self.match
        raise StopIteration

    def match(self, *args):
        """Indicate whether or not to enter a case suite"""
        if self.fall or not args:
            return True
        elif self.value in args:
            self.fall = True
            return True
        else:
            return False


c = "z"
for case in switch(c):
    if case("a"): pass  # only necessary if the rest of the suite is empty
    if case("c"): pass
    # ...
    if case("y"): pass
    if case("z"):
        print("c is lowercase!")
        break
    if case("A"): pass
    # ...
    if case("Z"):
        print("c is uppercase!")
        break
    if case():  # default
        print("I dunno what c was!")

這種實(shí)現(xiàn)相對(duì)復(fù)雜一點(diǎn)，而且用起來也不是很舒服，又需要for又需要if（還不如直接if/else痛快）。當(dāng)然也有好處，就是可以把相同結(jié)果的case放一起，而且case里可以寫更多東西，不僅僅是一個(gè)方法名。

寫在最后

最后我們還是回到Python推崇的方法來處理switch/case問題，一般我們可以通過字典來處理這種多分支的問題，舉例說明。

MAPPING = {
    "cn": cn,
    "us": us
}

lang = "cn"
result = MAPPING.get(lang, default=us)

是不是一目了然，不僅易于閱讀也易于維護(hù)。在字典中key是唯一的，value可以是任意類型的數(shù)據(jù)，可以是類或者是方法，所以足夠靈活。

關(guān)于作者：Toby Qin, Python 技術(shù)愛好者，目前從事測(cè)試開發(fā)相關(guān)工作，轉(zhuǎn)載請(qǐng)注明原文出處。
歡迎關(guān)注我的博客 https://betacat.online，你可以到我的公眾號(hào)中去當(dāng)吃瓜群眾。