mirror of
https://gitlab.com/flectra-community/reporting-engine.git
synced 2024-11-15 02:32:05 +00:00
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
|
from flectra import api, models
|
||
|
|
import qrcode
|
||
|
|
from qrcode.image import svg, pil
|
||
|
|
import io
|
||
|
|
|
||
|
|
|
||
|
|
class IrActionsReport(models.Model):
|
||
|
|
_inherit = 'ir.actions.report'
|
||
|
|
|
||
|
|
@api.model
|
||
|
|
def qr_generate(self, value, box_size=3, border=5,
|
||
|
|
factory='png', **kwargs):
|
||
|
|
factories = {
|
||
|
|
'png': pil.PilImage,
|
||
|
|
'svg': svg.SvgImage,
|
||
|
|
'svg-fragment': svg.SvgFragmentImage,
|
||
|
|
'svg-path': svg.SvgPathImage,
|
||
|
|
}
|
||
|
|
# Color parameters seem to be inverted in the library
|
||
|
|
back_color = kwargs.pop("back_color", "black")
|
||
|
|
fill_color = kwargs.pop("fill_color", "white")
|
||
|
|
try:
|
||
|
|
# Defaults to png if the argument is unknown
|
||
|
|
image_factory = factories.get(factory, pil.PilImage)
|
||
|
|
qr = qrcode.QRCode(
|
||
|
|
box_size=box_size, border=border,
|
||
|
|
image_factory=image_factory, **kwargs)
|
||
|
|
qr.add_data(value)
|
||
|
|
qr.make()
|
||
|
|
img = qr.make_image(fill_color=fill_color,
|
||
|
|
back_color=back_color)
|
||
|
|
arr = io.BytesIO()
|
||
|
|
img.save(arr)
|
||
|
|
return arr.getvalue()
|
||
|
|
except Exception:
|
||
|
|
raise ValueError("Cannot convert into barcode.")
|