from odoo import models


class SaleOrder(models.Model):
    _inherit = 'sale.order'

    def api_v1_single_response(self, sale_order, context=False):
        data = {
            'id': sale_order.id,
            'name': sale_order.name,
            'sales': sale_order.user_id.name,
            'amount_total': sale_order.amount_total,
            'purchase_order_name': sale_order.partner_purchase_order_name,
            'invoice_count': sale_order.invoice_count,
            'status': 'draft',
        }
        if sale_order.state == 'cancel':
            data['status'] = 'cancel'
        if sale_order.state in ['draft', 'sent']:
            data['status'] = 'draft'
            if sale_order.approval_status in ['pengajuan1', 'pengajuan2']:
                data['status'] = 'waiting'
        if sale_order.state == 'sale': 
            data['status'] = 'sale'
            for picking in sale_order.picking_ids:
                if picking.state == 'assigned':
                    data['status'] = 'shipping'
        if sale_order.state == 'done': 
            data['status'] = 'done'
        
        if context:
            if context == 'with_detail':
                res_users = self.env['res.users']
                data_with_detail = {
                    'purchase_order_file': True if sale_order.partner_purchase_order_file else False,
                    'payment_term': sale_order.payment_term_id.name or '',
                    'date_order': self.env['rest.api'].datetime_to_str(sale_order.date_order, '%d/%m/%Y %H:%M:%S'),
                    'products': [],
                    'address': {
                        'customer': res_users.api_address_response(sale_order.partner_id),
                        'invoice': res_users.api_address_response(sale_order.partner_invoice_id),
                        'shipping': res_users.api_address_response(sale_order.partner_shipping_id)
                    },
                    'invoices': []
                }
                for line in sale_order.order_line:
                    product = self.env['product.product'].api_single_response(line.product_id)
                    product['quantity'] = line.product_uom_qty
                    data_with_detail['products'].append(product)
                for invoice in sale_order.invoice_ids:
                    if invoice.state == 'posted':
                        data_with_detail['invoices'].append(self.env['account.move'].api_v1_single_response(invoice))
                data.update(data_with_detail)
        return data