快捷導(dǎo)航

抵御代碼復(fù)雜性使python函數(shù)更加Pythonic技巧示例詳解

更新時間：2024年01月18日 08:47:32 作者：簡訊Alfred?愛生活愛扣釘

這篇文章主要介紹了抵御代碼復(fù)雜性使python函數(shù)更加Pythonic技巧示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪

掌握設(shè)計優(yōu)雅返回結(jié)果的函數(shù)的藝術(shù)

在開發(fā)解決方案時，我們傾向于將復(fù)雜的實際問題提煉為更小、更易于管理的子問題，然后使用函數(shù)來解決這些問題。函數(shù)是冗余代碼的克星，也是我們抵御代碼復(fù)雜性的最強防線。

大多數(shù)函數(shù)在編寫過程中，關(guān)鍵是其返回值。函數(shù)產(chǎn)生結(jié)果的方式會極大地影響用戶調(diào)用函數(shù)時的體驗。掌握設(shè)計優(yōu)雅返回結(jié)果的函數(shù)的藝術(shù)是制作高質(zhì)量函數(shù)的基礎(chǔ)。

不要返回多種類型

Python 是如此的靈活，以至于我們可以很容易地做到在其他語言中很難做到的事情。例如：讓函數(shù)同時返回不同類型的結(jié)果。就像下面這樣：

def get_users(user_id=None):
    if user_id is not None:
        return User.get(user_id)
    else:
        return User.filter(is_active=True)
# Return single user
get_users(user_id=1)
# Return all users
get_users()

在上面的代碼片段中，當我們需要獲取單個用戶時，我們傳遞一個 "user_id " 參數(shù)，否則，如果我們傳遞一個 None 值，它將返回所有活躍用戶的列表。乍一看，這種設(shè)計似乎很合理。

但是，編寫類似功能強大的函數(shù)并不是一件好事。這是因為優(yōu)秀的函數(shù)必須具有單一職責(zé)。所謂 "單一職責(zé)"，是指一個函數(shù)只做好一件事，而且目的明確。這樣的函數(shù)將來也不太可能隨著需求的變化而修改，同時也非常方便編寫單元測試。

返回多種類型的函數(shù)違反了 "單一職責(zé) "原則。一個好的函數(shù)應(yīng)始終提供一個穩(wěn)定的返回值，以盡量減少調(diào)用者的處理成本。就像上面的例子，我們應(yīng)該編寫兩個獨立的函數(shù) get_active_users() 和 get_user_by_id(user_id)。

使用類型提示定義返回類型

使用類型提示定義返回類型和顯式參數(shù)聲明。這樣，集成開發(fā)環(huán)境就能幫助您進行自動補全和類型檢查，從而在編輯的時候就發(fā)現(xiàn)錯誤。

例如：

def say_hello(name: str) -> str:
    return "Hello, " + name

-> 語法表示 say_hello() 函數(shù)將返回一個字符串。

使用部分函數(shù)構(gòu)造新函數(shù)

假設(shè)在這種情況下，您的代碼中有一個帶有很多參數(shù)的函數(shù) A，它非常適用。另一個函數(shù) B 調(diào)用 A 做一些工作，就像下面這樣：

def add(x, y):
    return x + y


def sum(value):
    # Calling add
    return add(100, value)

在上述示例中，我們可以使用 functools 模塊中的 partial() 函數(shù)來簡化它。

import functools

sum = functools.partial(add, 100)
sum(200)  # Output is 300

partial(func,*args,**kwargs) 以傳入的函數(shù)為基礎(chǔ)，使用變量參數(shù)構(gòu)造一個新函數(shù)。在合并當前調(diào)用參數(shù)和構(gòu)造參數(shù)后，對新函數(shù)的所有調(diào)用都將委托給原始函數(shù)。

因此，在使用部分函數(shù)時，可以將上面的求和函數(shù)定義修改為單行表達式，這樣會更加簡潔和直接。

拋出異常而不是返回錯誤

有時，您可能需要編寫同時返回結(jié)果和錯誤信息的函數(shù)：

def create_user(name):
    if len(name) > MAX_LENGTH_OF_NAME:
        return None, 'name of user is too long'
    if len(CURRENT_USERS) > MAX_USERS_QUOTA:
        return None, 'too many users'
    return User(name=name), ''def create_from_input():
    name = input()
    user, err_msg = create_user(name)
    if err_msg:
        print(f'create user failed: {err_msg}')
    else:
        print(f'user<{name}> created')

在上例中，create_user 函數(shù)的作用是創(chuàng)建一個新的用戶對象。同時，為了在發(fā)生錯誤時向調(diào)用者提供錯誤詳細信息，它利用了多返回值特性，將錯誤信息作為第二個結(jié)果返回。

但在 Python 中，這并不是解決此類問題的最佳方法。因為這種做法會增加調(diào)用者處理錯誤的成本，尤其是當許多函數(shù)都遵循這種規(guī)范，并且存在多層調(diào)用時。

在這種情況下，使用異常來處理錯誤過程是更習(xí)以為常的做法。因此，上述代碼可以重寫為：

class CreateUserError(Exception):
    """Exception for user creation failure"""
    pass

def create_user(name):
    """Create new user

    :raises: CreateUserError
    """
    if len(name) > MAX_LENGTH_OF_NAME:
        raise CreateUserError('name of user is too long')
    if len(CURRENT_USERS) > MAX_USERS_QUOTA:
        raise CreateUserError('Too many users')
    return User(name=name)


def create_for_input():
    name = input()
    try:
        user = create_user(name)
    except CreateUserError as e:
        print(f'create user failed: {e}')
    else:
        print(f'user<{name}> created')

在使用拋出異常而不是返回結(jié)果、錯誤信息后，整個錯誤處理過程乍一看變化不大，但實際上在一些細節(jié)上有很大不同：

• 新版函數(shù)的返回值類型更加穩(wěn)定，它將始終只返回用戶類型或拋出異常。

• 異常與返回值的不同之處在于，異常在被捕獲之前會不斷向調(diào)用棧的上層報告。因此，create_user 的一級調(diào)用者可以完全省略異常處理，將其留給上層處理。