~technofluid-team/openobject-addons/technofluid_multiple_installations

from mx.DateTime import RelativeDateTime, now, DateTime, localtime

from pychart import *

import pychart.legend

from report.misc import choice_colors

 

#

# Draw a graph

# 

class GanttCanvas(object):

        def __init__(self, io, convertors=(lambda x:x,lambda x:x)):

                self._datas = {}

                self._canvas = canvas.init(fname=io, format='pdf')

                self._canvas.set_author("Tiny ERP")

                self._names = {}

                self._conv = convertors

                self._min = 0

                self._max = 0

 

        def add(self, user, name, datas):

                if user not in self._datas:

                        self._datas[user] = []

                for f in datas:

                        x = map(self._conv[0], f)

                        if x[0]<self._min or not self._min:

                                self._min = x[0]

                        if x[1]>self._max or not self._max:

                                self._max = x[1]

                        self._datas[user].append( (name, x))

                        self._names.setdefault(name, x[0])

 

        def draw(self):

                colors = choice_colors(len(self._datas.keys()))

                user_color = {}

                for user in self._datas.keys():

                        user_color[user] = colors.pop()

 

                names = []

                for n in self._names:

                        names.append((self._names[n], n))

                names.sort()

                names.reverse()

                def _interval_get(*args):

                        result = []

                        for i in range(20):

                                d = localtime(self._min + (((self._max-self._min)/20)*(i+1)))

                                res = DateTime(d.year, d.month, d.day).ticks()

                                if (not result) or result[-1]<>res:

                                        result.append(res)

                        return result

 

                ar = area.T(y_coord = category_coord.T(names, 1),

                        x_grid_style=line_style.gray50_dash1,

                        x_grid_interval=_interval_get,

                        x_range = (self._min,self._max),

                        x_axis=axis.X(label="Date", format=self._conv[1]),

                        y_axis=axis.Y(label="Tasks"),

                        legend = legend.T(), size = (680,450))

 

                for user in self._datas:

                        chart_object.set_defaults(interval_bar_plot.T, direction="horizontal", data=self._datas[user])

                        f = fill_style.Plain()

                        f.bgcolor = user_color[user]

                        ar.add_plot(interval_bar_plot.T(fill_styles = [f, None], label=user, cluster=(0,1)))

 

                ar.draw(self._canvas)

 

        def close(self):

                self._canvas.close()

 

if __name__ == '__main__':

        date_to_int = lambda x: int(x.ticks())

        int_to_date = lambda x: '/a60{}'+localtime(x).strftime('%d %m %Y')

        gt = GanttCanvas('test.pdf', convertors=(date_to_int, int_to_date))

        gt.add('nicoe', 'Graphe de gantt', [(DateTime(2005,6,12), DateTime(2005,6,13))])

        gt.add('nicoe', 'Tarifs', [(DateTime(2005,6,19), DateTime(2005,6,21))])

        gt.add('gaetan', 'Calcul des prix', [(DateTime(2005,6,12), DateTime(2005,6,13))])

        gt.add('nico', 'Mise a jour du site', [(DateTime(2005,6,13), DateTime(2005,6,16))])

        gt.add('tom', 'Coucou', [(DateTime(2005,6,11), DateTime(2005,6,12))])

        gt.draw()

        gt.close()