from odoo import fields, models, api


class StockPicking(models.Model):
    _inherit = 'stock.picking'
    is_internal_use = fields.Boolean('Internal Use', help='flag which is internal use or not')
    account_id = fields.Many2one('account.account', string='Account')
    efaktur_id = fields.Many2one('vit.efaktur', string='Faktur Pajak')
    is_efaktur_exported = fields.Boolean(string='Is eFaktur Exported')
    date_efaktur_exported = fields.Datetime(string='eFaktur Exported Date')

    # Delivery Order
    driver_departure_date = fields.Datetime(
        string='Driver Departure Date',
        readonly=True,
        copy=False
    )
    driver_arrival_date = fields.Datetime(
        string='Driver Arrival Date',
        readonly=True,
        copy=False
    )
    delivery_tracking_no = fields.Char(
        string='Delivery Tracking Number',
        readonly=True,
        copy=False
    )
    driver_id = fields.Many2one(
        comodel_name='res.users',
        string='Driver',
        readonly=True,
        copy=False
    )
    picking_code = fields.Char(
        string="Picking Code",
        readonly=True,
        copy=False
    )

    @api.onchange('picking_type_id')
    def _onchange_operation_type(self):
        self.is_internal_use = self.picking_type_id.is_internal_use
        return

    @api.model
    def create(self, vals):
        if not self.picking_code:
            vals['picking_code'] = self.env['ir.sequence'].next_by_code('stock.picking.code') or '0'

        self._use_faktur(vals)
        return super(StockPicking, self).create(vals)

    def write(self, vals):
        self._use_faktur(vals)
        return super(StockPicking, self).write(vals)

    def _use_faktur(self, vals):
        if vals.get('efaktur_id', False):
            self.env['vit.efaktur'].search(
                [
                    ('id', '=', vals['efaktur_id'])
                ],
                limit=1
            ).is_used = True

            if self.efaktur_id.id != vals['efaktur_id']:
                self.efaktur_id.is_used = False
        return True