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
        }
        if context:
            if context == 'with_detail':
                res_users = self.env['res.users']
                data_with_detail = {
                    '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': [self.env['account.move'].api_v1_single_response(x) for x in sale_order.invoice_ids]
                }
                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)
                data.update(data_with_detail)
        return data