Pythonにおける例外を投げる関数

例外は、条件を使って関数内に記述して投げると非常に便利です。

リストの長さが 3 以下の場合にのみリスト要素の合計を返し、それ以外の場合は独自の例外 ToBigLength を投げる関数 get_sum を作成しましょう。

pythondef get_sum(lst):
  if len(lst) <= 3:
    return sum(lst)
  else:
    raise ToBigLength

次に、関数 get_sum にリストを渡し、結果をコンソールに出力してみましょう。

pythonlst = [1, 2, 3, 4]

print(get_sum(lst)) # 6 を出力します

関数内では、複数のタイプの例外をキャッチするように設定することもできます。 もう一つ例外クラス ToSmallLength を作成しましょう。

pythonclass ToSmallLength(Exception):
  pass

ここで、関数内に条件を記述します。リストの長さがゼロの場合は ToSmallLength を投げ、どの例外にも該当しない場合はリストの全要素の合計を返すようにします。

pythondef get_sum(lst):
  if len(lst) > 3:
    raise ToBigLength
  if len(lst) == 0:
    raise ToSmallLength
  else:
    return sum(lst)

便利のために、関数のパラメータとして渡されるリスト、関数自体、そしてその呼び出しを try ブロックで囲むことができます。 作成した2つの例外のキャッチは except ブロックに配置します。

pythontry:
  lst = [1, 2, 3]
  res = get_sum(lst)
  print(res)
except ToBigLength:
  print('error 1')
except ToSmallLength:
  print('error 2')