diff --git a/Scripts/Libs/CleaningPipeline/sql_endpoint.py b/Scripts/Libs/CleaningPipeline/sql_endpoint.py index 66ba1ea..31f9ae4 100644 --- a/Scripts/Libs/CleaningPipeline/sql_endpoint.py +++ b/Scripts/Libs/CleaningPipeline/sql_endpoint.py @@ -133,6 +133,11 @@ class SqlEndpoint(): GROUP BY RelationshipURI; """ return pd.read_sql_query(QUERY, self.sql_engine) + + def get_dataframe_from_query(self, query: str, params=None): + if params is None: + return pd.read_sql_query(query, self.sql_engine) + return pd.read_sql_query(query, self.sql_engine, params=params)