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()