# -*- coding=utf-8 -*-
import sys
import random
import datetime
from openpyxl import Workbook
wb = Workbook()
ws = wb.active
reload(sys)
sys.setdefaultencoding('utf8')
var = {}
var["add1"] = [1,2,3,4,5,6,7,8]
var["add2"] = [2,5,6,7]
var["add3"] = [1,5,6]
var["add4"] = [5]
var["add5"] = [4]
var["add6"] = [1,2,3]
var["add7"] = [1,2]
var["add8"] = [1]
var["add9"] = [0]
var["add0"] = [1,2,3,4,5,6,7,8]
var["sub9"] = [1,2,3,4,5,6,7,8,9]
var["sub8"] = [1,2,3,4,5,6,7,8]
var["sub7"] = [1,2,5,6,7]
var["sub6"] = [1,5,6]
var["sub5"] = [5]
var["sub4"] = [1,2,3,4]
var["sub3"] = [1,2,3]
var["sub2"] = [1,2]
var["sub1"] = [1]
var["sub0"] = [0]
sa = {
    "1":"-",
    "2":"+"
}
def getNum():
    return random.randrange(1, 10)
def getU(f):
    u = random.randrange(1, 3)
    if f>9:
        f = random.randrange(0, 10)
    if (u == 2):
        return "+",var["add%s" % f]
    else:
        return "-", var["sub%s" % f]
def getV(a):
    al = len(a)
    qz = 0
    if (al != 1):
        qz = random.randrange(1, al)
    return qz
def getRes(a,b,u):
    if(u=="+"):
        return a +b
    else:
        return a-b
def recursive_formula(num,count,arr):
    if len(arr)==0:
      arr.append(str(num))
    u1, a = getU(num)
    count = count - 1
    if count>0:
        v = a[getV(a)]
        arr.append(str(u1))
        arr.append(str(v))
        r = getRes(num,v,u1)
        return recursive_formula(r, count, arr)
    else:
        return arr 
data = []
cell = 4  #生成excel 4列表格
total = 100 #生成100题
while True:
    a = recursive_formula(getNum(), 5, [])
    if len(data)>=total:
        break
    if "0" not in a:
        data.append("".join(a) + "=")

print "共%d题" % len(data)
rows = []
for item in data:
    rows.append(item)
    if len(rows)==cell:
        ws.append(rows)
        del rows[:]
version = datetime.datetime.now().strftime('%Y%m%d%H%M%S')
wb.save("珠心算题目生成-%d-%d-%s.xlsx" % (cell,len(data),version))
wb.close()