# This file is part of Tryton.  The COPYRIGHT file at the top level of
# this repository contains the full copyright notices and license terms.
from trytond.model import ModelView, ModelSQL, fields, Workflow
from trytond.pool import Pool
from trytond.transaction import Transaction
from trytond.modules.currency.fields import Monetary
from trytond.modules.product import price_digits
from decimal import Decimal
from datetime import date


class SaleOrder (Workflow, ModelView, ModelSQL):
    "Sale Order"
    __name__ = 'sale.order'

    company = fields.Many2One(
        'company.company', "Company", required=True
    )
    party = fields.Many2One(
        'party.party', "Party", required=True
    )
    pickup_location = fields.Selection(
        [("on_site", "On Site"),
         ("at_home", "At Home")], 'Pickup Location'
    )
    order_address = fields.Many2One(
        'party.address', 'Address'
    )
    order_mobile = fields.Function(
        fields.Char('Mobile'),
        'on_change_with_order_mobile'
    )
    lines = fields.One2Many(
        'order.line', 'order', 'Lines'
    )
    date = fields.Date("Date", required=True
    )
    currency = fields.Many2One(
        'currency.currency', 'Currency', required=True
    )
    total_order = fields.Function(
        Monetary("Total", currency='currency', digits='currency'),
        'on_change_with_total_order'
    )
    number = fields.Char(
        "Number", readonly=True
    )
    state = fields.Selection([
        ('draft', 'Draft'),
        ('confirmed', 'Confirmed')], "State"
    )

    @classmethod
    def __setup__(cls):
        super(SaleOrder, cls).__setup__()
        cls._buttons.update({
            'confirm': {},
        })
        cls._transitions |= set((
            ('draft', 'confirmed'),
        ))

    @classmethod
    @ModelView.button
    @Workflow.transition('confirmed')
    def confirm(cls, orders):
        for order in orders:
            if order.number:
                continue
            else:
                cls.set_code([order])

    @classmethod
    def set_code(cls, orders):
        for order in orders:
            order.number = cls.get_number()[0].get()
            order.save()

    @classmethod
    def get_number(cls):
        pool = Pool()
        Sequence = pool.get('ir.sequence')
        order_sequence = Sequence.search([(
            'sequence_type.name', '=', "Order")])
        return order_sequence

    @staticmethod
    def default_state():
        return 'draft'

    @staticmethod
    def default_company():
        return Transaction().context.get('company')

    @staticmethod
    def default_date():
        return date.today()

    @fields.depends('lines')
    def on_change_with_total_order(self, name=None):
        total = Decimal('0.0')
        if self.lines:
            for line in self.lines:
                if line.total_amount:
                    total += Decimal(line.total_amount)
        return total

    @classmethod
    def default_currency(cls, **pattern):
        pool = Pool()
        Company = pool.get('company.company')
        company = pattern.get('company')
        if not company:
            company = cls.default_company()
        if company:
            return Company(company).currency.id

    @fields.depends('party')
    def on_change_with_order_mobile(self, name=None):
        if self.party:
            pool = Pool()
            ContactMechanism = pool.get('party.contact_mechanism')
            mechanisms = ContactMechanism.search([
                ('party', '=', self.party.id),
                ('type', '=', 'mobile')
            ])
            if mechanisms:
                return mechanisms[0].value
        return None


class OrderLine(ModelView, ModelSQL):
    "Order Line"
    __name__ = 'order.line'

    company = fields.Many2One(
        'company.company', "Company", required=True
    )
    order = fields.Many2One(
        'sale.order', "Sale"
    )
    currency = fields.Many2One(
        'currency.currency', 'Currency', required=True
    )
    product = fields.Many2One(
        'product.product', 'Product', required=True
    )
    unit = fields.Many2One(
        'product.uom', 'Unit'
    )
    product_uom_category = fields.Function(
        fields.Many2One('product.uom.category', 'Product UOM Category'),
        'on_change_with_product_uom_category'
    )
    quantity = fields.Float(
        "Quantity", digits=('unit')
    )
    unitprice = Monetary(
        "Unit Price", digits=price_digits, currency='currency'
    )
    total_amount = fields.Function(
        Monetary("Total Amount", currency='currency', digits='currency'),
        'on_change_with_total_amount'
    )

    @staticmethod
    def default_company():
        return Transaction().context.get('company')

    @classmethod
    def default_currency(cls, **pattern):
        pool = Pool()
        Company = pool.get('company.company')
        company = pattern.get('company')
        if not company:
            company = cls.default_company()
        if company:
            return Company(company).currency.id

    @fields.depends('product')
    def on_change_with_product_uom_category(self, name=None):
        if self.product:
            return self.product.default_uom.category.id
        return None

    @fields.depends('quantity', 'unitprice')
    def on_change_with_total_amount(self, name=None):
        if self.unitprice and self.quantity:
            total_amount = self.unitprice * Decimal(self.quantity)

            return total_amount