From 82f4612f53187bb167794cc0695ff16e3ccceb5e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Thibault=20Delavall=C3=A9e?= <tde@openerp.com>
Date: Mon, 30 Jul 2012 12:33:01 +0200
Subject: [PATCH] [REF] pos.category: category_image removed and replaced by
 image, image_medium and image_small, to standardize through OpenERP.

bzr revid: tde@openerp.com-20120730103301-rw0s7077m3ed7vj3
---
 addons/point_of_sale/point_of_sale.py | 54 +++++++++++++++++----------
 1 file changed, 35 insertions(+), 19 deletions(-)

diff --git a/addons/point_of_sale/point_of_sale.py b/addons/point_of_sale/point_of_sale.py
index d6eae7e3ddf..93345c45d55 100644
--- a/addons/point_of_sale/point_of_sale.py
+++ b/addons/point_of_sale/point_of_sale.py
@@ -31,6 +31,7 @@ from PIL import Image
 
 import netsvc
 from osv import fields, osv
+import tools
 from tools.translate import _
 from decimal import Decimal
 import decimal_precision as dp
@@ -1174,20 +1175,21 @@ class pos_category(osv.osv):
         res = self.name_get(cr, uid, ids, context=context)
         return dict(res)
 
-    def _get_small_image(self, cr, uid, ids, prop, unknow_none, context=None):
-        result = {}
+    def _get_image(self, cr, uid, ids, name, args, context=None):
+        result = dict.fromkeys(ids, False)
         for obj in self.browse(cr, uid, ids, context=context):
-            if not obj.category_image:
-                result[obj.id] = False
-                continue
-
-            image_stream = io.BytesIO(obj.category_image.decode('base64'))
-            img = Image.open(image_stream)
-            img.thumbnail((120, 100), Image.ANTIALIAS)
-            img_stream = StringIO.StringIO()
-            img.save(img_stream, "JPEG")
-            result[obj.id] = img_stream.getvalue().encode('base64')
+            resized_image_dict = tools.get_resized_images(obj.image)
+            result[obj.id] = {
+                'image_medium': resized_image_dict['image_medium'],
+                'image_small': resized_image_dict['image_small'],
+                }
         return result
+    
+    def _set_image(self, cr, uid, id, name, value, args, context=None):
+        return self.write(cr, uid, [id], {'image': tools.resize_image_big(value)}, context=context)
+    
+    def onchange_image(self, cr, uid, ids, value, context=None):
+        return {'value': tools.get_resized_images(value)}
 
     _columns = {
         'name': fields.char('Name', size=64, required=True, translate=True),
@@ -1195,20 +1197,34 @@ class pos_category(osv.osv):
         'parent_id': fields.many2one('pos.category','Parent Category', select=True),
         'child_id': fields.one2many('pos.category', 'parent_id', string='Children Categories'),
         'sequence': fields.integer('Sequence', help="Gives the sequence order when displaying a list of product categories."),
-        'category_image': fields.binary('Image'),
-        'category_image_small': fields.function(_get_small_image, string='Small Image', type="binary",
+        'image': fields.binary("Image",
+            help="This field holds the image used for the category. "\
+                 "The image is base64 encoded, and PIL-supported. "\
+                 "It is limited to a 1024x1024 px image."),
+        'image_medium': fields.function(_get_image, fnct_inv=_set_image,
+            string="Medium-sized image", type="binary", multi="_get_image",
             store = {
-                'pos.category': (lambda self, cr, uid, ids, c={}: ids, ['category_image'], 10),
-            }),
+                'pos.category': (lambda self, cr, uid, ids, c={}: ids, ['image'], 10),
+            },
+            help="Medium-sized image of the category. It is automatically "\
+                 "resized as a 180x180 px image, with aspect ratio kept. "\
+                 "Use this field in form views or some kanban views."),
+        'image_small': fields.function(_get_image, fnct_inv=_set_image,
+            string="Smal-sized image", type="binary", multi="_get_image",
+            store = {
+                'pos.category': (lambda self, cr, uid, ids, c={}: ids, ['image'], 10),
+            },
+            help="Small-sized image of the category. It is automatically "\
+                 "resized as a 50x50 px image, with aspect ratio keps. "\
+                 "Use this field anywhere a small image is required."),
     }
 
     def _get_default_image(self, cr, uid, context=None):
         image_path = openerp.modules.get_module_resource('point_of_sale', 'images', 'default_category_photo.png')
-        return open(image_path, 'rb').read().encode('base64')
-
+        return tools.resize_image_big(open(image_path, 'rb').read().encode('base64'))
 
     _defaults = {
-        'category_image': _get_default_image,
+        'image': _get_default_image,
     }
 
 pos_category()