# -*- coding: utf-8 -*-
##############################################################################
#
#    OpenERP, Open Source Management Solution
#    Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Affero General Public License as
#    published by the Free Software Foundation, either version 3 of the
#    License, or (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU Affero General Public License for more details.
#
#    You should have received a copy of the GNU Affero General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
##############################################################################

import time

from openerp.osv import fields, osv
import openerp.addons.decimal_precision as dp
from openerp.tools.translate import _
import product

class stock_landed_cost(osv.osv):
    _name = 'stock.landed.cost'
    _description = 'Stock Landed Cost'

    def _total_amount(self, cr, uid, ids, name, args, context=None):
        result = {}
        for cost in self.browse(cr, uid, ids, context=context):
            total = 0.0
            for line in cost.cost_lines:
                total += line.price_unit
            result[cost.id] = total
        return result

    def _get_cost_line(self, cr, uid, ids, context=None):
        result = {}
        for line in self.pool.get('stock.landed.cost.lines').browse(cr, uid, ids, context=context):
            result[line.cost_id.id] = True
        return result.keys()

    def onchange_pickings(self, cr, uid, ids, picking_ids=None):
        result = {'valuation_adjustment_lines': []}
        line_obj = self.pool.get('stock.valuation.adjustment.lines')
        picking_obj = self.pool.get('stock.picking')
        lines = []
 
        for cost in self.browse(cr, uid, ids):
            line_ids = [line.id for line in cost.valuation_adjustment_lines]
            line_obj.unlink(cr, uid, line_ids)

        picking_ids = picking_ids and picking_ids[0][2] or False
        if not picking_ids:
            return {'value': result}

        for picking in picking_obj.browse(cr, uid, picking_ids):
            for move in picking.move_lines:
                total_cost = 0.0
                total_qty = 0.0
                weight = move.product_id and move.product_id.weight
                volume = move.product_id and move.product_id.volume
                for quant in move.quant_ids:
                    total_cost += quant.cost
                    total_qty += quant.qty
                vals = dict(product_id = move.product_id.id, move_id = move.id, quantity = move.product_uom_qty, former_cost = total_cost * total_qty, weight = weight, volume = volume, flag = 'original')
                lines.append(vals)
        result['valuation_adjustment_lines'] = lines
        return {'value': result}

    _columns = {
        'name': fields.char('Name', size=256, required=True, states={'open': [('readonly', True)]}),
        'date': fields.date('Date', required=True, states={'open': [('readonly', True)]}),
        'picking_ids': fields.many2many('stock.picking', string='Pickings', states={'open': [('readonly', True)]}),
        'cost_lines': fields.one2many('stock.landed.cost.lines', 'cost_id', 'Cost Lines', states={'open': [('readonly', True)]}),
        'valuation_adjustment_lines': fields.one2many('stock.valuation.adjustment.lines', 'cost_id', 'Valuation Adjustments', states={'open': [('readonly', True)]}),
        'description': fields.text('Item Description', states={'open': [('readonly', True)]}),
        'amount_total': fields.function(_total_amount, type='float', string='Total', digits_compute=dp.get_precision('Account'),
            store={
                'stock.landed.cost': (lambda self, cr, uid, ids, c={}: ids, ['cost_lines'], 20),
                'stock.landed.cost.lines': (_get_cost_line, ['price_unit', 'quantity', 'cost_id'], 20),
            }
        ),
        'state':fields.selection([('draft', 'Draft'), ('open', 'Open'), ('cancel', 'Cancelled')], 'State', readonly=True),
    }

    _defaults = {
        'state': 'draft',
        'date': lambda *a: time.strftime('%Y-%m-%d'),
    }

    def _create_accounting_entries(self, cr, uid, cost, valuation_lines, context=None):
        product_obj = self.pool.get('product.product')
        for line in valuation_lines:
            credit_account_id = line.product_id.property_stock_account_input and line.product_id.property_stock_account_input.id or False
            debit_account_id = line.cost_line_id and line.cost_line_id.product_id and line.cost_line_id.product_id.property_account_expense and \
                                line.cost_line_id.product_id.property_account_expense.id or False
            if not credit_account_id:
                raise osv.except_osv(_('Error!'), _('Please configure Stock Input Account for product: %s.') % (line.product_id.name))
            if not debit_account_id:
                raise osv.except_osv(_('Error!'), _('Please configure Stock Expense Account for product: %s.') % (line.cost_line_id and line.cost_line_id.product_id and line.cost_line_id.product_id.name))
            accounts = product_obj.get_product_accounts(cr, uid, line.product_id.id, context=context)
            journal_id = accounts['stock_journal']
            self._create_account_move(cr, uid, cost, line, credit_account_id=credit_account_id, debit_account_id=debit_account_id, journal_id=journal_id, context=context)
        return True

    def _prepare_account_move_line(self, cr, uid, cost, line, credit_account_id, debit_account_id, context=None):
        """
        Generate the account.move.line values to track the landed cost.
        """
        debit_line_vals = {
            'name': cost.name,
            'product_id': line.product_id.id,
            'quantity': line.quantity,
            'debit': line.additional_landed_cost,
            'account_id': debit_account_id
        }
        credit_line_vals = {
            'name': cost.name,
            'product_id': line.product_id.id,
            'quantity': line.quantity,
            'credit': line.additional_landed_cost,
            'account_id': credit_account_id
        }
        return [(0, 0, debit_line_vals), (0, 0, credit_line_vals)]

    def _create_account_move(self, cr, uid, cost, line, credit_account_id, debit_account_id, journal_id, context=None):
        move_lines = self._prepare_account_move_line(cr, uid, cost, line, credit_account_id, debit_account_id, context=context)
        vals = {
            'journal_id': journal_id,
            'line_id': move_lines,
            'period_id': self.pool.get('account.period').find(cr, uid, cost.date, context=context)[0],
            'date': cost.date,
            'ref': cost.name
        }
        move = self.pool.get('account.move').create(cr, uid, vals, context=context)
        self.pool.get('stock.valuation.adjustment.lines').write(cr, uid, line.id, {'account_move_id': move}, context=context)
        return True

    def button_validate(self, cr ,uid, ids, context=None):
        quant_obj = self.pool.get('stock.quant')
        for cost in self.browse(cr, uid, ids, context=context):
            for line in cost.valuation_adjustment_lines:
                per_unit = line.final_cost / line.quantity
                diff = per_unit - line.former_cost_per_unit
                quants = [quant for quant in line.move_id.quant_ids if line.move_id]
                for quant in quants:
                    if quant.cost < 0:
                        new_cost = quant.cost - diff
                    else:
                        new_cost = quant.cost + diff
                    quant_obj.write(cr, uid, quant.id, {'cost': new_cost}, context=context)
            self._create_accounting_entries(cr, uid, cost, cost.valuation_adjustment_lines, context=context)
            self.write(cr, uid, cost.id, {'state': 'open'}, context=context)
        return True

    def compute_landed_cost(self, cr, uid, ids, context=None):
        line_obj = self.pool.get('stock.valuation.adjustment.lines')
        for cost in self.browse(cr, uid, ids, context=None):
            total_qty = 0.0
            total_cost = 0.0
            total_weight = 0.0
            total_volume = 0.0
            total_line = 0.0
            for line in cost.valuation_adjustment_lines:
                if line.flag == 'original':
                    total_qty += line.quantity
                    total_cost += line.former_cost
                    total_weight += line.weight
                    total_volume += line.volume
                    total_line += 1

        unlink_ids = line_obj.search(cr, uid, [('cost_id', 'in', ids), ('flag', '=', 'duplicate')], context=context)
        line_obj.unlink(cr, uid, unlink_ids, context=context)
        for cost in self.browse(cr, uid, ids, context=None):
            count = 0.0
            for line in cost.cost_lines:
                count += 1
                for valuation in cost.valuation_adjustment_lines:
                    if count == 1:
                        line_obj.write(cr, uid, valuation.id, {'cost_line_id': line.id}, context=context)
                        continue
                    line_obj.copy(cr, uid, valuation.id, default={'cost_line_id': line.id, 'flag': 'duplicate'}, context=context)

        for cost in self.browse(cr, uid, ids, context=None):
            dict = {}
            for line in cost.cost_lines:
                for valuation in cost.valuation_adjustment_lines:
                    if valuation.cost_line_id and valuation.cost_line_id.id == line.id:
                        if line.split_method == 'by_quantity':
                            per_unit = (line.price_unit / total_qty)
                            value = valuation.quantity * per_unit
                            if valuation.id not in dict:
                                dict[valuation.id] = value
                            else:
                                dict[valuation.id] += value
                        elif line.split_method == 'by_weight':
                            per_unit = (line.price_unit / total_weight if total_weight else total_line)
                            value = valuation.quantity * per_unit
                            if valuation.id not in dict:
                                dict[valuation.id] = value
                            else:
                                dict[valuation.id] += value
                        elif line.split_method == 'by_volume':
                            per_unit = (line.price_unit / total_volume if total_volume else total_line)
                            value = valuation.quantity * per_unit
                            if valuation.id not in dict:
                                dict[valuation.id] = value
                            else:
                                dict[valuation.id] += value
                        elif line.split_method == 'equal':
                            per_unit = (line.price_unit / total_line)
                            if valuation.id not in dict:
                                dict[valuation.id] = per_unit
                            else:
                                dict[valuation.id] += per_unit
                        elif line.split_method == 'by_current_cost_price':
                            per_unit = (line.price_unit / total_cost)
                            value = valuation.former_cost * per_unit
                            if valuation.id not in dict:
                                dict[valuation.id] = value
                            else:
                                dict[valuation.id] += value

        for key, value in dict.items():
            line_obj.write(cr, uid, key, {'additional_landed_cost': value}, context=context)

        return True

