detquantlib 5.0.1__tar.gz → 5.1.0__tar.gz

{detquantlib-5.0.1 → detquantlib-5.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: detquantlib
-Version: 5.0.1
+Version: 5.1.0
 Summary: An internal library containing functions and classes that can be used across Quant models.
 License-File: LICENSE.txt
 Author: Dynamic Energy Trading

{detquantlib-5.0.1 → detquantlib-5.1.0}/detquantlib/data/databases/detdatabase.py

@@ -93,12 +93,15 @@ class DetDatabase:
         """Disconnects the SQLAlchemy engine and releases all underlying resources."""
         self.engine.dispose()
 
-    def query_db(self, query: str) -> pd.DataFrame:
+    def query_db(self, query: str, params: dict = None) -> pd.DataFrame:
         """
         Short utility method to make an SQL query to the database.
 
         Args:
             query: SQL query
+            params: Dict of query parameters. Example:
+                >> query = text('''SELECT * FROM MyTable WHERE start_date > :start_date''')
+                >> df = obj.query_db(query, params={"start_date": start_date})
 
         Returns:
             Dataframe containing the queried data
@@ -107,7 +110,7 @@ class DetDatabase:
             Exception: Raises an error if the SQL query fails.
         """
         try:
-            df = pd.read_sql_query(query, con=self.engine)
+            df = pd.read_sql_query(query, params=params, con=self.engine)
         except Exception:
             # If query fails, close connection before raising the error
             self.terminate_engine()

{detquantlib-5.0.1 → detquantlib-5.1.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "detquantlib"
-version = "5.0.1"
+version = "5.1.0"
 description = "An internal library containing functions and classes that can be used across Quant models."
 authors = ["Dynamic Energy Trading"]
 readme = "README.md"