擅长:python、mysql、java
<p>添加以下函数</p>
<pre><code>def get_input(*names):
while True:
try:
for i, name in enumerate(names, 1):
print('[{:>2}] {}'.format(i, name))
choice = int(input()) - 1
if choice >= 0 and choice < len(names):
return names[choice]
except Exception:
print('Bad choice, try again...')
</code></pre>
<p>像这样使用-<code>get_input('Warrior', 'Mage', 'Archer')</code></p>
<p>输出示例:</p>
<pre><code>To play enter [Play]: Play
you were a.....
Choose your class!
[ 1] Warrior
[ 2] Mage
[ 3] Archer
4
[ 1] Warrior
[ 2] Mage
[ 3] Archer
Bad choice, try again...
[ 1] Warrior
[ 2] Mage
[ 3] Archer
dd
Bad choice, try again...
[ 1] Warrior
[ 2] Mage
[ 3] Archer
fdsg
Bad choice, try again...
[ 1] Warrior
[ 2] Mage
[ 3] Archer
1
Warrior
</code></pre>