paul@0 | 1 | # -*- coding: iso-8859-1 -*- |
paul@0 | 2 | """ |
paul@0 | 3 | MoinMoin - SVGChart |
paul@0 | 4 | |
paul@0 | 5 | @copyright: 2011 by Paul Boddie <paul@boddie.org.uk> |
paul@0 | 6 | @license: GNU GPL (v2 or later), see COPYING.txt for details. |
paul@0 | 7 | """ |
paul@0 | 8 | |
paul@2 | 9 | from MoinMoin import wikiutil |
paul@5 | 10 | from MoinMoin.action import AttachFile |
paul@3 | 11 | from SVGChartSupport import get_chart, get_dimensions |
paul@0 | 12 | |
paul@5 | 13 | try: |
paul@5 | 14 | from hashlib import sha1 |
paul@5 | 15 | except ImportError: |
paul@5 | 16 | from sha import new as sha1 |
paul@5 | 17 | |
paul@0 | 18 | Dependencies = ["pages"] |
paul@0 | 19 | |
paul@0 | 20 | # Parser support. |
paul@0 | 21 | |
paul@0 | 22 | class Parser: |
paul@0 | 23 | |
paul@0 | 24 | "Format CSV-like data as an SVG chart." |
paul@0 | 25 | |
paul@0 | 26 | extensions = [".csv"] |
paul@0 | 27 | |
paul@0 | 28 | def __init__(self, raw, request, **kw): |
paul@2 | 29 | |
paul@2 | 30 | """ |
paul@2 | 31 | Initialise the parser with the given 'raw' data, 'request' and any |
paul@2 | 32 | keyword arguments that may have been supplied. |
paul@2 | 33 | """ |
paul@2 | 34 | |
paul@0 | 35 | self.raw = raw |
paul@0 | 36 | self.request = request |
paul@0 | 37 | self.data = [row.split() for row in raw.split("\n")] |
paul@0 | 38 | |
paul@2 | 39 | attrs, msg = wikiutil.parseAttributes(request, kw.get("format_args", "")) |
paul@2 | 40 | |
paul@2 | 41 | # The attributes returned from the formatting arguments are encoded like |
paul@2 | 42 | # strings. |
paul@2 | 43 | |
paul@3 | 44 | quotes = '"' + "'" |
paul@3 | 45 | self.xorigin = float((attrs.get("xorigin") or '0').strip(quotes)) |
paul@3 | 46 | self.yorigin = float((attrs.get("yorigin") or '0').strip(quotes)) |
paul@3 | 47 | width = (attrs.get("width") or '').strip(quotes) or None |
paul@3 | 48 | height = (attrs.get("height") or '').strip(quotes) or None |
paul@3 | 49 | self.width = width and int(width) |
paul@3 | 50 | self.height = height and int(height) |
paul@2 | 51 | |
paul@0 | 52 | def format(self, fmt): |
paul@0 | 53 | |
paul@0 | 54 | "Format a chart using the given formatter 'fmt'." |
paul@0 | 55 | |
paul@0 | 56 | request = self.request |
paul@0 | 57 | page = request.page |
paul@0 | 58 | _ = request.getText |
paul@0 | 59 | |
paul@5 | 60 | data = "%s,%s,%s,%s,%s" % (self.width, self.height, self.xorigin, self.yorigin, self.raw) |
paul@5 | 61 | filename = "SVGChart-%s.svg" % sha1(data).hexdigest() |
paul@5 | 62 | pagename = page.page_name |
paul@0 | 63 | |
paul@5 | 64 | if not AttachFile.exists(request, pagename, filename): |
paul@3 | 65 | width, height, chart = get_chart(self.data, self.width, self.height, self.xorigin, self.yorigin, |
paul@2 | 66 | styles_url="%s/%s/css/svgchart.css" % (request.cfg.url_prefix_static, request.theme.name)) |
paul@5 | 67 | AttachFile.add_attachment(request, pagename, filename, chart.encode("utf-8")) |
paul@3 | 68 | else: |
paul@3 | 69 | width, height = get_dimensions(self.data, self.width, self.height, self.xorigin, self.yorigin) |
paul@0 | 70 | |
paul@0 | 71 | request.write(fmt.div(1, css_class="svgchart")) |
paul@5 | 72 | request.write(fmt.transclusion(1, data=AttachFile.getAttachUrl(pagename, filename, request), width=width, height=height)) |
paul@0 | 73 | request.write(fmt.text(_("SVG chart of CSV data."))) |
paul@0 | 74 | request.write(fmt.transclusion(0)) |
paul@0 | 75 | request.write(fmt.div(0)) |
paul@0 | 76 | |
paul@0 | 77 | # vim: tabstop=4 expandtab shiftwidth=4 |