change update_user used function from wazuh.security to wazuh.rbac.orm

build-docker-images/wazuh-manager/config/create_user.py

@@ -13,13 +13,12 @@ SPECIAL_CHARS = "@$!%*?&-_"
 
 
 try:
-    from wazuh.rbac.orm import check_database_integrity
+    from wazuh.rbac.orm import check_database_integrity, AuthenticationManager
     from wazuh.security import (
         create_user,
         get_users,
         get_roles,
         set_user_role,
-        update_user,
     )
 except ModuleNotFoundError as e:
     logging.error("No module 'wazuh' found.")
@@ -54,10 +53,9 @@ def disable_user(uid):
     # assure there must be at least one character from each group
     random_pass = random_pass + ''.join([random.choice(chars) for chars in [string.ascii_lowercase, string.digits, string.ascii_uppercase, SPECIAL_CHARS]])
     random_pass = ''.join(random.sample(random_pass,len(random_pass)))
-    update_user(
+    with AuthenticationManager() as auth:
-        user_id=[
+        auth.update_user(
-            str(uid),
+            user_id=uid,
-        ],
             password=random_pass,
         )
 
@@ -90,10 +88,9 @@ if __name__ == "__main__":
     else:
         # modify an existing user ("wazuh" or "wazuh-wui")
         uid = initial_users[username]
-        update_user(
+        with AuthenticationManager() as auth:
-            user_id=[
+            auth.update_user(
-                str(uid),
+                user_id=uid,
-            ],
                 password=password,
             )
     # disable unused default users