71 lines
2.5 KiB
Python
71 lines
2.5 KiB
Python
|
"""Register a repository."""
|
||
|
|
from aiogithubapi import AIOGitHubAPIException
|
||
|
|
|
||
|
|
from custom_components.hacs.helpers.classes.exceptions import (
|
||
|
|
HacsException,
|
||
|
|
HacsExpectedException,
|
||
|
|
)
|
||
|
|
from custom_components.hacs.share import get_hacs
|
||
|
|
|
||
|
|
from ...repositories import RERPOSITORY_CLASSES
|
||
|
|
|
||
|
|
|
||
|
|
# @concurrent(15, 5)
|
||
|
|
async def register_repository(full_name, category, check=True, ref=None):
|
||
|
|
"""Register a repository."""
|
||
|
|
hacs = get_hacs()
|
||
|
|
|
||
|
|
if full_name in hacs.common.skip:
|
||
|
|
if full_name != "hacs/integration":
|
||
|
|
raise HacsExpectedException(f"Skipping {full_name}")
|
||
|
|
|
||
|
|
if category not in RERPOSITORY_CLASSES:
|
||
|
|
raise HacsException(f"{category} is not a valid repository category.")
|
||
|
|
|
||
|
|
repository = RERPOSITORY_CLASSES[category](full_name)
|
||
|
|
if check:
|
||
|
|
try:
|
||
|
|
await repository.async_registration(ref)
|
||
|
|
if hacs.status.new:
|
||
|
|
repository.data.new = False
|
||
|
|
if repository.validate.errors:
|
||
|
|
hacs.common.skip.append(repository.data.full_name)
|
||
|
|
if not hacs.status.startup:
|
||
|
|
hacs.log.error("Validation for %s failed.", full_name)
|
||
|
|
if hacs.system.action:
|
||
|
|
raise HacsException(f"::error:: Validation for {full_name} failed.")
|
||
|
|
return repository.validate.errors
|
||
|
|
if hacs.system.action:
|
||
|
|
repository.logger.info("%s Validation completed", repository)
|
||
|
|
else:
|
||
|
|
repository.logger.info("%s Registration completed", repository)
|
||
|
|
except AIOGitHubAPIException as exception:
|
||
|
|
hacs.common.skip.append(repository.data.full_name)
|
||
|
|
raise HacsException(
|
||
|
|
f"Validation for {full_name} failed with {exception}."
|
||
|
|
) from None
|
||
|
|
|
||
|
|
exists = (
|
||
|
|
False
|
||
|
|
if str(repository.data.id) == "0"
|
||
|
|
else [x for x in hacs.repositories if str(x.data.id) == str(repository.data.id)]
|
||
|
|
)
|
||
|
|
|
||
|
|
if exists:
|
||
|
|
if exists[0] in hacs.repositories:
|
||
|
|
hacs.repositories.remove(exists[0])
|
||
|
|
|
||
|
|
else:
|
||
|
|
if hacs.hass is not None and (
|
||
|
|
(check and repository.data.new) or hacs.status.new
|
||
|
|
):
|
||
|
|
hacs.hass.bus.async_fire(
|
||
|
|
"hacs/repository",
|
||
|
|
{
|
||
|
|
"action": "registration",
|
||
|
|
"repository": repository.data.full_name,
|
||
|
|
"repository_id": repository.data.id,
|
||
|
|
},
|
||
|
|
)
|
||
|
|
hacs.repositories.append(repository)
|