configuration.py

#The COPYRIGHT file at the top level of this repository contains the full
#copyright notices and license terms.
from trytond.model import Model, ModelView, ModelSQL, ModelSingleton, fields
from trytond.pool import Pool
from trytond.transaction import Transaction

__all__ = ['Configuration', 'ConfigurationCompany']


class Configuration(ModelSingleton, ModelSQL, ModelView):
    'Truck Configuration'
    __name__ = 'truck.configuration'

    order_sequence = fields.Function(fields.Many2One('ir.sequence',
            'Order Sequence', domain=[
                ('code', '=', 'truck.order'),
                ], required=True),
        'get_company_config', setter='set_company_config')

    @classmethod
    def get_company_config(self, configs, names):
        pool = Pool()
        CompanyConfig = pool.get('truck.configuration.company')

        company_id = Transaction().context.get('company')
        company_configs = CompanyConfig.search([
                ('company', '=', company_id),
                ])

        res = {}
        for fname in names:
            res[fname] = {
                configs[0].id: None,
                }
            if company_configs:
                val = getattr(company_configs[0], fname)
                if isinstance(val, Model):
                    val = val.id
                res[fname][configs[0].id] = val
        return res

    @classmethod
    def set_company_config(self, configs, name, value):
        pool = Pool()
        CompanyConfig = pool.get('truck.configuration.company')

        company_id = Transaction().context.get('company')
        company_configs = CompanyConfig.search([
                ('company', '=', company_id),
                ])
        if company_configs:
            company_config = company_configs[0]
        else:
            company_config = CompanyConfig(company=company_id)
        setattr(company_config, name, value)
        company_config.save()


class ConfigurationCompany(ModelSQL):
    'Truck Configuration per Company'
    __name__ = 'truck.configuration.company'

    company = fields.Many2One('company.company', 'Company', required=True,
        ondelete='CASCADE', select=True)
    order_sequence = fields.Many2One('ir.sequence',
            'Order Sequence', domain=[
                ('code', '=', 'truck.order'),
                ])