mirrors/recipes
1
0
Fork 0
mirror of https://github.com/TandoorRecipes/recipes.git synced 2026-01-08 07:38:26 -05:00
Code Issues Packages Projects Releases Wiki Activity

update tests and fix small bug in connector_manager

Browse Source
This commit is contained in:
Mikhail Epifanov
2024-01-13 00:24:58 +01:00
parent 1a37961ceb
commit 50eb232fff
2 changed files with 19 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
cookbook/connectors/connector_manager.py
View File

@@ -13,12 +13,13 @@ from django_scopes import scope
from cookbook.connectors.connector import Connector
from cookbook.connectors.homeassistant import HomeAssistant
from cookbook.models import ShoppingListEntry, Recipe, MealPlan, Space
from cookbook.models import ShoppingListEntry, Recipe, MealPlan, Space, HomeAssistantConfig, ConnectorConfig
multiprocessing.set_start_method('fork') # https://code.djangoproject.com/ticket/31169
QUEUE_MAX_SIZE = 25
REGISTERED_CLASSES: UnionType = ShoppingListEntry | Recipe | MealPlan | Connector
REGISTERED_CLASSES: UnionType = ShoppingListEntry | Recipe | MealPlan
CONNECTOR_UPDATE_CLASSES: UnionType = HomeAssistantConfig | ConnectorConfig
class ActionType(Enum):
@@ -35,7 +36,7 @@ class Work:
class ConnectorManager:
_queue: Queue
_listening_to_classes = REGISTERED_CLASSES
_listening_to_classes = REGISTERED_CLASSES | CONNECTOR_UPDATE_CLASSES
def __init__(self):
self._queue = multiprocessing.Queue(maxsize=QUEUE_MAX_SIZE)
@@ -81,10 +82,10 @@ class ConnectorManager:
break
# If a Connector was changed/updated, refresh connector from the database for said space
refresh_connector_cache = isinstance(item.instance, Connector)
refresh_connector_cache = isinstance(item.instance, CONNECTOR_UPDATE_CLASSES)
space: Space = item.instance.space
connectors: Optional[List[Connector]] = _connectors.get(space.name, None)
connectors: Optional[List[Connector]] = _connectors.get(space.name)
if connectors is None or refresh_connector_cache:
with scope(space=space):