53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
from esphome.const import CONF_SIZE, CONF_TEXT
|
|
from esphome.cpp_generator import MockObjClass
|
|
|
|
from ..defines import CONF_MAIN
|
|
from ..lv_validation import lv_color, lv_text
|
|
from ..lvcode import LocalVariable, lv, lv_expr
|
|
from ..schemas import TEXT_SCHEMA
|
|
from ..types import WidgetType, lv_obj_t
|
|
from . import Widget
|
|
|
|
CONF_QRCODE = "qrcode"
|
|
CONF_DARK_COLOR = "dark_color"
|
|
CONF_LIGHT_COLOR = "light_color"
|
|
|
|
QRCODE_SCHEMA = TEXT_SCHEMA.extend(
|
|
{
|
|
cv.Optional(CONF_DARK_COLOR, default="black"): lv_color,
|
|
cv.Optional(CONF_LIGHT_COLOR, default="white"): lv_color,
|
|
cv.Required(CONF_SIZE): cv.int_,
|
|
}
|
|
)
|
|
|
|
|
|
class QrCodeType(WidgetType):
|
|
def __init__(self):
|
|
super().__init__(
|
|
CONF_QRCODE,
|
|
lv_obj_t,
|
|
(CONF_MAIN,),
|
|
QRCODE_SCHEMA,
|
|
modify_schema=TEXT_SCHEMA,
|
|
)
|
|
|
|
def get_uses(self):
|
|
return "canvas", "img", "label"
|
|
|
|
async def obj_creator(self, parent: MockObjClass, config: dict):
|
|
dark_color = await lv_color.process(config[CONF_DARK_COLOR])
|
|
light_color = await lv_color.process(config[CONF_LIGHT_COLOR])
|
|
size = config[CONF_SIZE]
|
|
return lv_expr.call("qrcode_create", parent, size, dark_color, light_color)
|
|
|
|
async def to_code(self, w: Widget, config):
|
|
if (value := config.get(CONF_TEXT)) is not None:
|
|
value = await lv_text.process(value)
|
|
with LocalVariable("qr_text", cg.std_string, value, modifier="") as str_obj:
|
|
lv.qrcode_update(w.obj, str_obj.c_str(), str_obj.size())
|
|
|
|
|
|
qr_code_spec = QrCodeType()
|