NaNtic/trytond-wbs_certification
1
0
Fork 0
mirror of https://github.com/NaN-tic/trytond-wbs_certification.git synced 2023-12-14 03:42:52 +01:00
Code Issues Projects Releases Wiki Activity

Add children and pending quantity on certification lines

Browse source
This commit is contained in:
Sergi Almacellas Abellana 2016-03-14 16:23:15 +01:00
parent ece714d87b
commit 98baa738c9
6 changed files with 93 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

79
certification.py
View file

@ -2,7 +2,7 @@
# copyright notices and license terms.
from trytond.pool import Pool, PoolMeta
from trytond.model import Workflow, ModelView, ModelSQL, fields
from trytond.pyson import Bool, Eval
from trytond.pyson import If, Bool, Eval
from trytond.transaction import Transaction
import datetime
@ -30,6 +30,12 @@ class Certification(Workflow, ModelSQL, ModelView):
depends=['party'])
lines = fields.One2Many('wbs.certification.line', 'certification',
'Lines')
lines_tree = fields.Function(fields.One2Many('wbs.certification.line',
'certification', 'Lines',
domain=[
('parent', '=', None),
]),
'get_lines_tree', 'set_lines_tree')
state = fields.Selection([
('draft', 'Draft'),
('proposal', 'Proposal'),
@ -97,6 +103,15 @@ class Certification(Workflow, ModelSQL, ModelView):
def default_invoice_state():
return 'pending'
def get_lines_tree(self, name):
return [l.id for l in self.lines if not l.parent]
@classmethod
def set_lines_tree(cls, certifications, name, value):
cls.write(certifications, {
'lines': value,
})
@classmethod
@ModelView.button
@Workflow.transition('draft')
@ -133,11 +148,10 @@ class Certification(Workflow, ModelSQL, ModelView):
def create_lines(self, wbs, parent=None):
CL = Pool().get('wbs.certification.line')
for l in wbs.childs:
if l.type in ('subtotal', 'total'):
continue
cl = CL()
cl.certification = self
cl.wbs = l
cl.type = l.type
cl.parent = parent
cl.save()
if l.childs:
@ -175,48 +189,70 @@ class Certification(Workflow, ModelSQL, ModelView):
if default is None:
default = {}
default['lines'] = []
default['lines_tree'] = []
new_certs = super(Certification, cls).copy(certifications,
default=default)
for cert, new_cert in zip(certifications, new_certs):
new_default = default.copy()
new_default['certification'] = new_cert.id
Line.copy(cert.lines, default=new_default)
new_default = {
'certification': new_cert.id,
}
Line.copy(cert.lines_tree, default=new_default)
return new_certs
class CertificationLine(ModelSQL, ModelView):
'WBS Certification Line'
__name__ = 'wbs.certification.line'
certification = fields.Many2One('wbs.certification', 'Certification',
required=True, ondelete='CASCADE')
party = fields.Function(fields.Many2One('party.party', 'Party'),
'on_change_with_party')
type = fields.Selection([
('group', 'Group'),
('line', 'Line'),
], 'Type', required=True)
wbs = fields.Many2One('wbs', 'WBS',
domain=[
('type', '=', 'line'),
])
('type', '=', Eval('type')),
('party', '=', Eval('party')),
],
depends=['type', 'party'])
wbs_product = fields.Function(fields.Many2One('product.product',
'Product'), 'get_wbs_field')
wbs_quantity = fields.Function(fields.Float('WBS Quantity'),
'get_wbs_field')
wbs_pending_quantity = fields.Function(fields.Float(
'WBS Pending Quantity'),
'get_wbs_field')
wbs_unit = fields.Function(fields.Many2One('product.uom', 'Unit'),
'get_wbs_field')
quantity = fields.Float('Quantity')
quantity = fields.Float('Quantity',
states={
'invisible': Eval('type') != 'line',
},
domain=[
If(Bool(Eval('quantity', 0)),
('quantity', '<=', Eval('wbs_pending_quantity', 0)),
(),
)
],
depends=['wbs_pending_quantity', 'type'])
wbs_pending_quantity = fields.Function(fields.Float(
'WBS Pending Quantity'),
'get_wbs_field')
parent = fields.Many2One('wbs.certification.line', 'Parent', select=True,
left='left', right='right', ondelete='CASCADE',
domain=[
('certification', '=', Eval('certification')),
# compatibility with sale_subchapters
],
depends=['certification'])
left = fields.Integer('Left', required=True, select=True)
right = fields.Integer('Right', required=True, select=True)
children = fields.One2Many('wbs.certification.line', 'parent', 'Children',
domain=[
('certification', '=', Eval('certification')),
],
depends=['certification'])
left = fields.Integer('Left', required=True, select=True)
right = fields.Integer('Right', required=True, select=True)
@staticmethod
def default_left():
@ -226,6 +262,14 @@ class CertificationLine(ModelSQL, ModelView):
def default_right():
return 0
@staticmethod
def default_type():
return 'line'
def on_change_with_party(self, name=None):
if self.certification and self.certification.party:
return self.certification.party.id
def get_wbs_field(self, name):
if not self.wbs:
return None
@ -236,11 +280,10 @@ class CertificationLine(ModelSQL, ModelView):
@classmethod
def copy(cls, lines, default=None):
# TODO: To test
if default is None:
default = {}
default['wbs'] = None
default['children'] = []
default.setdefault('quantity')
default.setdefault('children', [])
new_lines = []
for line in lines:
new_line, = super(CertificationLine, cls).copy([line], default)