class stock_landed_cost_lines(osv.osv):
    _name = 'stock.landed.cost.lines'
    _description = 'Stock Landed Cost Lines'

    def onchange_product_id(self, cr, uid, ids, product_id=False, context=None):
        result = {}
        if not product_id:
            return {'value': {'quantity': 0.0, 'price_unit': 0.0}}

        product = self.pool.get('product.product').browse(cr, uid, product_id, context=context)
        result['name'] = product.name
        result['split_method'] = product.split_method
        result['price_unit'] = product.standard_price
        result['account_id'] = product.property_account_expense and product.property_account_expense.id or False
        return {'value': result}

    _columns = {
        'name': fields.char('Description', size=256),
        'cost_id': fields.many2one('stock.landed.cost', 'Landed Cost', required=True, ondelete='cascade'),
        'product_id': fields.many2one('product.product', 'Product', required=True),
        'price_unit': fields.float('Unit Price', required=True, digits_compute= dp.get_precision('Product Price')),
        'split_method': fields.selection(product.SPLIT_METHOD, string='Split Method', required=True),
        'account_id': fields.many2one('account.account', 'Account', domain=[('type','<>','view'), ('type', '<>', 'closed')]),
    }

class stock_valuation_adjustment_lines(osv.osv):
    _name = 'stock.valuation.adjustment.lines'
    _description = 'Stock Valuation Adjustment Lines'

    def _amount_final(self, cr, uid, ids, name, args, context=None):
        result = {}
        for line in self.browse(cr, uid, ids, context=context):
            result[line.id] = {
                'former_cost_per_unit': 0.0,
                'final_cost': 0.0,
            }
            result[line.id]['former_cost_per_unit'] = (line.former_cost / line.quantity or 1.0)
            result[line.id]['final_cost'] = (line.former_cost + line.additional_landed_cost)
        return result

    _columns = {
        'name': fields.char('Description', size=256),
        'cost_id': fields.many2one('stock.landed.cost', 'Landed Cost', required=True, ondelete='cascade'),
        'cost_line_id': fields.many2one('stock.landed.cost.lines', 'Cost Line', readonly=True),
        'move_id': fields.many2one('stock.move', 'Stock Move', readonly=True),
        'account_move_id': fields.many2one('account.move', 'Journal Entry', readonly=True),
        'product_id': fields.many2one('product.product', 'Product', required=True),
        'quantity': fields.float('Quantity', digits_compute= dp.get_precision('Product Unit of Measure'), required=True),
        'weight': fields.float('Weight', digits_compute= dp.get_precision('Product Unit of Measure')),
        'volume': fields.float('Volume', digits_compute= dp.get_precision('Product Unit of Measure')),
        'former_cost': fields.float('Former Cost', digits_compute= dp.get_precision('Product Price')),
        'former_cost_per_unit': fields.function(_amount_final, multi='cost', string='Former Cost(Per Unit)', type='float', digits_compute= dp.get_precision('Account'), store=True),
        'additional_landed_cost': fields.float('Additional Landed Cost', digits_compute= dp.get_precision('Product Price')),
        'final_cost': fields.function(_amount_final, multi='cost', string='Final Cost', type='float', digits_compute= dp.get_precision('Account'), store=True),
        'flag': fields.selection([('original', 'Original'), ('duplicate', 'Duplicate')], 'Flag', readonly=True),
    }

    _defaults = {
        'quantity': 1.0,
        'weight': 1.0,
        'volume': 1.0,
        'flag': 'original',
    }

# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: