eiru_purchases/controllers/main.py

# -*- coding: utf-8 -*-
from openerp import http
from openerp.http import request
from werkzeug.wrappers import Response
from werkzeug.datastructures import Headers
from datetime import datetime
from dateutil.relativedelta import relativedelta as rd
from dateutil.parser import parse
from gzip import GzipFile
from StringIO import StringIO as IO
import simplejson as json
import gzip
import logging

LOGGER = logging.getLogger(__name__)
DATE_FORMAT = '%Y-%m-%d'
GZIP_COMPRESSION_LEVEL = 9

class Purchases(http.Controller):

    '''
        Get server date to send
    '''
    def get_server_date(self):
        return datetime.now().strftime(DATE_FORMAT)

    '''
        Get current user information
    '''
    def get_user(self):
        user = request.env.user

        return {
            'id': user.id,
            'name': user.name,
            'displayName': user.display_name,
            'currency': {
                'id': user.company_id.currency_id.id,
                'name': user.company_id.currency_id.name,
                'displayName': user.company_id.currency_id.display_name,
                'symbol': user.company_id.currency_id.symbol
            },
            'company': {
                'id': user.company_id.id,
                'name': user.company_id.name,
                'displayName': user.company_id.display_name
            }
        }

    '''
        Get currencies
    '''
    def get_currencies(self):
        return [{
            'id': currency.id,
            'name': currency.name,
            'displayName': currency.display_name,
            'base': currency.base,
            'accuracy': currency.accuracy,
            'rateSilent': currency.rate_silent,
            'rounding': currency.rounding,
            'symbol': currency.symbol,
            'position': currency.position,
            'decimalSeparator': currency.decimal_separator,
            'decimalPlaces': currency.decimal_places,
            'thousandsSeparator': currency.thousands_separator
        } for currency in request.env['res.currency'].search([('active', '=', True)])]

    '''
        Get all active journals
    '''
    def get_journals(self):
        return [{
            'id': journal.id,
            'name': journal.name,
            'displayName': journal.display_name,
            'code': journal.code,
            'cashControl': journal.cash_control,
            'type': journal.type,
            'currency': {
                'id': journal.currency.id,
                'name': journal.currency.name,
                'displayName': journal.currency.display_name
            },
            'defaultDebitAccount': {
                'id': journal.default_debit_account_id.id,
                'name': journal.default_debit_account_id.name,
                'displayName': journal.default_debit_account_id.display_name,
                'code': journal.default_debit_account_id.code,
                'exchange_rate': journal.default_credit_account_id.exchange_rate,
                'foreignBalance': journal.default_credit_account_id.foreign_balance,
                'reconcile': journal.default_credit_account_id.reconcile,
                'debit': journal.default_credit_account_id.debit,
                'credit': journal.default_credit_account_id.credit,
                'currencyMode': journal.default_credit_account_id.currency_mode,
                'companyCurrency': {
                    'id': journal.default_credit_account_id.company_currency_id.id,
                    'name': journal.default_credit_account_id.company_currency_id.name,
                    'displayName': journal.default_credit_account_id.company_currency_id.display_name,
                },
                'currency': {
                    'id': journal.default_credit_account_id.currency_id.id,
                    'name': journal.default_credit_account_id.currency_id.name,
                    'displayName': journal.default_credit_account_id.currency_id.display_name
                }

            }
        } for journal in request.env['account.journal'].search([('type', 'in', ['bank', 'cash']), ('default_debit_account_id.currency_id', '=', False), ('active', '=', True)], order='id')]

    '''
        Get all active suppliers
    '''
    def get_suppliers(self):
        return [{
            'id': supplier.id,
            'name': supplier.name,
            'displayName': supplier.display_name,
            'imageMedium': supplier.image_medium,
            'phone': supplier.phone,
            'mobile': supplier.mobile,
            'email': supplier.email
        } for supplier in request.env['res.partner'].search([('supplier', '=', True), ('active', '=', True)])]

    '''
        Get all purchasable and active products
    '''
    def get_products(self):
        return [{
            'id': product.id,
            'name': product.name,
            'displayName': product.display_name,
            'ean13': product.ean13,
            'imageMedium': product.image_medium,
            'standardPrice': product.standard_price,
            'variantCount': product.product_variant_count,
            'quantity': 1,
            'price': product.standard_price,
            'variants': [{
                'id': variant.id,
                'name': variant.name,
                'displayName': variant.display_name,
                'ean13': variant.ean13,
                'imageMedium': variant.image_medium,
                'standardPrice': variant.standard_price,
                'quantity': 1,
                'price': variant.standard_price
            } for variant in product.product_variant_ids if variant.active]
        } for product in request.env['product.template'].search([('purchase_ok', '=', True), ('standard_price', '>=', 0), ('active', '=', True)])]
        
    '''
        Get all incoming and active picking types
    '''
    def get_picking_types(self):
        return [{
            'id': picking_type.id,
            'name': picking_type.name,
            'displayName': picking_type.display_name
        } for picking_type in request.env['stock.picking.type'].search([('code', '=', 'incoming'), ('active', '=', True)])]
    
    '''
        Get all active payment terms
    '''
    def get_payment_terms(self):
        return [{
            'id': payment_term.id,
            'name': payment_term.name,
            'displayName': payment_term.display_name,
            'lines': [{
                'id': line.id,
                'days': line.days,
                'days2': line.days2,
                'value': line.value,
                'valueAmount': line.value_amount
            } for line in payment_term.line_ids]
        } for payment_term in request.env['account.payment.term'].search([('active', '=', True)])]

    '''
        Make JSON response to send
    '''
    def make_response(self, data=None, status=200):
        return Response(json.dumps(data), status=status, content_type='application/json')

    '''
        Make GZIP to JSON response
    '''
    def make_gzip_response(self, data=None, status=200):
        gzip_buffer = IO()
        with GzipFile(mode='wb', compresslevel=GZIP_COMPRESSION_LEVEL, fileobj=gzip_buffer) as gzip_file:
            gzip_file.write(json.dumps(data))
        
        contents = gzip_buffer.getvalue()
        gzip_buffer.close()

        headers = Headers()
        headers.add('Content-Encoding', 'gzip')
        headers.add('Vary', 'Accept-Encoding')
        headers.add('Content-Length', len(contents))

        return Response(contents, status=status, headers=headers, content_type='application/json')

    '''
    '''
    def make_info_log(self, log):
        LOGGER.info('\033[1;34m[INFO] --> \033[m{}'.format(log))

    '''
        New purchase resource route
    '''
    @http.route('/eiru_purchases/init', auth='user', methods=['GET'], cors='*')
    def init_purchase(self, **kw):
        self.make_info_log('Sending JSON response')

        return self.make_gzip_response({
            'date': self.get_server_date(),
            'user': self.get_user(),
            'currencies': self.get_currencies(),
            'journals': self.get_journals(),
            'suppliers': self.get_suppliers(),
            'products': self.get_products(),
            'pickingTypes': self.get_picking_types(),
            'paymentTerms': self.get_payment_terms()
        })

    '''
        Create supplier and return data
    '''
    @http.route('/eiru_purchases/create_supplier', type='json', auth='user', methods=['POST'], cors='*')
    def create_supplier(self, **kw):
        self.make_info_log('Creating supplier')

        supplier = request.env['res.partner'].create({
            'name': kw.get('name'),
            'ruc': kw.get('ruc'),
            'phone': kw.get('phone'),
            'supplier': True,
            'customer': False
        })

        return {
            'id': supplier.id,
            'name': supplier.name,
            'displayName': supplier.display_name,
            'imageMedium': supplier.image_medium,
            'phone': supplier.phone,
            'mobile': supplier.mobile,
            'email': supplier.email
        }

    '''
        Create product and return data
    '''
    @http.route('/eiru_purchases/create_product', type='json', auth='user', methods=['POST'], cors='*')
    def create_product(self, **kw):
        self.make_info_log('Creating product')

        product = request.env['product.template'].create({
            'name': kw.get('name'),
            'standard_price': float(kw.get('price')),
            'ean13': kw.get('ean13')
        })

        return {
            'id': product.id,
            'name': product.name,
            'displayName': product.display_name,
            'ean13': product.ean13,
            'imageMedium': product.image_medium,
            'listPrice': product.list_price,
            'variantCount': product.product_variant_count,
            'variants': [{
                'id': variant.id,
                'name': variant.name,
                'displayName': variant.display_name,
                'ean13': variant.ean13,
                'imageMedium': variant.image_medium,
                'standardPrice': variant.standard_price
            } for variant in product.product_variant_ids if variant.active]
        }

    '''
        Get currency id based on journal
    '''
    def get_currency_id(self, journal_id):
        journal = request.env['account.journal'].browse(journal_id)
        return journal.default_debit_account_id.currency_id.id or journal.default_debit_account_id.company_currency_id.id


    '''
        Check currency in pricelist and return it
    '''
    def get_pricelist_id(self, currency_id):
        pricelist = request.env['product.pricelist'].search([('active', '=', True), ('type', '=', 'sale')])

        if not True in pricelist.mapped(lambda p: p.currency_id.id == currency_id):
            pricelist = pricelist[0].copy()
            pricelist.write({
                'currency_id': currency_id
            })
        else:
            pricelist = pricelist.filtered(lambda p: p.currency_id.id == currency_id)
        
        return pricelist.id

    '''
        Get default location
    '''
    def get_stock_location_id(self):
        stock_location = request.env['stock.location'].search([('usage', '=', 'internal')])

        return stock_location.id

    '''
        Create purchase order from cart and return id
    '''
    def create_purchase_order(self, supplier_id, cart_items, date_order, currency_id, pricelist_id, payment_term_id=None):
        return request.env['purchase.order'].create({
            'name': '/',
            'partner_id': supplier_id,
            'order_line': [[0, False, {
                'name': line.get('name'),
                'date_planned': date_order,
                'product_id': int(line.get('id')),
                'product_qty': float(line.get('quantity')),
                'price_unit': float(line.get('price'))
            }] for line in cart_items],
            'date_order': date_order,
            'currency_id': currency_id,
            'pricelist_id': pricelist_id,
            'payment_term_id': payment_term_id,
            'location_id': self.get_stock_location_id(),
            'invoice_method': 'manual',
            'state': 'draft'
        })

    '''
        Confirm purchase order
    '''
    def confirm_purchase_order(self, purchase_order_id):
        purchase_order = request.env['purchase.order'].browse(purchase_order_id)
        purchase_order.write({
            'state': 'confirmed'
        })

        purchase_order.action_purchase_confirm()

        for picking in purchase_order.picking_ids:
                picking.force_assign()
                picking.action_done()
    
    '''
        Purchase order
    '''
    def create_invoice(self, purchase_order_id):
        purchase_order_obj = request.env['purchase.order']
        purchase_order_line_obj = request.env['purchase.order.line']
        purchase_order_line_invoice_obj = request.env['purchase.order.line_invoice']
        account_invoice_line_obj = request.env['account.invoice.line']

        purchase_order = purchase_order_obj.browse(purchase_order_id)
        invoice_lines = {}

        for line in purchase_order.order_line:
            if (not line.invoiced) and (line.state not in ('draft', 'cancel')):
                if not line.partner_id.id in invoice_lines:
                    invoice_lines[line.partner_id.id] = []

                account_id = purchase_order_obj._choose_account_from_po_line(line)
                account_invoice_line_values = purchase_order_obj._prepare_inv_line(account_id, line)
                account_invoice_line_values.update({
                    'origin': line.order_id.name
                })

                account_invoice_line_id = account_invoice_line_obj.create(account_invoice_line_values)
            
                purchase_order_line_obj.with_context({
                    'invoiced': True,
                    'invoice_lines': [(4, account_invoice_line_id.id)]
                }).write([line.id])

                invoice_lines[line.partner_id.id].append((line, account_invoice_line_id.id))

        invoice_ids = []
        for value in invoice_lines.values():
            invoice_line = map(lambda x : x[1], value)
            orders = list(set(map(lambda x : x[0].order_id, value)))

            new_id = purchase_order_line_invoice_obj._make_invoice_by_partner(orders[0].partner_id, orders, invoice_line)
            invoice_ids.append(new_id)

        return invoice_ids

    '''
        Purchase processing resource route
    ''' 
    @http.route('/eiru_purchases/process', type='json', auth='user', methods=['POST'], cors='*')
    def process_purchase(self, **kw):
        self.make_info_log('Processing purchase...')

        # # Get date
        date_now = datetime.now().strftime(DATE_FORMAT)
        self.make_info_log('Getting date')

        # Get currency
        currency_id = self.get_currency_id(kw.get('journalId'))
        self.make_info_log('Getting currency')

        # Get pricelist
        pricelist_id = self.get_pricelist_id(currency_id)
        self.make_info_log('Product pricelist checked')

        # Create purchase order 
        purchase_order = self.create_purchase_order(kw.get('supplierId'), kw.get('items'), date_now, currency_id, pricelist_id, kw.get('paymentTermId'))
        self.make_info_log('Purchase order created')

        # Confirm purchase 
        self.confirm_purchase_order(purchase_order.id)
        self.make_info_log('Purchase order confirmed')

        # Create invoice
        invoice_ids = self.create_invoice(purchase_order.id)
        self.make_info_log('Invoice created')

        print(invoice_ids)

        return {
            'status': 'ok'
        }