/+junk/pygooglechart-py3k

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/%2Bjunk/pygooglechart-py3k

« back to all changes in this revision

Viewing changes to pygooglechart.py

  • Committer: gak
  • Date: 2007-12-13 22:35:59 UTC
  • Revision ID: git-v1:a7ad1afe3f08b7de8b3a3cbdc7165d4f88ddbe92
version bump, added BadContentTypeException, added a few examples, added COPYING licence, code is more PEP8 friendly, download() checks for content type and raises on bad http codes, add_data returns the index of the dataset, Line doesn't allow being an instance.

Show diffs side-by-side

added added

removed removed

Lines of Context:
pygooglechart.py
160
160
    RIGHT = 'r'
161
161
    TYPES = (BOTTOM, TOP, LEFT, RIGHT)
162
162
 
163
 
    def __init__(self, axis_index, axis_type, **kw):
164
 
        assert(axis_type in Axis.TYPES)
 
163
    def __init__(self, axis, **kw):
 
164
        assert(axis in Axis.TYPES)
165
165
        self.has_style = False
166
 
        self.axis_index = axis_index
167
 
        self.axis_type = axis_type
 
166
        self.index = None
168
167
        self.positions = None
169
168
 
170
 
    def set_index(self, axis_index):
171
 
        self.axis_index = axis_index
 
169
    def set_index(self, index):
 
170
        self.index = index
172
171
 
173
172
    def set_positions(self, positions):
174
173
        self.positions = positions
182
181
 
183
182
    def style_to_url(self):
184
183
        bits = []
185
 
        bits.append(str(self.axis_index))
 
184
        bits.append(str(self.index))
186
185
        bits.append(self.colour)
187
186
        if self.font_size is not None:
188
187
            bits.append(str(self.font_size))
192
191
 
193
192
    def positions_to_url(self):
194
193
        bits = []
195
 
        bits.append(str(self.axis_index))
 
194
        bits.append(str(self.index))
196
195
        bits += [str(a) for a in self.positions]
197
196
        return ','.join(bits)
198
197
 
199
198
 
200
199
class LabelAxis(Axis):
201
200
 
202
 
    def __init__(self, axis_index, axis_type, values, **kwargs):
203
 
        Axis.__init__(self, axis_index, axis_type, **kwargs)
 
201
    def __init__(self, axis, values, **kwargs):
 
202
        Axis.__init__(self, axis, **kwargs)
204
203
        self.values = [str(a) for a in values]
205
204
 
206
205
    def __repr__(self):
207
 
        return '%i:|%s' % (self.axis_index, '|'.join(self.values))
 
206
        return '%i:|%s' % (self.index, '|'.join(self.values))
208
207
 
209
208
 
210
209
class RangeAxis(Axis):
211
210
 
212
 
    def __init__(self, axis_index, axis_type, low, high, **kwargs):
213
 
        Axis.__init__(self, axis_index, axis_type, **kwargs)
 
211
    def __init__(self, axis, low, high, **kwargs):
 
212
        Axis.__init__(self, axis, **kwargs)
214
213
        self.low = low
215
214
        self.high = high
216
215
 
217
216
    def __repr__(self):
218
 
        return '%i,%s,%s' % (self.axis_index, self.low, self.high)
 
217
        return '%i,%s,%s' % (self.index, self.low, self.high)
219
218
 
220
219
# Chart Classes
221
220
# -----------------------------------------------------------------------------
253
252
            Chart.BACKGROUND: None,
254
253
            Chart.CHART: None,
255
254
        }
256
 
#        self.axis = {
257
 
#            Axis.TOP: None,
258
 
#            Axis.BOTTOM: None,
259
 
#            Axis.LEFT: None,
260
 
#            Axis.RIGHT: None,
261
 
#        }
262
 
        self.axis = []
 
255
        self.axis = {
 
256
            Axis.TOP: None,
 
257
            Axis.BOTTOM: None,
 
258
            Axis.LEFT: None,
 
259
            Axis.RIGHT: None,
 
260
        }
263
261
        self.markers = []
264
262
 
265
263
    # URL generation
411
409
    # Axis Labels
412
410
    # -------------------------------------------------------------------------
413
411
 
414
 
    def set_axis_labels(self, axis_type, values):
415
 
        assert(axis_type in Axis.TYPES)
