switch.py

# Copyright Andy Chu
class switch(object):
  """A ContextManager that translates to a C switch statement."""

  def __init__(self, value):
    # type: (int) -> None
    self.value = value

  def __enter__(self):
    # type: () -> switch
    return self

  def __exit__(self, type, value, traceback):
    # type: (Any, Any, Any) -> bool
    return False  # Allows a traceback to occur

  def __call__(self, *cases):
    # type: (*Any) -> bool
    return self.value in cases

 with switch(x) as case:
  if case(0):
    print('zero')
    print('zero')

  elif case(1, 2):
    print('one or two')

  elif case(3, 4):
    print('three or four')

  else:
    print('default')
    print('another')