Mercurial > sqlpython
comparison plothandler.py @ 153:ebdd20cfba69
plots finally editable
| author | catherine@Elli.myhome.westell.com |
|---|---|
| date | Mon, 29 Sep 2008 05:28:28 -0400 |
| parents | |
| children |
comparison
equal
deleted
inserted
replaced
| 152:c26bc528cb05 | 153:ebdd20cfba69 |
|---|---|
| 1 import shelve, pickle, cx_Oracle, datetime, sys | |
| 2 shelvename = 'plot.shelve' | |
| 3 | |
| 4 try: | |
| 5 import pylab | |
| 6 class Plot(object): | |
| 7 plottable_types = (cx_Oracle.NUMBER, datetime.datetime) | |
| 8 def __init__(self): | |
| 9 self.legends = [] | |
| 10 self.yserieslists = [] | |
| 11 self.xticks = [] | |
| 12 def build(self, sqlSession): | |
| 13 self.title = sqlSession.tblname | |
| 14 self.xlabel = sqlSession.curs.description[0][0] | |
| 15 self.datatypes = [d[1] for d in sqlSession.curs.description] | |
| 16 for (colNum, datatype) in enumerate(self.datatypes): | |
| 17 if colNum > 0 and datatype in self.plottable_types: | |
| 18 yseries = [row[colNum] for row in sqlSession.rows] | |
| 19 if max(yseries) is not None: | |
| 20 self.yserieslists.append(yseries) | |
| 21 self.legends.append(sqlSession.curs.description[colNum][0]) | |
| 22 if self.datatypes[0] in self.plottable_types: | |
| 23 self.xvalues = [r[0] for r in sqlSession.rows] | |
| 24 else: | |
| 25 self.xvalues = range(sqlSession.curs.rowcount) | |
| 26 self.xticks = [r[0] for r in sqlSession.rows] | |
| 27 def shelve(self): | |
| 28 s = shelve.open(shelvename,'c') | |
| 29 for k in ('xvalues xticks yserieslists title legends xlabel'.split()): | |
| 30 s[k] = getattr(self, k) | |
| 31 s.close() | |
| 32 # reading pickles fails with EOF error, don't understand | |
| 33 def unshelve(self): | |
| 34 s = shelve.open(shelvename) | |
| 35 self.__dict__.update(s) | |
| 36 s.close() | |
| 37 self.draw() | |
| 38 def draw(self): | |
| 39 if not self.yserieslists: | |
| 40 print 'At least one quantitative column needed to plot.' | |
| 41 return None | |
| 42 for yseries in self.yserieslists: | |
| 43 pylab.plot(self.xvalues, yseries, '-o') | |
| 44 if self.xticks: | |
| 45 pylab.xticks(self.xvalues, self.xticks) | |
| 46 pylab.xlabel(self.xlabel) | |
| 47 pylab.title(self.title) | |
| 48 pylab.legend(self.legends) | |
| 49 pylab.show() | |
| 50 | |
| 51 except ImportError: | |
| 52 class Plot(object): | |
| 53 def build(self, sqlSession): | |
| 54 pass | |
| 55 def save(self): | |
| 56 pass | |
| 57 def draw(self): | |
| 58 return 'Must install python-matplotlib to plot query results.' |