summaryrefslogtreecommitdiff
path: root/indoteknik_api/controllers/controller.py
diff options
context:
space:
mode:
Diffstat (limited to 'indoteknik_api/controllers/controller.py')
-rw-r--r--indoteknik_api/controllers/controller.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/indoteknik_api/controllers/controller.py b/indoteknik_api/controllers/controller.py
index a3fb6fcb..7ceebbe3 100644
--- a/indoteknik_api/controllers/controller.py
+++ b/indoteknik_api/controllers/controller.py
@@ -186,11 +186,14 @@ class Controller(http.Controller):
@http.route('/api/image/<model>/<field>/<id>', auth='public', methods=['GET'])
def get_image(self, model, field, id, **kw):
+ model_name = model
+
model = request.env[model].sudo().search([('id', '=', id)], limit=1)
image = model[field] if model[field] else ''
+ model_with_watermark = ['product.template', 'product.product']
watermark = kw.get('watermark', '')
- if watermark.lower() == 'true':
+ if watermark.lower() == 'true' or model_name in model_with_watermark:
image = self.add_watermark_to_image(image)
response_headers = [