first commit

modules/product_image/routes.py

@@ -0,0 +1,96 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.
+from urllib.parse import unquote
+
+import trytond.config as config
+from trytond.protocols.wrappers import (
+    HTTPStatus, Response, abort, with_pool, with_transaction)
+from trytond.wsgi import app
+
+
+@app.route(
+    '/product/image/<code>/<base64:database_name>',
+    methods={'GET'})
+@app.route(
+    '/product/image/<code>/<base64:database_name>/<name>',
+    methods={'GET'})
+@with_pool
+@with_transaction()
+def product_image(request, pool, code, name=None):
+    Template = pool.get('product.template')
+    try:
+        template, = Template.search([
+                ('code', '=', unquote(code)),
+                ])
+    except ValueError:
+        abort(HTTPStatus.NOT_FOUND)
+    return _image(request, pool, template)
+
+
+@app.route(
+    '/product/variant/image/<code>/<base64:database_name>',
+    methods={'GET'})
+@app.route(
+    '/product/variant/image/<code>/<base64:database_name>/<name>',
+    methods={'GET'})
+@with_pool
+@with_transaction()
+def variant_image(request, pool, code, name=None):
+    Product = pool.get('product.product')
+    try:
+        product, = Product.search([
+                ('code', '=', unquote(code)),
+                ])
+    except ValueError:
+        abort(HTTPStatus.NOT_FOUND)
+    return _image(request, pool, product)
+
+
+@app.route(
+    '/product-category/image/<code>/<base64:database_name>',
+    methods={'GET'})
+@app.route(
+    '/product-category/image/<code>/<base64:database_name>/<name>',
+    methods={'GET'})
+@with_pool
+@with_transaction()
+def category_image(request, pool, code, name=None):
+    Category = pool.get('product.category')
+    try:
+        category, = Category.search([
+                ('code', '=', unquote(code)),
+                ])
+    except ValueError:
+        abort(HTTPStatus.NOT_FOUND)
+    return _image(request, pool, category)
+
+
+def _image(request, pool, record):
+    images = record.get_images(request.args)
+    if not images:
+        abort(HTTPStatus.NOT_FOUND)
+    try:
+        image = list(images)[int(request.args.get('i', 0))]
+    except IndexError:
+        abort(HTTPStatus.NOT_FOUND)
+    except ValueError:
+        abort(HTTPStatus.BAD_REQUEST)
+    try:
+        size = int(request.args.get('s', 400))
+    except ValueError:
+        abort(HTTPStatus.BAD_REQUEST)
+    try:
+        width = int(request.args.get('w', size))
+    except ValueError:
+        abort(HTTPStatus.BAD_REQUEST)
+    try:
+        height = int(request.args.get('h', size))
+    except ValueError:
+        abort(HTTPStatus.BAD_REQUEST)
+
+    response = Response(image.get((width, height)), mimetype='image/jpeg')
+    response.headers['Cache-Control'] = (
+        'max-age=%s, public' % config.getint(
+            'product', 'image_timeout', default=365 * 24 * 60 * 60))
+    response.add_etag()
+    return response