- 1 錯誤和異常
- 1.1 簡介
- 1.1.1 語法錯誤
- 1.1.2 異常
- 1.2 丟擲異常
- 1.2.1 丟擲原裝異常
- 1.2.2 assert異常
- 1.2.3 丟擲使用者自定義異常
- 1.3 異常處理
- 1.3.1 try/except
- 1.3.2 try/except...else
- 1.3.3 try-finally 語句
- 1.3.4 with 關鍵字
- 1.4 分析記錄錯誤
- 1.4.1 分析錯誤
- 1.4.2 記錄錯誤
- 1.5 單元測試
- 1.5.1 unittest單元測試
- 1.5.2 setUp與tearDown
- 1.5.3 文件測試
- 1.1 簡介
1 錯誤和異常
1.1 簡介
Python 有兩種錯誤很容易辨認：語法錯誤和異常
Python assert（斷言）用於判斷一個表示式，在表示式條件為 false 的時候觸發異常。
1.1.1 語法錯誤
Python 的語法錯誤或者稱之為解析錯，是初學者經常碰到的，如下例項
>>> while True print('Hello world')
File "<stdin>", line 1, in ?
while True print('Hello world')
^
SyntaxError: invalid syntax
這個例子中，函式 print() 被檢查到有錯誤，是它前面缺少了一個冒號 : 。
語法分析器指出了出錯的一行，並且在最先找到的錯誤的位置標記了一個小小的箭頭。
1.1.2 異常
即便 Python 程式的語法是正確的，在執行它的時候，也有可能發生錯誤。執行期檢測到的錯誤被稱為異常。
大多數的異常都不會被程式處理，都以錯誤資訊的形式展現在這裡:
>>> 10 * (1/0) # 0 不能作為除數，觸發異常
Traceback (most recent call last):
File "<stdin>", line 1, in ?
ZeroDivisionError: division by zero
>>> 4 + spam*3 # spam 未定義，觸發異常
Traceback (most recent call last):
File "<stdin>", line 1, in ?
NameError: name 'spam' is not defined
>>> '2' + 2 # int 不能與 str 相加，觸發異常
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: can only concatenate str (not "int") to str
異常以不同的型別出現，這些型別都作為資訊的一部分列印出來: 例子中的型別有 ZeroDivisionError，NameError 和 TypeError。
錯誤資訊的前面部分顯示了異常發生的上下文，並以呼叫棧的形式顯示具體資訊。
1.2 丟擲異常
1.2.1 丟擲原裝異常
Python 使用 raise 語句丟擲一個指定的異常。
raise語法格式如下：raise [Exception [, args [, traceback]]]
以下例項如果 x 大於 5 就觸發異常:
x = 10
if x > 5:
raise Exception('x 不能大於 5。x 的值為: {}'.format(x))
執行以上程式碼會觸發異常：
Traceback (most recent call last):
File "test.py", line 3, in <module>
raise Exception('x 不能大於 5。x 的值為: {}'.format(x))
Exception: x 不能大於 5。x 的值為: 10
raise 唯一的一個引數指定了要被丟擲的異常。它必須是一個異常的例項或者是異常的類（也就是 Exception 的子類）。
如果只想知道這是否丟擲了一個異常，並不想去處理它，那麼一個簡單的 raise 語句就可以再次把它丟擲。
>>> try:
raise NameError('HiThere') # 模擬一個異常。
except NameError:
print('An exception flew by!')
raise
An exception flew by!
Traceback (most recent call last):
File "<stdin>", line 2, in ?
NameError: HiThere
1.2.2 assert異常
Python assert（斷言）用於判斷一個表示式，在表示式條件為 false 的時候觸發異常。
斷言可以在條件不滿足程式執行的情況下直接返回錯誤，而不必等待程式執行後出現崩潰的情況，例如我們的程式碼只能在 Linux 系統下執行，可以先判斷當前系統是否符合條件。
語法格式如下：assert expression或者附帶引數assert expression [, arguments]
等價於：
if not expression:
raise AssertionError
或者帶引數
if not expression:
raise AssertionError(arguments)
以下為 assert 使用例項：
>>> assert True # 條件為 true 正常執行
>>> assert False # 條件為 false 觸發異常
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
AssertionError
>>> assert 1==1 # 條件為 true 正常執行
>>> assert 1==2 # 條件為 false 觸發異常
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
AssertionError
>>> assert 1==2, '1 不等於 2'
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
AssertionError: 1 不等於 2
>>>
以下例項判斷當前系統是否為 Linux，如果不滿足條件則直接觸發異常，不必執行接下來的程式碼：
import sys
assert ('linux' in sys.platform), "該程式碼只能在 Linux 下執行"
1.2.3 丟擲使用者自定義異常
可以透過建立一個新的異常類來擁有自己的異常。異常類繼承自 Exception 類，可以直接繼承，或者間接繼承，例如:
>>> class MyError(Exception):
def __init__(self, value):
self.value = value
def __str__(self):
return repr(self.value)
>>> try:
raise MyError(2*2)
except MyError as e:
print('My exception occurred, value:', e.value)
My exception occurred, value: 4
>>> raise MyError('oops!')
Traceback (most recent call last):
File "<stdin>", line 1, in ?
__main__.MyError: 'oops!'
在這個例子中，類 Exception 預設的 __init__() 被覆蓋。
當建立一個模組有可能丟擲多種不同的異常時，一種通常的做法是為這個包建立一個基礎異常類，然後基於這個基礎類為不同的錯誤情況建立不同的子類:
class Error(Exception):
"""Base class for exceptions in this module."""
pass
class InputError(Error):
"""Exception raised for errors in the input.
Attributes:
expression -- input expression in which the error occurred
message -- explanation of the error
"""
def __init__(self, expression, message):
self.expression = expression
self.message = message
class TransitionError(Error):
"""Raised when an operation attempts a state transition that's not
allowed.
Attributes:
previous -- state at beginning of transition
next -- attempted new state
message -- explanation of why the specific transition is not allowed
"""
def __init__(self, previous, next, message):
self.previous = previous
self.next = next
self.message = message
大多數的異常的名字都以"Error"結尾，就跟標準的異常命名一樣。
1.3 異常處理
1.3.1 try/except
異常捕捉可以使用 try/except 語句
以下例子中，讓使用者輸入一個合法的整數，但是允許使用者中斷這個程式（使用 Control-C 或者作業系統提供的方法）。使用者中斷的資訊會引發一個 KeyboardInterrupt 異常。
while True:
try:
x = int(input("請輸入一個數字: "))
break
except ValueError:
print("您輸入的不是數字，請再次嘗試輸入！")
try 語句按照如下方式工作；
- 執行 try 子句（在關鍵字 try 和關鍵字 except 之間的語句）。
- 如果沒有異常發生，忽略 except 子句，try 子句執行後結束。
- 如果在執行 try 子句的過程中發生了異常，那麼 try 子句餘下的部分將被忽略。如果異常的型別和
except之後的名稱相符，那麼對應的 except 子句將被執行。 - 如果一個異常沒有與任何的 except 匹配，那麼這個異常將會傳遞給上層的 try 中。
一個 try 語句可能包含多個except子句，分別來處理不同的特定的異常。最多隻有一個分支會被執行。
處理程式將只針對對應的 try 子句中的異常進行處理，而不是其他的 try 的處理程式中的異常。
一個except子句可以同時處理多個異常，這些異常將被放在一個括號裡成為一個元組，例如:
except (RuntimeError, TypeError, NameError):
pass
最後一個except子句可以忽略異常的名稱，它將被當作萬用字元使用。可以使用這種方法列印一個錯誤資訊，然後使用raise再次把異常丟擲。
import sys
try:
f = open('myfile.txt')
s = f.readline()
i = int(s.strip())
except OSError as err:
print("OS error: {0}".format(err))
except ValueError:
print("Could not convert data to an integer.")
except:
print("Unexpected error:", sys.exc_info()[0])
raise
1.3.2 try/except...else
try/except 語句還有一個可選的 else 子句，如果使用這個子句，那麼必須放在所有的 except 子句之後。
else 子句將在 try 子句沒有發生任何異常的時候執行。
以下例項在 try 語句中判斷檔案是否可以開啟，如果開啟檔案時正常的沒有發生異常則執行 else 部分的語句，讀取檔案內容：
for arg in sys.argv[1:]:
try:
f = open(arg, 'r')
except IOError:
print('cannot open', arg)
else:
print(arg, 'has', len(f.readlines()), 'lines')
f.close()
使用 else 子句比把所有的語句都放在 try 子句裡面要好，這樣可以避免一些意想不到，而 except 又無法捕獲的異常。
異常處理並不僅僅處理那些直接發生在 try 子句中的異常，而且還能處理子句中呼叫的函式（甚至間接呼叫的函式）裡丟擲的異常。例如:
>>> def this_fails():
x = 1/0
>>> try:
this_fails()
except ZeroDivisionError as err:
print('Handling run-time error:', err)
Handling run-time error: int division or modulo by zero
1.3.3 try-finally 語句
try-finally 語句無論是否發生異常都將執行最後的程式碼。
以下例項中 finally 語句無論異常是否發生都會執行：
try:
runoob()
except AssertionError as error:
print(error)
else:
try:
with open('file.log') as file:
read_data = file.read()
except FileNotFoundError as fnf_error:
print(fnf_error)
finally:
print('這句話，無論異常是否發生都會執行。')
下面是一個更加複雜的例子（在同一個 try 語句裡包含 except 和 finally 子句）:
>>> def divide(x, y):
try:
result = x / y
except ZeroDivisionError:
print("division by zero!")
else:
print("result is", result)
finally:
print("executing finally clause")
>>> divide(2, 1)
result is 2.0
executing finally clause
>>> divide(2, 0)
division by zero!
executing finally clause
>>> divide("2", "1")
executing finally clause
Traceback (most recent call last):
File "<stdin>", line 1, in ?
File "<stdin>", line 3, in divide
TypeError: unsupported operand type(s) for /: 'str' and 'str'
1.3.4 with 關鍵字
Python 中的 with 語句用於異常處理，封裝了 try…except…finally 編碼正規化，提高了易用性。
with 語句使程式碼更清晰、更具可讀性， 它簡化了檔案流等公共資源的管理。
在處理檔案物件時使用 with 關鍵字是一種很好的做法。
我們可以看下以下幾種程式碼例項：不使用 with，也不使用 try…except…finally
file = open('./test_runoob.txt', 'w')
file.write('hello world !')
file.close()
以上程式碼如果在呼叫 write 的過程中，出現了異常，則 close 方法將無法被執行，因此資源就會一直被該程式佔用而無法被釋放。 接下來我們呢可以使用 try…except…finally 來改進程式碼：
file = open('./test_runoob.txt', 'w')
try:
file.write('hello world')
finally:
file.close()
以上程式碼我們對可能發生異常的程式碼處進行 try 捕獲，發生異常時執行 except 程式碼塊，finally 程式碼塊是無論什麼情況都會執行，所以檔案會被關閉，不會因為執行異常而佔用資源。
使用 with 關鍵字：
with open('./test_runoob.txt', 'w') as file:
file.write('hello world !')
使用 with 關鍵字系統會自動呼叫 f.close() 方法， with 的作用等效於 try/finally 語句是一樣的。
我們可以在執行 with 關鍵字後檢驗檔案是否關閉：
>>> with open('./test_runoob.txt') as f:
... read_data = f.read()
>>> # 檢視檔案是否關閉
>>> f.closed
True
with 語句實現原理建立在上下文管理器之上。上下文管理器是一個實現 __enter__ 和 __exit__ 方法的類。使用 with 語句確保在巢狀塊的末尾呼叫 __exit__ 方法。這個概念類似於 try...finally 塊的使用。
with open('./test_runoob.txt', 'w') as my_file:
my_file.write('hello world!')
以上例項將 hello world! 寫到 ./test_runoob.txt 檔案上。
在檔案物件中定義了 __enter__ 和 __exit__ 方法，即檔案物件也實現了上下文管理器，首先呼叫 __enter__ 方法，然後執行 with 語句中的程式碼，最後呼叫 __exit__ 方法。 即使出現錯誤，也會呼叫 __exit__ 方法，也就是會關閉檔案流。
1.4 分析記錄錯誤
1.4.1 分析錯誤
如果錯誤沒有被捕獲，它就會一直往上拋，最後被Python直譯器捕獲，列印一個錯誤資訊，然後程式退出。來看看err.py：
# err.py:
def foo(s):
return 10 / int(s)
def bar(s):
return foo(s) * 2
def main():
bar('0')
main()
執行，結果如下：
$ python err.py
Traceback (most recent call last):
File "err.py", line 11, in <module>
main()
File "err.py", line 9, in main
bar('0')
File "err.py", line 6, in bar
return foo(s) * 2
File "err.py", line 3, in foo
return 10 / int(s)
ZeroDivisionError: integer division or modulo by zero
出錯並不可怕，可怕的是不知道哪裡出錯了。解讀錯誤資訊是定位錯誤的關鍵。我們從上往下可以看到整個錯誤的呼叫函式鏈：
錯誤資訊第1行：
Traceback (most recent call last):
告訴我們這是錯誤的跟蹤資訊。
第2行：
File "err.py", line 11, in <module>
main()
呼叫main()出錯了，在程式碼檔案err.py的第11行程式碼，但原因是第9行：
File "err.py", line 9, in main
bar('0')
呼叫bar('0')出錯了，在程式碼檔案err.py的第9行程式碼，但原因是第6行：
File "err.py", line 6, in bar
return foo(s) * 2
原因是return foo(s) * 2這個語句出錯了，但這還不是最終原因，繼續往下看：
File "err.py", line 3, in foo
return 10 / int(s)
原因是return 10 / int(s)這個語句出錯了，這是錯誤產生的源頭，因為下面列印了：
ZeroDivisionError: integer division or modulo by zero
根據錯誤型別ZeroDivisionError，我們判斷，int(s)本身並沒有出錯，但是int(s)返回0，在計算10 / 0時出錯，至此，找到錯誤源頭。
1.4.2 記錄錯誤
如果不捕獲錯誤，自然可以讓Python直譯器來列印出錯誤堆疊，但程式也被結束了。既然我們能捕獲錯誤，就可以把錯誤堆疊列印出來，然後分析錯誤原因，同時，讓程式繼續執行下去。
Python內建的logging模組可以非常容易地記錄錯誤資訊：
# err.py
import logging
def foo(s):
return 10 / int(s)
def bar(s):
return foo(s) * 2
def main():
try:
bar('0')
except StandardError, e:
logging.exception(e)
main()
print 'END'
同樣是出錯，但程式列印完錯誤資訊後會繼續執行，並正常退出：
$ python err.py
ERROR:root:integer division or modulo by zero
Traceback (most recent call last):
File "err.py", line 12, in main
bar('0')
File "err.py", line 8, in bar
return foo(s) * 2
File "err.py", line 5, in foo
return 10 / int(s)
ZeroDivisionError: integer division or modulo by zero
END
透過配置，logging還可以把錯誤記錄到日誌檔案裡，方便事後排查。
1.5 單元測試
1.5.1 unittest單元測試
為了編寫單元測試，我們需要引入Python自帶的unittest模組，編寫mydict_test.py如下：
import unittest
from mydict import Dict
class TestDict(unittest.TestCase):
def test_init(self):
d = Dict(a=1, b='test')
self.assertEquals(d.a, 1)
self.assertEquals(d.b, 'test')
self.assertTrue(isinstance(d, dict))
def test_key(self):
d = Dict()
d['key'] = 'value'
self.assertEquals(d.key, 'value')
def test_attr(self):
d = Dict()
d.key = 'value'
self.assertTrue('key' in d)
self.assertEquals(d['key'], 'value')
def test_keyerror(self):
d = Dict()
with self.assertRaises(KeyError):
value = d['empty']
def test_attrerror(self):
d = Dict()
with self.assertRaises(AttributeError):
value = d.empty
編寫單元測試時，我們需要編寫一個測試類，從unittest.TestCase繼承。
以test開頭的方法就是測試方法，不以test開頭的方法不被認為是測試方法，測試的時候不會被執行。
對每一類測試都需要編寫一個test_xxx()方法。由於unittest.TestCase提供了很多內建的條件判斷，我們只需要呼叫這些方法就可以斷言輸出是否是我們所期望的。最常用的斷言就是assertEquals()：self.assertEquals(abs(-1), 1) ， 斷言函式返回的結果與1相等
另一種重要的斷言就是期待丟擲指定型別的Error，比如透過d['empty']訪問不存在的key時，斷言會丟擲KeyError：
with self.assertRaises(KeyError):
value = d['empty']
而透過d.empty訪問不存在的key時，我們期待丟擲AttributeError：
with self.assertRaises(AttributeError):
value = d.empty
執行單元測試
一旦編寫好單元測試，我們就可以執行單元測試。最簡單的執行方式是在mydict_test.py的最後加上兩行程式碼：
if __name__ == '__main__':
unittest.main()
這樣就可以把mydict_test.py當做正常的python指令碼執行，另一種更常見的方法是在命令列透過引數-m unittest直接執行單元測試：$ python -m unittest mydict_test
1.5.2 setUp與tearDown
可以在單元測試中編寫兩個特殊的setUp()和tearDown()方法。這兩個方法會分別在每呼叫一個測試方法的前後分別被執行。
setUp()和tearDown()方法有什麼用呢？設想測試需要啟動一個資料庫，這時，就可以在setUp()方法中連線資料庫，在tearDown()方法中關閉資料庫，這樣，不必在每個測試方法中重複相同的程式碼：
class TestDict(unittest.TestCase):
def setUp(self):
print 'setUp...'
def tearDown(self):
print 'tearDown...'
可以再次執行測試看看每個測試方法呼叫前後是否會列印出setUp...和tearDown...。
1.5.3 文件測試
當我們編寫註釋時，如果寫上這樣的註釋：
def abs(n):
'''
Function to get absolute value of number.
Example:
>>> abs(1)
1
>>> abs(-1)
1
>>> abs(0)
0
'''
return n if n >= 0 else (-n)
無疑更明確地告訴函式的呼叫者該函式的期望輸入和輸出。並且，Python內建的文件測試（doctest）模組可以直接提取註釋中的程式碼並執行測試。
doctest嚴格按照Python互動式命令列的輸入和輸出來判斷測試結果是否正確。只有測試異常的時候，可以用...表示中間一大段煩人的輸出。
用doctest來測試上次編寫的Dict類：
class Dict(dict):
'''
Simple dict but also support access as x.y style.
>>> d1 = Dict()
>>> d1['x'] = 100
>>> d1.x
100
>>> d1.y = 200
>>> d1['y']
200
>>> d2 = Dict(a=1, b=2, c='3')
>>> d2.c
'3'
>>> d2['empty']
Traceback (most recent call last):
...
KeyError: 'empty'
>>> d2.empty
Traceback (most recent call last):
...
AttributeError: 'Dict' object has no attribute 'empty'
'''
def __init__(self, **kw):
super(Dict, self).__init__(**kw)
def __getattr__(self, key):
try:
return self[key]
except KeyError:
raise AttributeError(r"'Dict' object has no attribute '%s'" % key)
def __setattr__(self, key, value):
self[key] = value
if __name__=='__main__':
import doctest
doctest.testmod()
執行python mydict.py：
$ python mydict.py
什麼輸出也沒有。這說明我們編寫的doctest執行都是正確的。如果程式有問題，比如把__getattr__()方法註釋掉，再執行就會報錯：
$ python mydict.py
**********************************************************************
File "mydict.py", line 7, in __main__.Dict
Failed example:
d1.x
Exception raised:
Traceback (most recent call last):
...
AttributeError: 'Dict' object has no attribute 'x'
**********************************************************************
File "mydict.py", line 13, in __main__.Dict
Failed example:
d2.c
Exception raised:
Traceback (most recent call last):
...
AttributeError: 'Dict' object has no attribute 'c'
**********************************************************************
注意到最後兩行程式碼。當模組正常匯入時，doctest不會被執行。只有在命令列執行時，才執行doctest。所以，不必擔心doctest會在非測試環境下執行。