[IMP] report_qr: black, isort

This commit is contained in:
Joan Sisquella
2020-03-02 10:59:46 +01:00
committed by Kiplangat Dan
parent 1a52e3dab5
commit ba66252734
4 changed files with 41 additions and 45 deletions

View File

@@ -1,20 +1,21 @@
from odoo import api, models
import qrcode
from qrcode.image import svg, pil
import io
import qrcode
from qrcode.image import pil, svg
from odoo import api, models
class IrActionsReport(models.Model):
_inherit = 'ir.actions.report'
_inherit = "ir.actions.report"
@api.model
def qr_generate(self, value, box_size=3, border=5,
factory='png', **kwargs):
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,
"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")
@@ -23,12 +24,11 @@ class IrActionsReport(models.Model):
# 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)
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)
img = qr.make_image(fill_color=fill_color, back_color=back_color)
arr = io.BytesIO()
img.save(arr)
return arr.getvalue()