addons/stock/report/stock_graph.py

   1 # -*- coding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
   6 #
   7 #    This program is free software: you can redistribute it and/or modify
   8 #    it under the terms of the GNU Affero General Public License as
   9 #    published by the Free Software Foundation, either version 3 of the
  10 #    License, or (at your option) any later version.
  11 #
  12 #    This program is distributed in the hope that it will be useful,
  13 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 #    GNU Affero General Public License for more details.
  16 #
  17 #    You should have received a copy of the GNU Affero General Public License
  18 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 #
  20 ##############################################################################
  21 from pychart import *
  22 import pychart.legend
  23 import time
  24 from report.misc import choice_colors
  25
  26 #
  27 # Draw a graph for stocks
  28 #
  29 class stock_graph(object):
  30     def __init__(self, io):
  31         self._datas = {}
  32         self._canvas = canvas.init(fname=io, format='pdf')
  33         self._canvas.set_author("OpenERP")
  34         self._canvas.set_title("Stock Level Forecast")
  35         self._names = {}
  36         self.val_min = ''
  37         self.val_max = ''
  38
  39     def add(self, product_id, product_name, datas):
  40         if hasattr(product_name, 'replace'):
  41             product_name=product_name.replace('/', '//')
  42         if product_id not in self._datas:
  43             self._datas[product_id] = {}
  44         self._names[product_id] = product_name
  45         for (dt,stock) in datas:
  46             if not dt in self._datas[product_id]:
  47                 self._datas[product_id][dt]=0
  48             self._datas[product_id][dt]+=stock
  49             if self.val_min:
  50                 self.val_min = min(self.val_min,dt)
  51             else:
  52                 self.val_min = dt
  53             self.val_max = max(self.val_max,dt)
  54
  55     def draw(self):
  56         colors = choice_colors(len(self._datas.keys()))
  57         user_color = {}
  58         for user in self._datas.keys():
  59             user_color[user] = colors.pop()
  60
  61         val_min = int(time.mktime(time.strptime(self.val_min,'%Y-%m-%d')))
  62         val_max = int(time.mktime(time.strptime(self.val_max,'%Y-%m-%d')))
  63
  64         plots = []
  65         for product_id in self._datas:
  66             f = fill_style.Plain()
  67             f.bgcolor = user_color[user]
  68             datas = self._datas[product_id].items()
  69             datas = map(lambda x: (int(time.mktime(time.strptime(x[0],'%Y-%m-%d'))),x[1]), datas)
  70             datas.sort()
  71             datas2 = []
  72             val = 0
  73             for d in datas:
  74                 val+=d[1]
  75
  76                 if len(datas2):
  77                     d2 = d[0]-60*61*24
  78                     if datas2[-1][0]<d2-1000:
  79                         datas2.append((d2,datas2[-1][1]))
  80                 datas2.append((d[0],val))
  81             if len(datas2) and datas2[-1][0]<val_max-100:
  82                 datas2.append((val_max, datas2[-1][1]))
  83             if len(datas2)==1:
  84                 datas2.append( (datas2[0][0]+100, datas2[0][1]) )
  85             st = line_style.T()
  86             st.color = user_color[product_id]
  87             st.width = 1
  88             st.cap_style=1
  89             st.join_style=1
  90             plot = line_plot.T(label=self._names[product_id], data=datas2, line_style=st)
  91             plots.append(plot)
  92
  93         interval = max((val_max-val_min)/15, 86400)
  94         x_axis = axis.X(format=lambda x:'/a60{}'+time.strftime('%Y-%m-%d',time.gmtime(x)), tic_interval=interval, label=None)
  95         ar = area.T(size = (620,435), x_range=(val_min,val_max+1), y_axis = axis.Y(format="%d", label="Virtual Stock (Unit)"), x_axis=x_axis)
  96         for plot in plots:
  97             ar.add_plot(plot)
  98         ar.draw(self._canvas)
  99
 100     def close(self):
 101         self._canvas.close()
 102
 103 if __name__ == '__main__':
 104     gt = stock_graph('test.pdf')
 105     gt.add(1, 'Pomme', [('2005-07-29', 6), ('2005-07-30', -2), ('2005-07-31', 4)])
 106     gt.add(2, 'Cailloux', [('2005-07-29', 9), ('2005-07-30', -4), ('2005-07-31', 2)])
 107     gt.draw()
 108     gt.close()
 109
 110 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
 111