Add environment variable checks for OMADA_SITE_ID, OMADA_GROUP_ID, and OMADA_PROXY_URL inside function calls
All checks were successful
Build and Push Docker Images / Build and Push Docker Images (push) Successful in 3m15s

This commit is contained in:
i701 2025-05-30 14:23:41 +05:00
parent 5600a5e8e6
commit e0dfc28590
Signed by: i701
GPG Key ID: 54A0DA1E26D8E587

View File

@ -19,19 +19,6 @@ OMADA_PROXY_URL = env("OMADA_PROXY_URL", default="") # type: ignore
OMADA_SITE_ID = env("OMADA_SITE_ID", default="") # type: ignore
OMADA_GROUP_ID = env("OMADA_GROUP_ID", default="") # type: ignore
if not OMADA_SITE_ID:
raise ValueError(
"OMADA_SITE_ID is not set. Please set it in your environment variables."
)
if not OMADA_GROUP_ID:
raise ValueError(
"OMADA_GROUP_ID is not set. Please set it in your environment variables."
)
if not OMADA_PROXY_URL:
raise ValueError(
"OMADA_PROXY_URL is not set. Please set it in your environment variables."
)
@shared_task
def add(x, y):
@ -76,6 +63,10 @@ def get_existing_omada_devices():
Get existing Omada devices from the database.
:return: List of existing device names.
"""
if not OMADA_PROXY_URL:
raise ValueError(
"OMADA_PROXY_URL is not set. Please set it in your environment variables."
)
try:
response = requests.get(
f"{OMADA_PROXY_URL}/9fd0cffa3475a74ae4e4d37de0d12414/api/v2/sites/66dcddb804aa0d2978cf145f/setting/profiles/groups",
@ -102,6 +93,18 @@ def add_new_devices_to_omada(new_devices: list[dict]):
Add new devices to Omada.
:param new_devices: List of new device names to add.
"""
if not OMADA_SITE_ID:
raise ValueError(
"OMADA_SITE_ID is not set. Please set it in your environment variables."
)
if not OMADA_GROUP_ID:
raise ValueError(
"OMADA_GROUP_ID is not set. Please set it in your environment variables."
)
if not OMADA_PROXY_URL:
raise ValueError(
"OMADA_PROXY_URL is not set. Please set it in your environment variables."
)
try:
PAYLOAD = {
"name": "REGISTERED_DEVICES",