setting_util.py: handle EOFError gracefully (^D)

This commit is contained in:
enkore 2015-06-17 14:04:06 +02:00
parent ff519fbde0
commit 5ce510c549

View File

@ -10,6 +10,7 @@ from i3pystatus import Module, SettingsBase
from i3pystatus.core import ClassFinder from i3pystatus.core import ClassFinder
from collections import defaultdict, OrderedDict from collections import defaultdict, OrderedDict
def signal_handler(signal, frame): def signal_handler(signal, frame):
sys.exit(0) sys.exit(0)
signal.signal(signal.SIGINT, signal_handler) signal.signal(signal.SIGINT, signal_handler)
@ -17,7 +18,11 @@ signal.signal(signal.SIGINT, signal_handler)
def get_int_in_range(prompt, _range): def get_int_in_range(prompt, _range):
while True: while True:
answer = input(prompt) try:
answer = input(prompt)
except EOFError:
print()
sys.exit(0)
try: try:
n = int(answer.strip()) n = int(answer.strip())
if n in _range: if n in _range: