Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 21:06:19 +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

42
script/hassfest/ssdp.py
View File

@@ -18,16 +18,15 @@ SSDP = {}
def sort_dict(value):
"""Sort a dictionary."""
return OrderedDict((key, value[key])
for key in sorted(value))
return OrderedDict((key, value[key]) for key in sorted(value))
def generate_and_validate(integrations: Dict[str, Integration]):
"""Validate and generate ssdp data."""
data = {
'st': defaultdict(list),
'manufacturer': defaultdict(list),
'device_type': defaultdict(list),
"st": defaultdict(list),
"manufacturer": defaultdict(list),
"device_type": defaultdict(list),
}
for domain in sorted(integrations):
@@ -36,7 +35,7 @@ def generate_and_validate(integrations: Dict[str, Integration]):
if not integration.manifest:
continue
ssdp = integration.manifest.get('ssdp')
ssdp = integration.manifest.get("ssdp")
if not ssdp:
continue
@@ -44,19 +43,19 @@ def generate_and_validate(integrations: Dict[str, Integration]):
try:
with open(str(integration.path / "config_flow.py")) as fp:
content = fp.read()
if (' async_step_ssdp' not in content and
'register_discovery_flow' not in content):
integration.add_error(
'ssdp', 'Config flow has no async_step_ssdp')
if (
" async_step_ssdp" not in content
and "register_discovery_flow" not in content
):
integration.add_error("ssdp", "Config flow has no async_step_ssdp")
continue
except FileNotFoundError:
integration.add_error(
'ssdp',
'SSDP info in a manifest requires a config flow to exist'
"ssdp", "SSDP info in a manifest requires a config flow to exist"
)
continue
for key in 'st', 'manufacturer', 'device_type':
for key in "st", "manufacturer", "device_type":
if key not in ssdp:
continue
@@ -69,22 +68,21 @@ def generate_and_validate(integrations: Dict[str, Integration]):
def validate(integrations: Dict[str, Integration], config: Config):
"""Validate ssdp file."""
ssdp_path = config.root / 'homeassistant/generated/ssdp.py'
config.cache['ssdp'] = content = generate_and_validate(integrations)
ssdp_path = config.root / "homeassistant/generated/ssdp.py"
config.cache["ssdp"] = content = generate_and_validate(integrations)
with open(str(ssdp_path), 'r') as fp:
with open(str(ssdp_path), "r") as fp:
if fp.read().strip() != content:
config.add_error(
"ssdp",
"File ssdp.py is not up to date. "
"Run python3 -m script.hassfest",
fixable=True
"File ssdp.py is not up to date. " "Run python3 -m script.hassfest",
fixable=True,
)
return
def generate(integrations: Dict[str, Integration], config: Config):
"""Generate ssdp file."""
ssdp_path = config.root / 'homeassistant/generated/ssdp.py'
with open(str(ssdp_path), 'w') as fp:
fp.write(config.cache['ssdp'] + '\n')
ssdp_path = config.root / "homeassistant/generated/ssdp.py"
with open(str(ssdp_path), "w") as fp:
fp.write(config.cache["ssdp"] + "\n")