122 lines
4.4 KiB
Python
122 lines
4.4 KiB
Python
# This file is part of Tryton. The COPYRIGHT file at the top level of
|
|
# this repository contains the full copyright notices and license terms.
|
|
from decimal import Decimal
|
|
|
|
from sql.aggregate import Sum
|
|
|
|
from trytond.model import fields
|
|
from trytond.pyson import Eval, Id
|
|
from trytond.transaction import Transaction
|
|
from trytond.pool import Pool, PoolMeta
|
|
from trytond.tools import reduce_ids, grouped_slice
|
|
|
|
__all__ = ['Work']
|
|
__metaclass__ = PoolMeta
|
|
|
|
|
|
class Work:
|
|
__name__ = 'project.work'
|
|
product = fields.Many2One('product.product', 'Product',
|
|
domain=[
|
|
('type', '=', 'service'),
|
|
('default_uom_category', '=', Id('product', 'uom_cat_time')),
|
|
])
|
|
list_price = fields.Numeric('List Price',
|
|
digits=(16, Eval('currency_digits', 2)), depends=['currency_digits'])
|
|
revenue = fields.Function(fields.Numeric('Revenue',
|
|
digits=(16, Eval('currency_digits', 2)),
|
|
depends=['currency_digits']), 'get_total')
|
|
cost = fields.Function(fields.Numeric('Cost',
|
|
digits=(16, Eval('currency_digits', 2)),
|
|
depends=['currency_digits']), 'get_total')
|
|
currency_digits = fields.Function(fields.Integer('Currency Digits'),
|
|
'on_change_with_currency_digits')
|
|
|
|
@classmethod
|
|
def _get_cost(cls, works):
|
|
pool = Pool()
|
|
Line = pool.get('timesheet.line')
|
|
Work = pool.get('timesheet.work')
|
|
transaction = Transaction()
|
|
cursor = transaction.cursor
|
|
|
|
costs = dict.fromkeys([w.id for w in works], 0)
|
|
|
|
table = cls.__table__()
|
|
work = Work.__table__()
|
|
line = Line.__table__()
|
|
|
|
work_ids = [w.id for w in works]
|
|
for sub_ids in grouped_slice(work_ids):
|
|
red_sql = reduce_ids(table.id, sub_ids)
|
|
# Group by cost_price as databases will manage differently
|
|
# the product of duration and cost_price
|
|
cursor.execute(*table.join(work,
|
|
condition=table.work == work.id
|
|
).join(line, condition=line.work == work.id
|
|
).select(table.id, line.cost_price, Sum(line.duration),
|
|
where=red_sql,
|
|
group_by=[table.id, line.cost_price]))
|
|
for work_id, cost, duration in cursor.fetchall():
|
|
# SQLite stores timedelta as float
|
|
if not isinstance(duration, float):
|
|
duration = duration.total_seconds()
|
|
# SQLite uses float for SUM
|
|
if not isinstance(cost, Decimal):
|
|
cost = Decimal(str(cost))
|
|
hours = duration / 60 / 60
|
|
costs[work_id] += Decimal(str(hours)) * cost
|
|
for work in works:
|
|
costs[work.id] = work.company.currency.round(costs[work.id])
|
|
return costs
|
|
|
|
@classmethod
|
|
def _get_revenue(cls, works):
|
|
return {w.id: (w.list_price * Decimal(str(w.effort_hours))
|
|
if w.list_price else Decimal(0))
|
|
for w in works}
|
|
|
|
@fields.depends('company')
|
|
def on_change_with_currency_digits(self, name=None):
|
|
if self.company:
|
|
return self.company.currency.digits
|
|
return 2
|
|
|
|
@staticmethod
|
|
def default_currency_digits():
|
|
Company = Pool().get('company.company')
|
|
company = Transaction().context.get('company')
|
|
if company:
|
|
company = Company(company)
|
|
return company.currency.digits
|
|
return 2
|
|
|
|
@fields.depends('product', 'party', 'company')
|
|
def on_change_product(self):
|
|
pool = Pool()
|
|
User = pool.get('res.user')
|
|
ModelData = pool.get('ir.model.data')
|
|
Uom = pool.get('product.uom')
|
|
Currency = pool.get('currency.currency')
|
|
|
|
if not self.product:
|
|
return
|
|
|
|
context = {}
|
|
|
|
if self.party:
|
|
context['customer'] = self.party.id
|
|
|
|
hour_uom = Uom(ModelData.get_id('product', 'uom_hour'))
|
|
|
|
with Transaction().set_context(context):
|
|
self.list_price = Uom.compute_price(self.product.default_uom,
|
|
self.product.list_price, hour_uom)
|
|
|
|
if self.company:
|
|
user = User(Transaction().user)
|
|
if user.company != self.company:
|
|
if user.company.currency != self.company.currency:
|
|
self.list_price = Currency.compute(user.company.currency,
|
|
self.list_price, self.company.currency)
|