Add dms handler

[pti/o2.git] / o2ims / service / handlers.py

diff --git a/o2ims/service/handlers.py b/o2ims/service/handlers.py
index 505549c..830e1ff 100644 (file)
--- a/o2ims/service/handlers.py
+++ b/o2ims/service/handlers.py
@@ -16,6 +16,7 @@
 from __future__ import annotations
 # from dataclasses import asdict
 from typing import List, Dict, Callable, Type
+from o2ims.service.auditor import dms_handler
 # TYPE_CHECKING
 from o2ims.domain import commands, events
 from o2ims.service.auditor import ocloud_handler
@@ -34,4 +35,5 @@ EVENT_HANDLERS = {
 
 COMMAND_HANDLERS = {
     commands.UpdateOCloud: ocloud_handler.update_ocloud,
+    commands.UpdateDms: dms_handler.update_dms
 }  # type: Dict[Type[commands.Command], Callable]