# -*- coding: utf-8 -*-
# Copyright 2020-Today TechKhedut.
# Part of TechKhedut. See LICENSE file for full copyright and licensing details.
from odoo import api, fields, models, _


class ConstructionPurchase(models.Model):
    _inherit = 'purchase.order'

    material_req_id = fields.Many2one('material.requisition', string="Material Requisition")
    project_id = fields.Many2one(related="material_req_id.project_id", string="Sub Project", store=True)
    job_order_id = fields.Many2one('job.order', string="Work order")
    purchase_order = fields.Selection([('equipment', 'Equipment'), ('labour', 'Labour'), ('overhead', 'Overhead')],
                                      string="Source Ref.")
    equipment_subcontract_id = fields.Many2one('equipment.subcontract', string="Subcontract")
    labour_subcontract_id = fields.Many2one('labour.subcontract', string="Subcontract ")
    overhead_subcontract_id = fields.Many2one('overhead.subcontract', string="Subcontract  ")

    def _prepare_invoice(self):
        res = super(ConstructionPurchase, self)._prepare_invoice()
        if self.material_req_id:
            res['material_req_id'] = self.material_req_id.id
        if self.job_order_id:
            res['job_order_id'] = self.job_order_id.id
            res['purchase_order'] = self.purchase_order
            res['equipment_subcontract_id'] = self.equipment_subcontract_id.id
            res['labour_subcontract_id'] = self.labour_subcontract_id.id
            res['overhead_subcontract_id'] = self.overhead_subcontract_id.id
        return res


class ConstructionBills(models.Model):
    _inherit = 'account.move'

    material_req_id = fields.Many2one('material.requisition', string="Material Requisition")
    project_id = fields.Many2one(related="material_req_id.project_id", string="Sub Project", store=True)
    job_order_id = fields.Many2one('job.order', string="Work order")
    purchase_order = fields.Selection([('equipment', 'Equipment'), ('labour', 'Labour'), ('overhead', 'Overhead')],
                                      string="Source Ref.")
    equipment_subcontract_id = fields.Many2one('equipment.subcontract', string="Subcontract")
    labour_subcontract_id = fields.Many2one('labour.subcontract', string="Subcontract ")
    overhead_subcontract_id = fields.Many2one('overhead.subcontract', string="Subcontract  ")


class ConstructionProduct(models.Model):
    _inherit = 'product.product'

    last_po_price = fields.Monetary(string="Last Purchase Price")
    is_material = fields.Boolean(string="Is Material")
    is_equipment = fields.Boolean(string="Is Equipment")
    is_labour = fields.Boolean(string="Is Labour")
    is_overhead = fields.Boolean(string="Is Overhead")
    is_expense = fields.Boolean(string="Is Expense")


class ConstructionWarehouse(models.Model):
    _inherit = 'stock.warehouse'

    project_id = fields.Many2one('tk.construction.project', string="Project")
    consume_stock_location_id = fields.Many2one('stock.location', string="Consume Location")


class ConstructionDelivery(models.Model):
    _inherit = 'stock.picking'

    code = fields.Selection(related='picking_type_id.code', store=True, string="Code ")
    transfer_id = fields.Many2one('internal.transfer', string="Transfer Ref.")
    consume_order_id = fields.Many2one('job.order', string="Consume Order Ref.")
    material_consume_id = fields.Many2one('material.consume', string="Material Consume Ref.")

    def button_validate(self):
        # Clean-up the context key at validation to avoid forcing the creation of immediate
        # transfers.
        ctx = dict(self.env.context)
        ctx.pop('default_immediate_transfer', None)
        self = self.with_context(ctx)

        # Sanity checks.
        if not self.env.context.get('skip_sanity_check', False):
            self._sanity_check()

        self.message_subscribe([self.env.user.partner_id.id])

        # Run the pre-validation wizards. Processing a pre-validation wizard should work on the
        # moves and/or the context and never call `_action_done`.
        if not self.env.context.get('button_validate_picking_ids'):
            self = self.with_context(button_validate_picking_ids=self.ids)
        res = self._pre_action_done_hook()
        if res is not True:
            return res

        # Call `_action_done`.
        pickings_not_to_backorder = self.filtered(lambda p: p.picking_type_id.create_backorder == 'never')
        if self.env.context.get('picking_ids_not_to_backorder'):
            pickings_not_to_backorder |= self.browse(self.env.context['picking_ids_not_to_backorder']).filtered(
                lambda p: p.picking_type_id.create_backorder != 'always'
            )
        pickings_to_backorder = self - pickings_not_to_backorder
        pickings_not_to_backorder.with_context(cancel_backorder=True)._action_done()
        pickings_to_backorder.with_context(cancel_backorder=False)._action_done()

        if self.user_has_groups('stock.group_reception_report') \
                and self.picking_type_id.auto_show_reception_report:
            lines = self.move_ids.filtered(lambda
                                               m: m.product_id.type == 'product' and m.state != 'cancel' and m.quantity_done and not m.move_dest_ids)
            if lines:
                # don't show reception report if all already assigned/nothing to assign
                wh_location_ids = self.env['stock.location']._search(
                    [('id', 'child_of', self.picking_type_id.warehouse_id.view_location_id.ids),
                     ('usage', '!=', 'supplier')])
                if self.env['stock.move'].search([
                    ('state', 'in', ['confirmed', 'partially_available', 'waiting', 'assigned']),
                    ('product_qty', '>', 0),
                    ('location_id', 'in', wh_location_ids),
                    ('move_orig_ids', '=', False),
                    ('picking_id', 'not in', self.ids),
                    ('product_id', 'in', lines.product_id.ids)], limit=1):
                    action = self.action_view_reception_report()
                    action['context'] = {'default_picking_ids': self.ids}
                    return action

        if self.transfer_id:
            return {
                'type': 'ir.actions.act_window',
                'name': _('Internal Transfer'),
                'res_model': 'internal.transfer',
                'res_id': self.transfer_id.id,
                'view_mode': 'form',
                'target': 'current'
            }
        return True