Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-25 05:26:47 +00:00
Code Activity

Black

Browse Source
This commit is contained in:
Paulus Schoutsen
2019-07-31 12:25:30 -07:00
parent da05dfe708
commit 4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
Download Patch File Download Diff File Expand all files Collapse all files

138
homeassistant/components/config/auth_provider_homeassistant.py
View File

@@ -5,41 +5,41 @@ from homeassistant.auth.providers import homeassistant as auth_ha
from homeassistant.components import websocket_api
WS_TYPE_CREATE = 'config/auth_provider/homeassistant/create'
SCHEMA_WS_CREATE = websocket_api.BASE_COMMAND_MESSAGE_SCHEMA.extend({
vol.Required('type'): WS_TYPE_CREATE,
vol.Required('user_id'): str,
vol.Required('username'): str,
vol.Required('password'): str,
})
WS_TYPE_CREATE = "config/auth_provider/homeassistant/create"
SCHEMA_WS_CREATE = websocket_api.BASE_COMMAND_MESSAGE_SCHEMA.extend(
{
vol.Required("type"): WS_TYPE_CREATE,
vol.Required("user_id"): str,
vol.Required("username"): str,
vol.Required("password"): str,
}
)
WS_TYPE_DELETE = 'config/auth_provider/homeassistant/delete'
SCHEMA_WS_DELETE = websocket_api.BASE_COMMAND_MESSAGE_SCHEMA.extend({
vol.Required('type'): WS_TYPE_DELETE,
vol.Required('username'): str,
})
WS_TYPE_DELETE = "config/auth_provider/homeassistant/delete"
SCHEMA_WS_DELETE = websocket_api.BASE_COMMAND_MESSAGE_SCHEMA.extend(
{vol.Required("type"): WS_TYPE_DELETE, vol.Required("username"): str}
)
WS_TYPE_CHANGE_PASSWORD = 'config/auth_provider/homeassistant/change_password'
SCHEMA_WS_CHANGE_PASSWORD = websocket_api.BASE_COMMAND_MESSAGE_SCHEMA.extend({
vol.Required('type'): WS_TYPE_CHANGE_PASSWORD,
vol.Required('current_password'): str,
vol.Required('new_password'): str
})
WS_TYPE_CHANGE_PASSWORD = "config/auth_provider/homeassistant/change_password"
SCHEMA_WS_CHANGE_PASSWORD = websocket_api.BASE_COMMAND_MESSAGE_SCHEMA.extend(
{
vol.Required("type"): WS_TYPE_CHANGE_PASSWORD,
vol.Required("current_password"): str,
vol.Required("new_password"): str,
}
)
async def async_setup(hass):
"""Enable the Home Assistant views."""
hass.components.websocket_api.async_register_command(
WS_TYPE_CREATE, websocket_create,
SCHEMA_WS_CREATE
WS_TYPE_CREATE, websocket_create, SCHEMA_WS_CREATE
)
hass.components.websocket_api.async_register_command(
WS_TYPE_DELETE, websocket_delete,
SCHEMA_WS_DELETE
WS_TYPE_DELETE, websocket_delete, SCHEMA_WS_DELETE
)
hass.components.websocket_api.async_register_command(
WS_TYPE_CHANGE_PASSWORD, websocket_change_password,
SCHEMA_WS_CHANGE_PASSWORD
WS_TYPE_CHANGE_PASSWORD, websocket_change_password, SCHEMA_WS_CHANGE_PASSWORD
)
return True
@@ -47,10 +47,10 @@ async def async_setup(hass):
def _get_provider(hass):
"""Get homeassistant auth provider."""
for prv in hass.auth.auth_providers:
if prv.type == 'homeassistant':
if prv.type == "homeassistant":
return prv
raise RuntimeError('Provider not found')
raise RuntimeError("Provider not found")
@websocket_api.require_admin
@@ -60,34 +60,43 @@ async def websocket_create(hass, connection, msg):
provider = _get_provider(hass)
await provider.async_initialize()
user = await hass.auth.async_get_user(msg['user_id'])
user = await hass.auth.async_get_user(msg["user_id"])
if user is None:
connection.send_message(websocket_api.error_message(
msg['id'], 'not_found', 'User not found'))
connection.send_message(
websocket_api.error_message(msg["id"], "not_found", "User not found")
)
return
if user.system_generated:
connection.send_message(websocket_api.error_message(
msg['id'], 'system_generated',
'Cannot add credentials to a system generated user.'))
connection.send_message(
websocket_api.error_message(
msg["id"],
"system_generated",
"Cannot add credentials to a system generated user.",
)
)
return
try:
await hass.async_add_executor_job(
provider.data.add_auth, msg['username'], msg['password'])
provider.data.add_auth, msg["username"], msg["password"]
)
except auth_ha.InvalidUser:
connection.send_message(websocket_api.error_message(
msg['id'], 'username_exists', 'Username already exists'))
connection.send_message(
websocket_api.error_message(
msg["id"], "username_exists", "Username already exists"
)
)
return
credentials = await provider.async_get_or_create_credentials({
'username': msg['username']
})
credentials = await provider.async_get_or_create_credentials(
{"username": msg["username"]}
)
await hass.auth.async_link_user(user, credentials)
await provider.data.async_save()
connection.send_message(websocket_api.result_message(msg['id']))
connection.send_message(websocket_api.result_message(msg["id"]))
@websocket_api.require_admin
@@ -97,29 +106,30 @@ async def websocket_delete(hass, connection, msg):
provider = _get_provider(hass)
await provider.async_initialize()
credentials = await provider.async_get_or_create_credentials({
'username': msg['username']
})
credentials = await provider.async_get_or_create_credentials(
{"username": msg["username"]}
)
# if not new, an existing credential exists.
# Removing the credential will also remove the auth.
if not credentials.is_new:
await hass.auth.async_remove_credentials(credentials)
connection.send_message(
websocket_api.result_message(msg['id']))
connection.send_message(websocket_api.result_message(msg["id"]))
return
try:
provider.data.async_remove_auth(msg['username'])
provider.data.async_remove_auth(msg["username"])
await provider.data.async_save()
except auth_ha.InvalidUser:
connection.send_message(websocket_api.error_message(
msg['id'], 'auth_not_found', 'Given username was not found.'))
connection.send_message(
websocket_api.error_message(
msg["id"], "auth_not_found", "Given username was not found."
)
)
return
connection.send_message(
websocket_api.result_message(msg['id']))
connection.send_message(websocket_api.result_message(msg["id"]))
@websocket_api.async_response
@@ -127,8 +137,9 @@ async def websocket_change_password(hass, connection, msg):
"""Change user password."""
user = connection.user
if user is None:
connection.send_message(websocket_api.error_message(
msg['id'], 'user_not_found', 'User not found'))
connection.send_message(
websocket_api.error_message(msg["id"], "user_not_found", "User not found")
)
return
provider = _get_provider(hass)
@@ -137,25 +148,30 @@ async def websocket_change_password(hass, connection, msg):
username = None
for credential in user.credentials:
if credential.auth_provider_type == provider.type:
username = credential.data['username']
username = credential.data["username"]
break
if username is None:
connection.send_message(websocket_api.error_message(
msg['id'], 'credentials_not_found', 'Credentials not found'))
connection.send_message(
websocket_api.error_message(
msg["id"], "credentials_not_found", "Credentials not found"
)
)
return
try:
await provider.async_validate_login(
username, msg['current_password'])
await provider.async_validate_login(username, msg["current_password"])
except auth_ha.InvalidAuth:
connection.send_message(websocket_api.error_message(
msg['id'], 'invalid_password', 'Invalid password'))
connection.send_message(
websocket_api.error_message(
msg["id"], "invalid_password", "Invalid password"
)
)
return
await hass.async_add_executor_job(
provider.data.change_password, username, msg['new_password'])
provider.data.change_password, username, msg["new_password"]
)
await provider.data.async_save()
connection.send_message(
websocket_api.result_message(msg['id']))
connection.send_message(websocket_api.result_message(msg["id"]))