jeans/supervisor
jeans/supervisor
1
0
Fork 0
mirror of https://github.com/home-assistant/supervisor.git synced 2025-07-22 08:36:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle group options for concurrency and throttle separately

Browse Source
This commit is contained in:
Stefan Agner 2025-07-10 15:08:21 +02:00
parent d95048ed60
commit 24a65dccd5
No known key found for this signature in database
GPG Key ID: AE01353D1E44747D
1 changed files with 17 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
supervisor/jobs/decorator.py
View File

@ -239,16 +239,23 @@ class Job(CoreSysAttributes):
job_group = cast(JobGroup, obj)
# Check for group-based parameters
group_based_params = [
self.concurrency
in (JobConcurrency.GROUP_REJECT, JobConcurrency.GROUP_QUEUE),
self.throttle in (JobThrottle.GROUP_THROTTLE, JobThrottle.GROUP_RATE_LIMIT),
]
if not job_group and any(group_based_params):
raise RuntimeError(
f"Job on {self.name} need to be a JobGroup to use group based limits!"
) from None
if not job_group:
if self.concurrency in (
JobConcurrency.GROUP_REJECT,
JobConcurrency.GROUP_QUEUE,
):
raise RuntimeError(
f"Job {self.name} uses group concurrency ({self.concurrency}) but is not on a JobGroup! "
f"The class must inherit from JobGroup to use GROUP_REJECT or GROUP_QUEUE."
) from None
if self.throttle in (
JobThrottle.GROUP_THROTTLE,
JobThrottle.GROUP_RATE_LIMIT,
):
raise RuntimeError(
f"Job {self.name} uses group throttling ({self.throttle}) but is not on a JobGroup! "
f"The class must inherit from JobGroup to use GROUP_THROTTLE or GROUP_RATE_LIMIT."
) from None
return job_group