addons/hr_timesheet_invoice/report/account_analytic_profit.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
  22 from report import report_sxw
  23 import pooler
  24
  25 class account_analytic_profit(report_sxw.rml_parse):
  26     def __init__(self, cr, uid, name, context):
  27         super(account_analytic_profit, self).__init__(cr, uid, name, context=context)
  28         self.localcontext.update({
  29             'lines': self._lines,
  30             'user_ids': self._user_ids,
  31             'journal_ids': self._journal_ids,
  32             'line': self._line,
  33         })
  34     def _user_ids(self, lines):
  35         user_obj=pooler.get_pool(self.cr.dbname).get('res.users')
  36         ids=list(set([b.user_id.id for b in lines]))
  37         return user_obj.browse(self.cr, self.uid, ids)
  38
  39     def _journal_ids(self, form, user_id):
  40         line_obj=pooler.get_pool(self.cr.dbname).get('account.analytic.line')
  41         journal_obj=pooler.get_pool(self.cr.dbname).get('account.analytic.journal')
  42         line_ids=line_obj.search(self.cr, self.uid, [
  43             ('date', '>=', form['date_from']),
  44             ('date', '<=', form['date_to']),
  45             ('journal_id', 'in', form['journal_ids'][0][2]),
  46             ('user_id', '=', user_id),
  47             ])
  48         ids=list(set([b.journal_id.id for b in line_obj.browse(self.cr, self.uid, line_ids)]))
  49         return journal_obj.browse(self.cr, self.uid, ids)
  50
  51     def _line(self, form, journal_ids, user_ids):
  52         pool=pooler.get_pool(self.cr.dbname)
  53         line_obj=pool.get('account.analytic.line')
  54         product_obj=pool.get('product.product')
  55         price_obj=pool.get('product.pricelist')
  56         ids=line_obj.search(self.cr, self.uid, [
  57                 ('date', '>=', form['date_from']),
  58                 ('date', '<=', form['date_to']),
  59                 ('journal_id', 'in', journal_ids),
  60                 ('user_id', 'in', user_ids),
  61                 ])
  62         res={}
  63         for line in line_obj.browse(self.cr, self.uid, ids):
  64             if line.account_id.pricelist_id:
  65                 if line.account_id.to_invoice:
  66                     if line.to_invoice:
  67                         id=line.to_invoice.id
  68                         name=line.to_invoice.name
  69                         discount=line.to_invoice.factor
  70                     else:
  71                         name="/"
  72                         discount=1.0
  73                         id = -1
  74                 else:
  75                     name="Fixed"
  76                     discount=0.0
  77                     id=0
  78                 pl=line.account_id.pricelist_id.id
  79                 price=price_obj.price_get(self.cr, self.uid, [pl], line.product_id.id, line.unit_amount or 1.0, line.account_id.partner_id.id)[pl]
  80             else:
  81                 name="/"
  82                 discount=1.0
  83                 id = -1
  84                 price=0.0
  85             if id not in res:
  86                 res[id]={'name': name, 'amount': 0, 'cost':0, 'unit_amount':0,'amount_th':0}
  87             xxx = round(price * line.unit_amount * (1-(discount or 0.0)), 2)
  88             res[id]['amount_th']+=xxx
  89             if line.invoice_id:
  90                 self.cr.execute('select id from account_analytic_line where invoice_id=%s', (line.invoice_id.id,))
  91                 tot = 0
  92                 for lid in self.cr.fetchall():
  93                     lid2 = line_obj.browse(self.cr, self.uid, lid[0])
  94                     pl=lid2.account_id.pricelist_id.id
  95                     price=price_obj.price_get(self.cr, self.uid, [pl], lid2.product_id.id, lid2.unit_amount or 1.0, lid2.account_id.partner_id.id)[pl]
  96                     tot += price * lid2.unit_amount * (1-(discount or 0.0))
  97                 if tot:
  98                     procent = line.invoice_id.amount_untaxed / tot
  99                     res[id]['amount'] +=  xxx * procent
 100                 else:
 101                     res[id]['amount'] += xxx
 102             else:
 103                 res[id]['amount'] += xxx
 104             res[id]['cost']+=line.amount
 105             res[id]['unit_amount']+=line.unit_amount
 106         for id in res:
 107             res[id]['profit']=res[id]['amount']+res[id]['cost']
 108             res[id]['eff']=res[id]['cost'] and '%d' % (-res[id]['amount'] / res[id]['cost'] * 100,) or 0.0
 109         return res.values()
 110
 111     def _lines(self, form):
 112         line_obj=pooler.get_pool(self.cr.dbname).get('account.analytic.line')
 113         ids=line_obj.search(self.cr, self.uid, [
 114             ('date', '>=', form['date_from']),
 115             ('date', '<=', form['date_to']),
 116             ('journal_id', 'in', form['journal_ids'][0][2]),
 117             ('user_id', 'in', form['employee_ids'][0][2]),
 118             ])
 119         return line_obj.browse(self.cr, self.uid, ids)
 120
 121 report_sxw.report_sxw('report.account.analytic.profit', 'account.analytic.line', 'addons/hr_timesheet_invoice/report/account_analytic_profit.rml', parser=account_analytic_profit)
 122
 123 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: