1111
1212from .api import NovastarClient
1313from .const import (
14+ CONF_ALLOW_RAW_COMMANDS ,
1415 CONF_ENCRYPTION ,
1516 CONF_PROJECT_ID ,
1617 CONF_SECRET_KEY ,
18+ DEFAULT_ALLOW_RAW_COMMANDS ,
1719 DEFAULT_ENCRYPTION ,
1820 DEFAULT_NAME ,
1921 DEFAULT_PORT ,
@@ -49,6 +51,7 @@ async def async_step_user(
4951 project_id = user_input [CONF_PROJECT_ID ]
5052 secret_key = user_input [CONF_SECRET_KEY ]
5153 encryption = user_input .get (CONF_ENCRYPTION , DEFAULT_ENCRYPTION )
54+ allow_raw_commands = user_input .get (CONF_ALLOW_RAW_COMMANDS , DEFAULT_ALLOW_RAW_COMMANDS )
5255
5356 # Validate credentials by testing connection
5457 client = NovastarClient (
@@ -71,6 +74,7 @@ async def async_step_user(
7174 CONF_PROJECT_ID : project_id ,
7275 CONF_SECRET_KEY : secret_key ,
7376 CONF_ENCRYPTION : encryption ,
77+ CONF_ALLOW_RAW_COMMANDS : allow_raw_commands ,
7478 },
7579 )
7680 errors ["base" ] = "cannot_connect"
@@ -84,6 +88,7 @@ async def async_step_user(
8488 vol .Required (CONF_PROJECT_ID ): str ,
8589 vol .Required (CONF_SECRET_KEY ): str ,
8690 vol .Optional (CONF_ENCRYPTION , default = DEFAULT_ENCRYPTION ): bool ,
91+ vol .Optional (CONF_ALLOW_RAW_COMMANDS , default = DEFAULT_ALLOW_RAW_COMMANDS ): bool ,
8792 vol .Optional (CONF_NAME , default = DEFAULT_NAME ): str ,
8893 }
8994 ),
@@ -151,6 +156,7 @@ async def async_step_manual(
151156 project_id = user_input [CONF_PROJECT_ID ]
152157 secret_key = user_input [CONF_SECRET_KEY ]
153158 encryption = user_input .get (CONF_ENCRYPTION , DEFAULT_ENCRYPTION )
159+ allow_raw_commands = user_input .get (CONF_ALLOW_RAW_COMMANDS , DEFAULT_ALLOW_RAW_COMMANDS )
154160
155161 # Validate credentials by testing connection
156162 client = NovastarClient (
@@ -173,6 +179,7 @@ async def async_step_manual(
173179 CONF_PROJECT_ID : project_id ,
174180 CONF_SECRET_KEY : secret_key ,
175181 CONF_ENCRYPTION : encryption ,
182+ CONF_ALLOW_RAW_COMMANDS : allow_raw_commands ,
176183 },
177184 )
178185 errors ["base" ] = "cannot_connect"
@@ -186,6 +193,7 @@ async def async_step_manual(
186193 vol .Required (CONF_PROJECT_ID ): str ,
187194 vol .Required (CONF_SECRET_KEY ): str ,
188195 vol .Optional (CONF_ENCRYPTION , default = DEFAULT_ENCRYPTION ): bool ,
196+ vol .Optional (CONF_ALLOW_RAW_COMMANDS , default = DEFAULT_ALLOW_RAW_COMMANDS ): bool ,
189197 vol .Optional (CONF_NAME , default = DEFAULT_NAME ): str ,
190198 }
191199 ),
@@ -225,6 +233,7 @@ async def async_step_credentials(
225233 CONF_PROJECT_ID : project_id ,
226234 CONF_SECRET_KEY : secret_key ,
227235 CONF_ENCRYPTION : encryption ,
236+ CONF_ALLOW_RAW_COMMANDS : DEFAULT_ALLOW_RAW_COMMANDS ,
228237 },
229238 )
230239 errors ["base" ] = "cannot_connect"
@@ -332,6 +341,7 @@ async def async_step_discovery_confirm(
332341 CONF_PROJECT_ID : project_id ,
333342 CONF_SECRET_KEY : secret_key ,
334343 CONF_ENCRYPTION : encryption ,
344+ CONF_ALLOW_RAW_COMMANDS : DEFAULT_ALLOW_RAW_COMMANDS ,
335345 },
336346 )
337347 errors ["base" ] = "cannot_connect"
0 commit comments