# yahoochart.py
import string, copy, urllib
from reportlab.lib import colors
from reportlab.graphics import shapes
from reportlab.graphics import renderPDF, renderPM
from reportlab.graphics.charts import linecharts
from reportlab.graphics.widgets import markers
mm = markers.makeMarker
def makeChart(titles, data, cats):
x, y, w, h = 50, 50, 300, 125
# make chart
lc = linecharts.HorizontalLineChart()
lc.x, lc.y = x, y
lc.width, lc.height = w, h
lc.data = data
lc.lines.strokeWidth = 1.5
lc.lines[0].strokeColor = colors.blue
lc.lines[0].symbol = mm('Circle')
lc.lines[1].strokeColor = colors.aqua
lc.lines[1].symbol = mm('Diamond')
lc.valueAxis.valueMin = 0
lc.valueAxis.valueStep = 25
lc.valueAxis.valueMax = 1.2*max(map(max, data))
lc.valueAxis.visibleGrid = 1
lc.valueAxis.gridStrokeColor = colors.black
lc.valueAxis.gridEnd = w
lc.categoryAxis.tickDown = 0
lc.categoryAxis.categoryNames = cats
lc.categoryAxis.labels.boxAnchor = 'e'
lc.categoryAxis.labels.dy = -2
lc.categoryAxis.labels.angle = 90
lc.categoryAxis.labels.fontSize = 8
drawing = shapes.Drawing(400, 200)
drawing.add(lc)
# make titles
tdata = [(70, 160, titles[0], colors.blue),
(70, 80, titles[1], colors.aqua)]
for x, y, str, col in tdata:
t = apply(shapes.String, [x, y, str])
t.fontName = "Helvetica-Bold"
t.fontSize = 10
t.fillColor = col
drawing.add(t)
return drawing
close = lambda l: float(l[:-1].split(',')[-2])
date = lambda l: l[:-1].split(',')[0]
base = "
http://table.finance.yahoo.com/"
format = "table.csv?a=1&b=1&c=2002&d=7&e=31&f=2002"
format = format + "&s=%s&y=0&g=w&ignore=.csv"
values = []
for val in ['ibm', 'aapl']:
url = base + format % val
lines = urllib.urlopen(url).readlines()
cats = map(date, lines[1:])
cats.reverse()
closeVals = map(close, lines[1:])
closeVals.reverse()
values.append(closeVals)
d = makeChart(['IBM', 'Apple'], values, cats)
renderPM.drawToFile(d, "yahoo.png", "PNG")
