Merge pull request #13 from pepkit/dev

v0.2.0

docs/changelog.md

@@ -2,6 +2,12 @@
 
 This project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html) and [Keep a Changelog](https://keepachangelog.com/en/1.0.0/) format.
 
+## [0.2.0] - 2021-10-25
+
+### Added
+
+- optional parameter for specify returning columns with `select_txt`
+
 ## [0.1.0] - 2021-06-24
 
 **This update introduces some backwards-incompatible changes due to database interface redesign**

pipestat/_version.py

@@ -1 +1 @@
-__version__ = "0.1.0"
+__version__ = "0.2.0"

pipestat/pipestat.py

@@ -1206,6 +1206,7 @@ def _retrieve_db(
 
     def select_txt(
         self,
+        columns: Optional[List[str]] = None,
         filter_templ: Optional[str] = "",
         filter_params: Optional[Dict[str, Any]] = {},
         table_name: Optional[str] = None,
@@ -1232,12 +1233,16 @@ def select_txt(
                 f"The {self.__class__.__name__} object is not backed by a database. "
                 f"This operation is not supported for file backend."
             )
+        ORM = self.get_orm(table_name or self.namespace)
         with self.session as s:
-            q = (
-                s.query(self.get_orm(table_name or self.namespace))
-                .filter(text(filter_templ))
-                .params(**filter_params)
-            )
+            if columns is not None:
+                q = (
+                    s.query(*[getattr(ORM, column) for column in columns])
+                    .filter(text(filter_templ))
+                    .params(**filter_params)
+                )
+            else:
+                q = s.query(ORM).filter(text(filter_templ)).params(**filter_params)
             if isinstance(offset, int):
                 q = q.offset(offset)
             if isinstance(limit, int):