فهرست منبع

refactor: expand the _get_manager() with User entity from taipy.auth

trgiangdo 7 ماه پیش
والد
کامیت
ef67923fd9
2فایلهای تغییر یافته به همراه13 افزوده شده و 2 حذف شده
  1. 12 2
      taipy/core/_entity/_reload.py
  2. 1 0
      taipy/core/common/_check_dependencies.py

+ 12 - 2
taipy/core/_entity/_reload.py

@@ -12,6 +12,8 @@
 import functools
 
 from .._manager._manager import _Manager
+from ..common._check_dependencies import EnterpriseEditionUtils
+from ..common._utils import _load_fct
 from ..notification import EventOperation, Notifier, _make_event
 
 
@@ -96,7 +98,7 @@ def _get_manager(manager: str) -> _Manager:
     from ..submission._submission_manager_factory import _SubmissionManagerFactory
     from ..task._task_manager_factory import _TaskManagerFactory
 
-    return {
+    managers = {
         "scenario": _ScenarioManagerFactory._build_manager(),
         "sequence": _SequenceManagerFactory._build_manager(),
         "data": _DataManagerFactory._build_manager(),
@@ -104,4 +106,12 @@ def _get_manager(manager: str) -> _Manager:
         "job": _JobManagerFactory._build_manager(),
         "task": _TaskManagerFactory._build_manager(),
         "submission": _SubmissionManagerFactory._build_manager(),
-    }[manager]  # type: ignore
+    }
+
+    if EnterpriseEditionUtils._using_enterprise():
+        _build_user_manager = _load_fct(
+            EnterpriseEditionUtils._TAIPY_AUTH_MODULE + ".user._user_manager_factory", "_UserManagerFactory"
+        )._build_manager
+        managers["user"] = _build_user_manager()
+
+    return managers[manager]

+ 1 - 0
taipy/core/common/_check_dependencies.py

@@ -32,6 +32,7 @@ def _check_dependency_is_installed(module_name: str, package_name: str) -> None:
 
 
 class EnterpriseEditionUtils:
+    _TAIPY_AUTH_MODULE = "taipy.auth"
     _TAIPY_ENTERPRISE_MODULE = "taipy.enterprise"
     _TAIPY_ENTERPRISE_CORE_MODULE = _TAIPY_ENTERPRISE_MODULE + ".core"