2
certification.xml
View file

@ -75,8 +75,8 @@
<record model="ir.ui.view" id="certification_line_view_list">
<field name="model">wbs.certification.line</field>
<field name="type">tree</field>
<field name="field_childs">children</field>
<field name="name">certification_line_list</field>
<field name="field_childs">children</field>
</record>
<record model="ir.action.act_window" id="act_certification_line">

4
view/certification_form.xml
View file

@ -16,8 +16,8 @@
colspan="2"/>
<newline/>
<notebook>
<page name="lines">
<field name="lines" colspan="4"/>
<page name="lines_tree">
<field name="lines_tree" colspan="4"/>
</page>
</notebook>
<group col="3" colspan="4" id="states">

7
view/certification_line_form.xml
View file

@ -2,10 +2,12 @@
<!-- The COPYRIGHT file at the top level of this repository contains the full
copyright notices and license terms. -->
<form string="WBS Certification Line">
<label name="certification"/>
<field name="certification" colspan="3"/>
<label name="wbs"/>
<field name="wbs"/>
<label name="certification"/>
<field name="certification"/>
<label name="type"/>
<field name="type"/>
<label name="wbs_product"/>
<field name="wbs_product"/>
<label name="wbs_unit"/>
@ -14,4 +16,5 @@
<field name="quantity"/>
<label name="wbs_quantity"/>
<field name="wbs_quantity"/>
<field name="children" colspan="4"/>
</form>

7
view/certification_line_list.xml
View file

@ -1,11 +1,14 @@
<?xml version="1.0"?>
<!-- The COPYRIGHT file at the top level of this repository contains the full
copyright notices and license terms. -->
<tree string="WBS Certification Lines" editable="top">
<field name="certification"/>
<tree string="WBS Certification Lines" editable="bottom">
<field name="type"/>
<field name="wbs"/>
<field name="wbs_product"/>
<field name="wbs_quantity"/>
<field name="wbs_unit"/>
<field name="quantity"/>
<field name="certification"/>
<field name="parent" tree_invisible="1"/>
<field name="children" tree_invisible="1"/>
</tree>

19
wbs.py
View file

@ -36,6 +36,13 @@ class WorkBreakdownStructure:
},
depends=['type', 'unit_digits']),
'get_certified_quantity')
pending_quantity = fields.Function(fields.Float('Pending Quantity',
digits=(16, Eval('unit_digits', 2)),
states={
'invisible': Eval('type') != 'line',
},
depends=['type', 'unit_digits']),
'get_pending_quantity')
progress = fields.Function(fields.Numeric('Progress', digits=(5, 4)),
'get_progress')
@ -67,6 +74,18 @@ class WorkBreakdownStructure:
result.update(dict(cursor.fetchall()))
return result
@classmethod
def get_pending_quantity(cls, records, name):
pool = Pool()
Uom = pool.get('product.uom')
result = {}
for wbs in records:
quantity = wbs.quantity - wbs.certified_quantity
if wbs.unit:
quantity = Uom.round(quantity, wbs.unit.rounding)
result[wbs.id] = quantity
return result
@classmethod
def get_progress(cls, records, name):
result = {}