Exception handling in Python

Exception handling in Python#

It is likely that you have raised Exceptions if you have typed all the previous commands of the tutorial. For example, you may have raised an exception if you entered a command with a typo.

Exceptions are raised by different kinds of errors arising when executing Python code. In your own code, you may also catch errors, or define custom error types. You may want to look at the descriptions of the the built-in Exceptions when looking for the right exception type.

Exceptions#

Exceptions are raised by errors in Python:

1/0

---------------------------------------------------------------------------
ZeroDivisionError                         Traceback (most recent call last)
Cell In[1], line 1
----> 1 1/0

ZeroDivisionError: division by zero

1 + 'e'

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
Cell In[2], line 1
----> 1 1 + 'e'

TypeError: unsupported operand type(s) for +: 'int' and 'str'

d = {1:1, 2:2}
d[3]

---------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)
Cell In[3], line 2
      1 d = {1:1, 2:2}
----> 2 d[3]

KeyError: 3

l = [1, 2, 3]
l[4]

---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
Cell In[4], line 2
      1 l = [1, 2, 3]
----> 2 l[4]

IndexError: list index out of range

l.foobar

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
Cell In[5], line 1
----> 1 l.foobar

AttributeError: 'list' object has no attribute 'foobar'

As you can see, there are different types of exceptions for different errors.

Catching exceptions#

try/except#

In [1]: while True:
   ....:     try:
   ....:         x = int(input('Please enter a number: '))
   ....:         break
   ....:     except ValueError:
   ....:         print('That was no valid number.  Try again...')
   ....:
Please enter a number: a
That was no valid number.  Try again...
Please enter a number: 1

In [2]: x
Out[9]: 1

try/finally#

In [1]: try:
   ....:     x = int(input('Please enter a number: '))
   ....: finally:
   ....:     print('Thank you for your input')
   ....:
Please enter a number: a
Thank you for your input
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
Cell In[10], line 2
      1 try:
----> 2     x = int(input('Please enter a number: '))
      3 finally:
      4     print('Thank you for your input')
ValueError: invalid literal for int() with base 10: 'a'

Important for resource management (e.g. closing a file)

Easier to ask for forgiveness than for permission#

def print_sorted(collection):
    try:
        collection.sort()
    except AttributeError:
        pass # The pass statement does nothing
    print(collection)

print_sorted([1, 3, 2])

[1, 2, 3]

print_sorted(set((1, 3, 2)))

{1, 2, 3}

print_sorted('132')

132

Raising exceptions#

Capturing and re-raising an exception:#

def filter_name(name):
    try:
        name = name.encode('ascii')
    except UnicodeError as e:
        if name == 'Gaël':
            print('OK, Gaël')
        else:
            raise e
    return name

filter_name('Gaël')

OK, Gaël

'Gaël'

filter_name('Stéfan')

---------------------------------------------------------------------------
UnicodeEncodeError                        Traceback (most recent call last)
Cell In[11], line 1
----> 1 filter_name('Stéfan')

Cell In[10], line 8, in filter_name(name)
      6         print('OK, Gaël')
      7     else:
----> 8         raise e
      9 return name

Cell In[10], line 3, in filter_name(name)
      1 def filter_name(name):
      2     try:
----> 3         name = name.encode('ascii')
      4     except UnicodeError as e:
      5         if name == 'Gaël':

UnicodeEncodeError: 'ascii' codec can't encode character '\xe9' in position 2: ordinal not in range(128)

Exceptions to pass messages between parts of the code:#

def achilles_arrow(x):
    if abs(x - 1) < 1e-3:
        raise StopIteration
    x = 1 - (1-x)/2.
    return x

x = 0

while True:
    try:
        x = achilles_arrow(x)
    except StopIteration:
        break

x

0.9990234375

Use exceptions to notify certain conditions are met (e.g. StopIteration) or not (e.g. custom error raising).

Contents