~openerp-commiter/openobject-addons/trunk-extra-addons

# -*- encoding: utf-8 -*-

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

# Copyright (C) 2009  Sharoon Thomas, Open Labs Business solutions      #

# Copyright (C) 2011 Akretion Sébastien BEAU sebastien.beau@akretion.com#

#                                                                       #

#This program is free software: you can redistribute it and/or modify   #

#it under the terms of the GNU 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 General Public License for more details.                           #

#                                                                       #

#You should have received a copy of the GNU General Public License      #

#along with this program.  If not, see <http://www.gnu.org/licenses/>.  #

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

import os

import shutil

import logging

import unicodedata

import base64, urllib

 

from osv import osv,fields

from tools.translate import _

 

class product_product(osv.osv):

    _inherit = "product.product"

 

    def copy(self, cr, uid, id, default=None, context=None):

        if not default:

            default = {}

        original = self.read(cr, uid, id, fields=['default_code', 'image_ids'], context=context)

        default.update({

            'default_code': original['default_code'] + '.copy',

            'images_ids': False,

        })

        local_media_repository = self.pool.get('res.company').get_local_media_repository(cr, uid, context=context)

        if local_media_repository:

            if original['image_ids']:

                old_path = os.path.join(local_media_repository, original['default_code'])

                if os.path.isdir(old_path):

                    try:

                        shutil.copytree(old_path, old_path+'.copy')

                    except:

                        logger = logging.getLogger('product_images_olbs')

                        logger.exception('error while trying to copy images from %s to %s', old_path, old_path+'.copy')

 

        return super(product_product, self).copy(cr, uid, id, default, context=context)

 

    def get_main_image(self, cr, uid, id, context=None):

        if isinstance(id, list):

            id = id[0]

        images_ids = self.read(cr, uid, id, ['image_ids'], context=context)['image_ids']

        if images_ids:

            return images_ids[0]

        return False

 

    def _get_main_image(self, cr, uid, ids, field_name, arg, context=None):

        res = {}

        img_obj = self.pool.get('product.images')

        for id in ids:

            image_id = self.get_main_image(cr, uid, id, context=context)

            if image_id:

                image = img_obj.browse(cr, uid, image_id, context=context)

                res[id] = image.file

            else:

                res[id] = False

        return res

 

    _columns = {

        'image_ids':fields.one2many(

                'product.images',

                'product_id',

                'Product Images'

        ),

        'default_code': fields.char('Reference', size=64, require='True'),

        'product_image': fields.function(_get_main_image, type="binary", method=True),

    }

 

    def write(self, cr, uid, ids, vals, context=None):

        if isinstance(ids, (int, long)):

            ids = [ids]

        # here we expect that the write on default_code is always on 1 product because there is an unique constraint on the default code

        if vals.get('default_code', False) and ids:

            local_media_repository = self.pool.get('res.company').get_local_media_repository(cr, uid, context=context)

            if local_media_repository:

                old_product = self.read(cr, uid, ids[0], ['default_code', 'image_ids'], context=context)

                res = super(product_product, self).write(cr, uid, ids, vals, context=context)

                if old_product['image_ids']:

                    if old_product['default_code'] != vals['default_code']:

                        old_path = os.path.join(local_media_repository, old_product['default_code'])

                        new_path = os.path.join(local_media_repository, vals['default_code'])

                        if os.path.isdir(old_path):

                            if not os.path.isdir(os.path.dirname(new_path)):

                                os.makedirs(os.path.dirname(new_path))

                            if os.path.isdir(new_path):

                                shutil.rmtree(new_path)

                            shutil.copytree(old_path, new_path)

                            shutil.rmtree(old_path)

                return res

        return super(product_product, self).write(cr, uid, ids, vals, context=context)

 

    #This constraint should be by default in openerp

    _sql_constraints = [('default_code', 'UNIQUE(default_code)',

                _('Default code should be uniq'))]

 

    def create_image_from_url(self, cr, uid, id, url, image_name=None, context=None):

        (filename, header) = urllib.urlretrieve(url)

        f = open(filename , 'rb')

        data = f.read()

        f.close()

        img = base64.encodestring(data)

        filename, extention = os.path.splitext(os.path.basename(url))

        data = {'name': image_name or filename,

            'extention': extention,

            'file': img,

            'product_id': id,

            }

        new_image_id = self.pool.get('product.images').create(cr, uid, data, context=context)

        return True

 

 

product_product()