diff --git a/neomi.py b/neomi.py index 12921a4..233eaff 100644 --- a/neomi.py +++ b/neomi.py @@ -1,3 +1,4 @@ +import enum import os import select import socket @@ -67,7 +68,7 @@ def drop_privileges(): except: die('Unable to drop privileges') -class Protocol: +class Protocol(enum.Enum): gopher, http = range(2) class RequestError(Exception):