fix select_keys

Signed-off-by: Praneeth Bedapudi <praneeth@bpraneeth.com>

liteindex/defined_index.py

@@ -345,6 +345,8 @@ def search(
         if not select_keys:
             select_keys = list(self.original_key_to_key_hash)
 
+        select_keys_hashes = [self.original_key_to_key_hash[k] for k in select_keys]
+
         sql_query, sql_params = search_query(
             table_name=self.name,
             query={self.original_key_to_key_hash[k]: v for k, v in query.items()},
@@ -354,10 +356,7 @@ def search(
             n=n,
             page=page_no,
             page_size=n if page_no else None,
-            select_columns=(
-                ["id", "updated_at"]
-                + [f'"{self.original_key_to_key_hash[k]}"' for k in select_keys]
-            ),
+            select_columns=(["id", "updated_at"] + select_keys_hashes),
         )
 
         results = {}
@@ -366,7 +365,7 @@ def search(
             _id, updated_at = result[:2]
             record = {
                 self.key_hash_to_original_key[h]: val
-                for h, val in zip(self.original_key_to_key_hash.values(), result[2:])
+                for h, val in zip(select_keys_hashes, result[2:])
             }
             for k, v in record.items():
                 if v is None:

setup.py

@@ -18,7 +18,7 @@
 EMAIL = "praneeth@bpraneeth.com"
 AUTHOR = "BEDAPUDI PRANEETH"
 REQUIRES_PYTHON = ">=3.6.0"
-VERSION = "0.0.2.dev11"
+VERSION = "0.0.2.dev12"
 
 # What packages are required for this module to be executed?
 REQUIRED = [