python生成PDF文件20以内加减法，给上小学的宝宝

小风儿

# -*- coding: utf-8 -*-
import random
from reportlab import platypus
from reportlab.lib.units import inch
from reportlab.lib.pagesizes import A4, landscape #最后生成横向打印的PDF
from reportlab.lib import colors

def genList():
ret = set()
while True:
a = random.randint(1, 20)
b = random.randint(1, 20)
c = random.randint(1, 20)
if a + b + c <= 20 and (a > 10 or b > 10 or c > 10):
ret.add( '%-2d + %-2d + %-2d =' % (a, b, c) )
if len(ret) == 25: break
while True:
a = random.randint(1, 20)
b = random.randint(1, 20)
c = random.randint(1, 20)
if a + b <= 20 and a + b - c >= 0 and (a > 10 or b > 10 or c > 10):
ret.add( '%-2d + %-2d - %-2d =' % (a, b, c) )
if len(ret) == 50: break
while True:
a = random.randint(1, 20)
b = random.randint(1, 20)
c = random.randint(1, 20)
if a - b >= 0 and a - b + c <= 20 and (a > 10 or b > 10 or c > 10):
ret.add( '%-2d - %-2d + %-2d =' % (a, b, c) )
if len(ret) == 75: break
while True:
a = random.randint(11, 20)
b = random.randint(1, 20)
c = random.randint(1, 20)
if a - b - c >= 0 and (a > 10 or b > 10 or c > 10):
ret.add( '%-2d - %-2d - %-2d =' % (a, b, c) )
if len(ret) == 100: break
ret = list(ret)
random.shuffle(ret)
return ret
'''for i in genList():
print i'''
def genTable():
data = []
items = genList()
for i in range(0, len(items), 5):
data.append(items[i:i+5])
# Courier 是 reportlab 缺省定义的等宽字体
# (0,0)/(-1,-1)说的是style运用范围，从左上到右下
tablestyle = [('FONT', (0,0), (-1,-1), 'Courier', 12),
('LINEAFTER', (0,0), (-2,-1), 1, colors.black)]
return platypus.Table(data, 2.2*inch, 0.35*inch, tablestyle)
import sys
try:
n = int(sys.argv[1]) #恩恩，生成一个多页的PDF
except:
n = 1
try:
fname = sys.argv[2]
except:
fname = '20plus100.pdf'
doc = platypus.SimpleDocTemplate(fname, topMargin=0.5*inch, bottomMargin=0.5*inch, title='DaDa Math', author='qyb')
elements = []
for i in range(n):
elements.append(genTable())
elements.append(platypus.flowables.PageBreak())
doc.pagesize = landscape(A4)
doc.build(elements)



  这个 功能需要用到python  的reportlab  包，，请下载安装