farm/models/event_order.py

# -*- coding: utf-8 -*-
# @authors: Alexander Ezquevo <alexander@acysos.com>
# Copyright (C) 2015  Acysos S.L.
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).

from openerp import models, fields, api, _
from openerp.exceptions import Warning

EVENT_STATES = [
    ('draft', 'Draft'),
    ('validated', 'Validated'),
    ]


class EventOrder(models.Model):
    _name = 'farm.event.order'
    _order = 'timestamp DESC'

    name = fields.Char(string='Reference', select=True)
    state = fields.Selection(string='State', selection=EVENT_STATES,
                             default='draft')
    animal_type = fields.Selection([
        ('male', 'Male'),
        ('female', 'Female'),
        ('individual', 'Individual'),
        ('group', 'Group'),
        ], string='Animal Type',
        select=True)
    specie = fields.Many2one(comodel_name='farm.specie', string='Specie',
                             select=True, required=True)
    event_type = fields.Selection([
        ('medication', 'Medications'),
        ('insemination', 'Inseminations'),
        ('pregnancy_diagnosis', 'Pregnancy Diagnosis'),
        ('abort', 'Aborts'),
        ('farrowing', 'Farrowings'),
        ('foster', 'Fosters'),
        ('feed', 'Feed'),
        ('weaning', 'Weanings'),
        ('trasformation_event', 'Trasformation Event'),
        ], string="Event Type", required=True, select=True)
    farm = fields.Many2one(comodel_name='stock.location', string='Farm',
                           required=True,
                           domain=[('usage', '=', 'view'), ])
    timestamp = fields.Datetime(string='Date & Time', requiered=True,
                                default=fields.Datetime.now())
    employee = fields.Many2one(comodel_name='res.users', string='Employee',
                               help='Employee that did the job.')
    medication_events = fields.One2many(comodel_name='farm.medication.event',
                                        inverse_name='job_order',
                                        string='Medication')
    insemination_events = fields.One2many(
        comodel_name='farm.insemination.event', inverse_name='job_order',
        string='Insemination')
    pregnancy_diagnosis_events = fields.One2many(
        comodel_name='farm.pregnancy_diagnosis.event',
        inverse_name='job_order',
        string='Pregnanci Diagnosis')
    abort_events = fields.One2many(comodel_name='farm.abort.event',
                                   inverse_name='job_order',
                                   string='Abort Events')
    farrowing_events = fields.One2many(comodel_name='farm.farrowing.event',
                                       inverse_name='job_order',
                                       string='Farrowings')
    foster_events = fields.One2many(comodel_name='farm.foster.event',
                                    inverse_name='job_order',
                                    string='Fosters')
    weaning_events = fields.One2many(comodel_name='farm.weaning.event',
                                     inverse_name='job_order',
                                     string='weaning')
    feed_events = fields.One2many(comodel_name='farm.feed.event',
                                  inverse_name='job_order',
                                  column1='feed_inventory',
                                  string='feed')
    trasformation_events = fields.One2many(
        comodel_name='farm.transformation.event',
        inverse_name='job_order', string='Tasformation Events')
    notes = fields.Text(string='Notes')

    @api.model
    @api.returns('self', lambda value: value.id)
    def create(self, vals):
        farm_line = self.env['farm.specie.farm_line'].search([
            ('farm', '=', vals['farm'])])
        if len(farm_line)==0:
            vals['name'] = '*'
        else:
            '''
            vals['name'] = farm_line[0].event_order_sequence.get(
                farm_line[0].event_order_sequence[0].code)
                '''
            vals['name'] = farm_line.event_order_sequence.get_id(
                farm_line.event_order_sequence.id)
        res = super(EventOrder, self).create(vals)
        return res

    def get_active_events(self):
        if self.event_type == 'medication':
            active_events = self.medication_events
        elif self.event_type == 'insemination':
            active_events = self.insemination_events
        elif self.event_type == 'pregnancy_diagnosis':
            active_events = self.pregnancy_diagnosis_events
        elif self.event_type == 'abort':
            active_events = self.abort_events
        elif self.event_type == 'farrowing':
            active_events = self.farrowing_events
        elif self.event_type == 'foster':
            active_events = self.foster_events
        elif self.event_type == 'feed':
            active_events = self.feed_events
        elif self.event_type == 'weaning':
            active_events = self.weaning_events
        elif self.event_type == 'trasformation_event':
            active_events = self.trasformation_events
        return active_events

    @api.one
    def confirm(self):
        if(self.confirm_event(self.get_active_events())):
                self.state = 'validated'
        else:
            raise Warning(
                _('There are no event associated with this work order'))

    def confirm_event(self, event_type):
        events = event_type.search([('state', '=', 'draft'),
                                    ('job_order', '=', self.id)])
        control = False
        for event in events:
            if event.farm.id != self.farm.id:
                raise Warning(
                    _('The values of farm are diferent on order and events'))
            elif self.animal_type:
                if event.animal_type != self.animal_type:
                    if self.event_type != 'medication':
                        raise Warning(
                            _('The values of animal type are diferent on order'
                              ' and events'))
            if event.specie != self.specie:
                raise Warning(
                    _('The values of specie are diferent on order and events'))
            else:
                control = event.confirm()
                if control is None:
                    control = False
        return control