[FIX] web_editor,website: add hook method in web editor to allow to add the website_id

Before this commit, the view create by editor (to change the scss file in asset for example) was not website specific.
Now we override (use the new hook method) to add the website_id.

1 files changed, 33 insertions, 0 deletions

diff --git a/addons/website/controllers/web_editor.py b/addons/website/controllers/web_editor.py
new file mode 100644
index 00000000000..d70e912018b
--- /dev/null
+++ b/addons/website/controllers/web_editor.py
@@ -0,0 +1,33 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+from odoo.addons.web_editor.controllers.main import Web_Editor
+from odoo.http import request
+
+
+class Web_Editor(Web_Editor):
+
+    def save_scss_view_hook(self):
+        res = super(Web_Editor, self).save_scss_view_hook()
+
+        website = request.env['website'].get_current_website()
+        if website:
+            res['website_id'] = website.id
+        return res
+
+    def save_scss_attachment_hook(self):
+        res = super(Web_Editor, self).save_scss_attachment_hook()
+
+        website = request.env['website'].get_current_website()
+        if website:
+            res['website_id'] = website.id
+        return res
+
+    def get_custom_attachment(self, custom_url, op='like'):
+        website = request.env['website'].get_current_website()
+        res = super(Web_Editor, self).get_custom_attachment(custom_url, op=op)
+        return res.with_context(website_id=website.id).filtered(lambda x: not x.website_id or x.website_id == website)
+
+    def get_custom_view(self, custom_url, op='='):
+        website = request.env['website'].get_current_website()
+        res = super(Web_Editor, self).get_custom_view(custom_url, op=op)
+        return res.with_context(website_id=website.id).filter_duplicate()