416
 
        values = [ urllib.quote(a) for a in values ]
417
 
        axis_index = len(self.axis)
418
 
        axis = LabelAxis(axis_index, axis_type, values)
419
 
        self.axis.append(axis)
420
 
        return axis_index
421
 
 
422
 
    def set_axis_range(self, axis_type, low, high):
423
 
        assert(axis_type in Axis.TYPES)
424
 
        axis_index = len(self.axis)
425
 
        axis = RangeAxis(axis_index, axis_type, low, high)
426
 
        self.axis.append(axis)
427
 
        return axis_index
428
 
 
429
 
    def set_axis_positions(self, axis_index, positions):
430
 
        try:
431
 
            self.axis[axis_index].set_positions(positions)
432
 
        except IndexError:
433
 
            raise InvalidParametersException('Axis index %i has not been ' \
434
 
                'created' % axis)
435
 
 
436
 
    def set_axis_style(self, axis_index, colour, font_size=None, \
437
 
            alignment=None):
438
 
        try:
439
 
            self.axis[axis_index].set_style(colour, font_size, alignment)
440
 
        except IndexError:
441
 
            raise InvalidParametersException('Axis index %i has not been ' \
442
 
                'created' % axis)
 
412
    def set_axis_labels(self, axis, values):
 
413
        assert(axis in Axis.TYPES)
 
414
        self.axis[axis] = LabelAxis(axis, values)
 
415
 
 
416
    def set_axis_range(self, axis, low, high):
 
417
        assert(axis in Axis.TYPES)
 
418
        self.axis[axis] = RangeAxis(axis, low, high)
 
419
 
 
420
    def set_axis_positions(self, axis, positions):
 
421
        assert(axis in Axis.TYPES)
 
422
        if not self.axis[axis]:
 
423
            raise InvalidParametersException('Please create an axis first')
 
424
        self.axis[axis].set_positions(positions)
 
425
 
 
426
    def set_axis_style(self, axis, colour, font_size=None, alignment=None):
 
427
        assert(axis in Axis.TYPES)
 
428
        if not self.axis[axis]:
 
429
            raise InvalidParametersException('Please create an axis first')
 
430
        self.axis[axis].set_style(colour, font_size, alignment)
443
431
 
444
432
    def axis_to_url(self):
445
433
        available_axis = []
448
436
        positions = []
449
437
        styles = []
450
438
        index = -1
451
 
        for axis in self.axis:
452
 
            available_axis.append(axis.axis_type)
 
439
        for position, axis in self.axis.items():
 
440
            if not axis:
 
441
                continue
 
442
            index += 1
 
443
            axis.set_index(index)
 
444
            available_axis.append(position)
453
445
            if isinstance(axis, RangeAxis):
454
446
                range_axis.append(repr(axis))
455
447
            if isinstance(axis, LabelAxis):
678
670
#        'aabbcc00', 0.5)
679
671
#    chart.fill_linear_stripes(Chart.CHART, 20, '204070', .2, '300040', .2,
680
672
#        'aabbcc00', 0.2)
681
 
    axis_left_index = chart.set_axis_range(Axis.LEFT, 0, 10)
682
 
    axis_left_index = chart.set_axis_range(Axis.LEFT, 0, 10)
683
 
    axis_left_index = chart.set_axis_range(Axis.LEFT, 0, 10)
684
 
    axis_right_index = chart.set_axis_range(Axis.RIGHT, 5, 30)
685
 
    axis_bottom_index = chart.set_axis_labels(Axis.BOTTOM, [1, 25, 95])
686
 
    chart.set_axis_positions(axis_bottom_index, [1, 25, 95])
687
 
    chart.set_axis_style(axis_bottom_index, '003050', 15)
 
673
    chart.set_axis_range(Axis.LEFT, 0, 10)
 
674
    chart.set_axis_range(Axis.RIGHT, 5, 30)
 
675
    chart.set_axis_labels(Axis.BOTTOM, [1, 25, 95])
 
676
    chart.set_axis_positions(Axis.BOTTOM, [1, 25, 95])
 
677
    chart.set_axis_style(Axis.BOTTOM, 'FFFFFF', 15)
688
678
 
689
679
#    chart.set_pie_labels(('apples', 'oranges', 'bananas'))
690
680