evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add charset to imap component. (#28258)

Browse source
This commit is contained in:
ZiroNL 2019-10-27 13:07:44 +01:00 • committed by Fabian Affolter
parent a9db2ead33
commit 6ac7796fb7
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
homeassistant/components/imap/sensor.py
View file

@ -23,6 +23,7 @@ _LOGGER = logging.getLogger(__name__)
CONF_SERVER = "server"
CONF_FOLDER = "folder"
CONF_SEARCH = "search"
CONF_CHARSET = "charset"
DEFAULT_PORT = 993
@ -35,6 +36,7 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
vol.Required(CONF_PASSWORD): cv.string,
vol.Required(CONF_SERVER): cv.string,
vol.Optional(CONF_PORT, default=DEFAULT_PORT): cv.port,
vol.Optional(CONF_CHARSET, default="utf-8"): cv.string,
vol.Optional(CONF_FOLDER, default="INBOX"): cv.string,
vol.Optional(CONF_SEARCH, default="UnSeen UnDeleted"): cv.string,
}
@ -49,6 +51,7 @@ async def async_setup_platform(hass, config, async_add_entities, discovery_info=
config.get(CONF_PASSWORD),
config.get(CONF_SERVER),
config.get(CONF_PORT),
config.get(CONF_CHARSET),
config.get(CONF_FOLDER),
config.get(CONF_SEARCH),
)
@ -62,13 +65,14 @@ async def async_setup_platform(hass, config, async_add_entities, discovery_info=
class ImapSensor(Entity):
"""Representation of an IMAP sensor."""
def __init__(self, name, user, password, server, port, folder, search):
def __init__(self, name, user, password, server, port, charset, folder, search):
"""Initialize the sensor."""
self._name = name or user
self._user = user
self._password = password
self._server = server
self._port = port
self._charset = charset
self._folder = folder
self._email_count = None
self._search = search
@ -150,7 +154,9 @@ class ImapSensor(Entity):
"""Check the number of found emails."""
if self._connection:
await self._connection.noop()
result, lines = await self._connection.search(self._search)
result, lines = await self._connection.search(
self._search, charset=self._charset
)
if result == "OK":
self._email_count = len(lines[0].split())