[feat] The configuration can also be loaded from an environment variable

This commit is contained in:
Sébastien Lucas
2018-10-23 20:27:52 +02:00
parent ea16a6306c
commit 096a864213
+4
View File
@@ -1,4 +1,5 @@
import yaml
from os import environ
import logging
_LOGGER = logging.getLogger(__name__)
@@ -6,6 +7,9 @@ _LOGGER = logging.getLogger(__name__)
def load_yaml(file):
try:
if environ.get("AQARA_MQTT_CONFIG") is not None:
stram = environ.get("AQARA_MQTT_CONFIG")
else:
stram = open(file, "r")
yaml_data = yaml.load(stram)
return yaml_data