Improved registry listing by using iterator
1 parent 30d1028 commit 64f014e636ea582eabc602a711393423e76e8327
@David Whale David Whale authored on 29 May 2016
Showing 1 changed file
View
17
src/setup_tool.py
 
def show_registry():
"""Show the registry as a numbered list"""
 
names = energenie.registry.names()
i=1
for name in names:
names = []
for name in energenie.registry.names():
print("%d. %s %s" % (i, name, energenie.registry.get(name)))
names.append(name)
i += 1
 
return names
 
 
def get_device_name():
names = show_registry()
 
try:
while True:
i = readin("Which device %s to %s? " % (1,len(names)))
i = readin("Which device %s to %s? " % (1, len(names)))
try:
device_index = int(i)
break # got it
if device_index < 1 or device_index > len(names):
print("Choose a number between %s and %s" % (1, len(names)))
else:
break # got it
except ValueError:
print("Must enter a number")
 
except KeyboardInterrupt: