1 files changed, 97 insertions, 0 deletions
diff --git a/addons/pos_cache/models/pos_cache.py b/addons/pos_cache/models/pos_cache.py
new file mode 100644
index 00000000000..0ae91b8bfeb
--- /dev/null
+++ b/addons/pos_cache/models/pos_cache.py
@@ -0,0 +1,97 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+from ast import literal_eval
+import cPickle
+
+from openerp import models, fields, api
+
+
+class pos_cache(models.Model):
+    _name = 'pos.cache'
+
+    cache = fields.Binary()
+    product_domain = fields.Text(required=True)
+    product_fields = fields.Text(required=True)
+
+    config_id = fields.Many2one('pos.config', ondelete='cascade', required=True)
+    compute_user_id = fields.Many2one('res.users', 'Cache compute user', required=True)
+
+    @api.model
+    def refresh_all_caches(self):
+        self.env['pos.cache'].search([]).refresh_cache()
+
+    @api.one
+    def refresh_cache(self):
+        products = self.env['product.product'].search(self.get_product_domain())
+        prod_ctx = products.with_context(pricelist=self.config_id.pricelist_id.id, display_default_code=False)
+        prod_ctx = prod_ctx.sudo(self.compute_user_id.id)
+        res = prod_ctx.read(self.get_product_fields())
+        datas = {
+            'cache': cPickle.dumps(res, protocol=cPickle.HIGHEST_PROTOCOL),
+        }
+
+        self.write(datas)
+
+    @api.model
+    def get_product_domain(self):
+        return literal_eval(self.product_domain)
+
+    @api.model
+    def get_product_fields(self):
+        return literal_eval(self.product_fields)
+
+    @api.model
+    def get_cache(self, domain, fields):
+        if not self.cache or domain != self.get_product_domain() or fields != self.get_product_fields():
+            self.product_domain = str(domain)
+            self.product_fields = str(fields)
+            self.refresh_cache()
+
+        return cPickle.loads(self.cache)
+
+
+class pos_config(models.Model):
+    _inherit = 'pos.config'
+
+    @api.one
+    @api.depends('cache_ids')
+    def _get_oldest_cache_time(self):
+        pos_cache = self.env['pos.cache']
+        oldest_cache = pos_cache.search([('config_id', '=', self.id)], order='write_date', limit=1)
+        if oldest_cache:
+            self.oldest_cache_time = oldest_cache.write_date
+
+    # Use a related model to avoid the load of the cache when the pos load his config
+    cache_ids = fields.One2many('pos.cache', 'config_id')
+    oldest_cache_time = fields.Datetime(compute='_get_oldest_cache_time', string='Oldest cache time', readonly=True)
+
+    def _get_cache_for_user(self):
+        pos_cache = self.env['pos.cache']
+        cache_for_user = pos_cache.search([('id', 'in', self.cache_ids.ids), ('compute_user_id', '=', self.env.uid)])
+
+        if cache_for_user:
+            return cache_for_user[0]
+        else:
+            return None
+
+    @api.multi
+    def get_products_from_cache(self, fields, domain):
+        cache_for_user = self._get_cache_for_user()
+
+        if cache_for_user:
+            return cache_for_user.get_cache(domain, fields)
+        else:
+            pos_cache = self.env['pos.cache']
+            pos_cache.create({
+                'config_id': self.id,
+                'product_domain': str(domain),
+                'product_fields': str(fields),
+                'compute_user_id': self.env.uid
+            })
+            new_cache = self._get_cache_for_user()
+            return new_cache.get_cache(domain, fields)
+
+    @api.one
+    def delete_cache(self):
+        # throw away the old caches
+        self.cache_ids.unlink()