get distinct_count

Signed-off-by: Praneeth Bedapudi <[email protected]>

liteindex/defined_index.py

@@ -31,6 +31,7 @@
 from .query_parser import (
     search_query,
     distinct_query,
+    distinct_count_query,
     count_query,
     delete_query,
     group_by_query,
@@ -571,6 +572,19 @@ def distinct(self, key, query={}):
             _[0] for _ in self.__connection.execute(sql_query, sql_params).fetchall()
         }
 
+    def distinct_count(self, key, query={}):
+        sql_query, sql_params = distinct_count_query(
+            table_name=self.name,
+            column=key,
+            query={k: v for k, v in query.items()},
+            schema=self.schema,
+        )
+
+        return {
+            _[0]: _[1]
+            for _ in self.__connection.execute(sql_query, sql_params).fetchall()
+        }
+
     def group(self, keys, query={}):
         if isinstance(keys, str):
             keys = [keys]

liteindex/query_parser.py

@@ -191,6 +191,21 @@ def distinct_query(table_name, column, query, schema):
     return query_str, params
 
 
+def distinct_count_query(table_name, column, query, schema):
+    # Prepare the query
+    where_conditions, params = parse_query(query, schema)
+
+    if schema[column] == "json":
+        query_str = f"SELECT JSON_EXTRACT({column}, '$[*]'), COUNT(*) FROM {table_name}"
+    else:
+        query_str = f"SELECT {column}, COUNT(*) FROM {table_name}"
+    if where_conditions:
+        query_str += f" WHERE {' AND '.join(where_conditions)}"
+    query_str += f" GROUP BY {column}"
+
+    return query_str, params
+
+
 def group_by_query(table_name, columns, query, schema):
     # Prepare the query
     where_conditions, params = parse_query(query, schema)

setup.py

@@ -18,7 +18,7 @@
 EMAIL = "[email protected]"
 AUTHOR = "BEDAPUDI PRANEETH"
 REQUIRES_PYTHON = ">=3.6.0"
-VERSION = "0.0.2.dev48"
+VERSION = "0.0.2.dev49"
 
 # What packages are required for this module to be executed?
 REQUIRED = []