From e9fbd56e4eb167f591483090b0329300ca2fa881 Mon Sep 17 00:00:00 2001
From: Simon Prickett <simon@redislabs.com>
Date: Thu, 6 Jan 2022 15:51:56 +0000
Subject: [PATCH] Changed the way that modules are detected to work with Redis
 Enterprise and OSS.

---
 aredis_om/checks.py | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)

diff --git a/aredis_om/checks.py b/aredis_om/checks.py
index 00c5084..882c69d 100644
--- a/aredis_om/checks.py
+++ b/aredis_om/checks.py
@@ -5,18 +5,19 @@ from aredis_om.connections import get_redis_connection
 
 
 @lru_cache(maxsize=None)
-async def get_modules(conn) -> List[str]:
-    modules = await conn.execute_command("module", "list")
-    return [m[1] for m in modules]
-
+async def check_for_command(conn, cmd):
+    try:
+        cmd_info = await conn.execute_command("command", "info", cmd)
+        return True
+    except TypeError:
+        return False
 
 @lru_cache(maxsize=None)
 async def has_redis_json(conn=None):
     if conn is None:
         conn = get_redis_connection()
-    names = await get_modules(conn)
-    return b"ReJSON" in names or "ReJSON" in names
-
+    command_exists = await check_for_command(conn, "json.set")
+    return command_exists
 
 @lru_cache(maxsize=None)
 async def has_redisearch(conn=None):
@@ -24,5 +25,5 @@ async def has_redisearch(conn=None):
         conn = get_redis_connection()
     if has_redis_json(conn):
         return True
-    names = await get_modules(conn)
-    return b"search" in names or "search" in names
+    command_exists = await check_for_command(conn, "ft.search")
+    return command_exists