104 lines
3.7 KiB
Python
104 lines
3.7 KiB
Python
# This file is part of lims_diagnosis module for Tryton.
|
|
# The COPYRIGHT file at the top level of this repository contains
|
|
# the full copyright notices and license terms.
|
|
|
|
from trytond.model import fields
|
|
from trytond.pool import Pool, PoolMeta
|
|
from trytond.pyson import Eval
|
|
|
|
|
|
class Fraction(metaclass=PoolMeta):
|
|
__name__ = 'lims.fraction'
|
|
|
|
diagnostician = fields.Function(fields.Many2One('lims.diagnostician',
|
|
'Diagnostician'), 'get_sample_field', searcher='search_sample_field')
|
|
|
|
def _order_sample_field(name):
|
|
def order_field(tables):
|
|
Sample = Pool().get('lims.sample')
|
|
field = Sample._fields[name]
|
|
table, _ = tables[None]
|
|
sample_tables = tables.get('sample')
|
|
if sample_tables is None:
|
|
sample = Sample.__table__()
|
|
sample_tables = {
|
|
None: (sample, sample.id == table.sample),
|
|
}
|
|
tables['sample'] = sample_tables
|
|
return field.convert_order(name, sample_tables, Sample)
|
|
return staticmethod(order_field)
|
|
order_diagnostician = _order_sample_field('diagnostician')
|
|
|
|
|
|
class Sample(metaclass=PoolMeta):
|
|
__name__ = 'lims.sample'
|
|
|
|
diagnostician = fields.Many2One('lims.diagnostician', 'Diagnostician')
|
|
diagnosis_template = fields.Many2One('lims.diagnosis.template',
|
|
'Diagnosis Template', domain=['OR', ('active', '=', True),
|
|
('id', '=', Eval('diagnosis_template'))])
|
|
|
|
@classmethod
|
|
def create(cls, vlist):
|
|
samples = super().create(vlist)
|
|
for sample in samples:
|
|
if not sample.diagnostician:
|
|
sample.diagnostician = cls.get_default_diagnostician(sample)
|
|
sample.diagnosis_template = (sample.party.diagnosis_template
|
|
and sample.party.diagnosis_template.id or None)
|
|
sample.save()
|
|
return samples
|
|
|
|
@staticmethod
|
|
def get_default_diagnostician(sample):
|
|
# 1st check party
|
|
if sample.party.diagnostician:
|
|
return sample.party.diagnostician.id
|
|
# 2nd check plant
|
|
if hasattr(sample, 'plant') and getattr(sample, 'plant'):
|
|
if sample.plant.diagnostician:
|
|
return sample.plant.diagnostician.id
|
|
# 3rd check services
|
|
for fraction in sample.fractions:
|
|
for service in fraction.services:
|
|
if service.analysis.diagnostician:
|
|
return service.analysis.diagnostician.id
|
|
# 4th check product type
|
|
if sample.product_type.diagnostician:
|
|
return sample.product_type.diagnostician.id
|
|
return None
|
|
|
|
def _get_dict_for_fast_copy(self):
|
|
def _many2one(value):
|
|
if value:
|
|
return str(value.id)
|
|
return "NULL"
|
|
|
|
res = super()._get_dict_for_fast_copy()
|
|
res['diagnostician'] = _many2one(self.diagnostician)
|
|
res['diagnosis_template'] = _many2one(self.diagnosis_template)
|
|
return res
|
|
|
|
|
|
class CreateSampleStart(metaclass=PoolMeta):
|
|
__name__ = 'lims.create_sample.start'
|
|
|
|
diagnostician = fields.Many2One('lims.diagnostician', 'Diagnostician')
|
|
|
|
|
|
class CreateSample(metaclass=PoolMeta):
|
|
__name__ = 'lims.create_sample'
|
|
|
|
def _get_samples_defaults(self, entry_id):
|
|
samples_defaults = super()._get_samples_defaults(entry_id)
|
|
|
|
diagnostician_id = None
|
|
if (hasattr(self.start, 'diagnostician') and
|
|
getattr(self.start, 'diagnostician')):
|
|
diagnostician_id = getattr(self.start, 'diagnostician').id
|
|
|
|
if diagnostician_id:
|
|
for sample in samples_defaults:
|
|
sample['diagnostician'] = diagnostician_id
|
|
return samples_defaults
|