Db: allow to specify a list as a WHERE clause value and they will be combined as a IN clause

mylib/db.py

@@ -192,6 +192,17 @@ class DB:
             params[param] = value
         return params
 
+    @staticmethod
+    def _get_unique_param_name(field, params):
+        """Return a unique parameter name based on specified field name"""
+        param = field
+        if field in params:
+            idx = 1
+            while param in params:
+                param = f"{field}_{idx}"
+                idx += 1
+        return param
+
     @classmethod
     def _format_where_clauses(cls, where_clauses, params=None, where_op=None):
         """
@@ -237,13 +248,19 @@ class DB:
         if isinstance(where_clauses, dict):
             sql_where_clauses = []
             for field, value in where_clauses.items():
-                param = field
+                if isinstance(value, list):
-                if field in params:
+                    param_names = []
-                    idx = 1
+                    for idx, v in enumerate(value):
-                    while param in params:
+                        param = cls._get_unique_param_name(f"{field}_{idx}", params)
-                        param = f"{field}_{idx}"
+                        cls._combine_params(params, **{param: v})
-                        idx += 1
+                        param_names.append(param)
-                cls._combine_params(params, {param: value})
+                    sql_where_clauses.append(
+                        f"{cls._quote_field_name(field)} IN "
+                        f"({', '.join([cls.format_param(param) for param in param_names])})"
+                    )
+                else:
+                    param = cls._get_unique_param_name(field, params)
+                    cls._combine_params(params, **{param: value})
                     sql_where_clauses.append(
                         f"{cls._quote_field_name(field)} = {cls.format_param(param)}"
                     )