Python 实现简单的加减算数游戏

#! /usr/bin/env python

from operator import add, sub
from random import randint, choice

ops = {'+': add, '-': sub}
MAXTRIES = 2

def doprob():
op = choice('+-')
nums = [randint(1,10) for i in range(2)]
nums.sort(reverse=True)
ans = ops[op](*nums)
pr = '%d %s %d = ' % (nums[0], op, nums[1])
oops = 0
while True:
try:
if int(raw_input(pr)) == ans:
print 'correct'
break
if oops == MAXTRIES:
print 'answer\n%s%d' % (pr, ans)
else:
print 'incorrect... try again'
oops += 1
except (KeyboardInterrupt, EOFError, ValueError):
print 'invalid input... try again'

def main():
while True:
doprob()
try:
opt = raw_input('Again? [y] ').lower()
if opt and opt[0] == 'n':
break
except (KeyboardInterrupt, EOFError):
break

if __name__ == '__main__':
main()

6 + 2 = 1
incorrect... try again
6 + 2 = 1
incorrect... try again
6 + 2 = 1
answer
6 + 2 = 8
6 + 2 = 1
answer
6 + 2 = 8
6 + 2 = 8
correct
Again? [y] y
10 + 7 = 17
correct
Again? [y] y
10 - 5 = 0
incorrect... try again
10 - 5 = 0
incorrect... try again
10 - 5 = 0
answer
10 - 5 = 5
10 - 5 = 5
correct
Again? [y] n