~openerp-mongolian/openerp-mongolization/7.0

# -*- coding: utf-8 -*-

##############################################################################

#

#    ShineERP, Enterprise Management Solution

#    Copyright (C) 2007-2013 ShineERP Co.,ltd (<http://www.erp.mn>). All Rights Reserved

#

#    Address : Suhkbaatar District, National ITPark, ShineERP LLC

#    Email : info@erp.mn

#    Phone : 976 + 11-318043

#

##############################################################################

 

from osv import osv,fields,orm

from tools.translate import _

import time

import netsvc, decimal, base64, os, time, xlrd

from tempfile import NamedTemporaryFile

 

class product_data_initial(osv.osv_memory):

    _name = 'product.data.initial'

    _description = 'Product Data Initial'

    

    _columns = {

        'data': fields.binary('Product Data File', required=True)

    }

    

    def import_data(self, cr, uid, ids, context={}):

        product_obj = self.pool.get('product.product')

        product_tmpl_obj = self.pool.get('product.template')

        product_category_obj = self.pool.get('product.category')

        product_uom_obj = self.pool.get('product.uom')

        

        product_uoms = {u'ш':'Unit(s)'}

        

        form = self.browse(cr, uid, ids[0])

        

        fileobj = NamedTemporaryFile('w+')

        fileobj.write(base64.decodestring(form.data))

        fileobj.seek(0)

        

        if not os.path.isfile(fileobj.name):

            raise osv.except_osv(u'Алдаа',u'Мэдээллийн файлыг уншихад алдаа гарлаа.\nЗөв файл эсэхийг шалгаад дахин оролдоно уу!')

        book = xlrd.open_workbook(fileobj.name)

        

        sheet = book.sheet_by_index(0)

        nrows = sheet.nrows

        

        rowi = 1

        

        while rowi < nrows :

            try:

                row = sheet.row(rowi)

                product_name = row[0].value

                product_category = u'' + row[1].value

                product_cost = row[2].value

                product_uom = row[3].value

                product_type = row[4].value

                product_supply_method = row[5].value

                product_procure_method = row[6].value

                product_sale_ok = row[7].value

                product_purchase_ok = row[8].value

                product_cost_method = row[9].value

                

                category_id = product_category_obj.search(cr, uid, [('name', '=', product_category)], limit=1)

                if len(category_id) ==0:

                    raise osv.except_osv('Error', 'Category does not exist! %s ' % product_category)

                product_uom_id = product_uom_obj.search(cr, uid, [('name', '=', product_uoms[product_uom] if product_uoms.has_key(product_uom) else '')], limit=1)

                if len(product_uom_id) ==0:

                    raise osv.except_osv('Error', 'Product uom does not exist! %s ' % product_uom)

                

                product_tmpl_id = product_tmpl_obj.create(cr, uid, {

                    'name': product_name,

                    'categ_id': category_id[0],

                    'standard_price': product_cost,

                    'uom_id': product_uom_id[0],

                    'type': product_type,

                    'supply_method': product_supply_method,

                    'procure_method': product_procure_method,

                    'purchase_ok': product_sale_ok,

                    'sale_ok': product_purchase_ok,

                    'cost_method': product_cost_method,

                    'company_id': 1,

                }, context=context)

                print 'created product_tmpl_id: ' + str(product_tmpl_id)

                product_id = product_obj.create(cr, uid, {

                    'product_tmpl_id': product_tmpl_id,

                    'active': True,

                    'valuation': 'manual_periodic',

                }, context=context)

                print 'created product_id: ' + str(product_id)

                

                rowi += 1

            except IndexError :

                raise osv.except_osv('Error', 'Excel sheet must be 10 columned : error on row %s ' % rowi)

        return True

product_data_initial()