Revert "Remove Enum to maintain Python 2 support"

sebastinas · sebastinas · commit b2445328bdb4 · 2020-10-19T22:29:14.000+02:00
This reverts commit ac93c74.
diff --git a/bpython/repl.py b/bpython/repl.py
@@ -36,6 +36,7 @@
 import traceback
 from itertools import takewhile
 from types import ModuleType
+from enum import Enum
 
 from pygments.token import Token
 from pygments.lexers import Python3Lexer
@@ -355,12 +356,10 @@ class SourceNotFound(Exception):
     """Exception raised when the requested source could not be found."""
 
 
-class LineTypeTranslator:
+class LineTypeTranslator(Enum):
     """Used when adding a tuple to all_logical_lines, to get input / output values
     having to actually type/know the strings"""
 
-    # TODO use Enum once we drop support for Python 2
-
     INPUT = "input"
     OUTPUT = "output"
 
