homeassistant/custom_components/hacs/hacsbase/task_factory.py

76 lines
2.5 KiB
Python

# pylint: disable=missing-docstring,invalid-name
import logging
import time
from datetime import timedelta
import asyncio
from aiogithubapi import AIOGitHubException
from custom_components.hacs.hacsbase.exceptions import HacsException
from custom_components.hacs.helpers.register_repository import register_repository
max_concurrent_tasks = asyncio.Semaphore(15)
sleeper = 5
logger = logging.getLogger("hacs.factory")
class HacsTaskFactory:
def __init__(self):
self.tasks = []
self.running = False
async def execute(self):
if not self.tasks:
logger.debug("No tasks to execute")
return
if self.running:
logger.debug("Allready executing tasks")
return
try:
self.running = True
logger.info("Processing %s tasks", len(self.tasks))
start = time.time()
await asyncio.gather(*self.tasks)
logger.info(
"Task processing of %s tasks completed in %s seconds",
len(self.tasks),
timedelta(seconds=round(time.time() - start)).seconds,
)
self.tasks = []
self.running = False
except RuntimeError:
logger.warning("RuntimeError, Clearing current tasks")
self.tasks = []
self.running = False
async def safe_common_update(self, repository):
async with max_concurrent_tasks:
try:
await repository.common_update()
except (AIOGitHubException, HacsException) as exception:
logger.error("%s - %s", repository.data.full_name, exception)
# Due to GitHub ratelimits we need to sleep a bit
await asyncio.sleep(sleeper)
async def safe_update(self, repository):
async with max_concurrent_tasks:
try:
await repository.update_repository()
except (AIOGitHubException, HacsException) as exception:
logger.error("%s - %s", repository.data.full_name, exception)
# Due to GitHub ratelimits we need to sleep a bit
await asyncio.sleep(sleeper)
async def safe_register(self, repo, category):
async with max_concurrent_tasks:
try:
await register_repository(repo, category)
except (AIOGitHubException, HacsException) as exception:
logger.error("%s - %s", repo, exception)
# Due to GitHub ratelimits we need to sleep a bit
await asyncio.sleep(sleeper)