File tree Expand file tree Collapse file tree 2 files changed +30
-0
lines changed
Expand file tree Collapse file tree 2 files changed +30
-0
lines changed Original file line number Diff line number Diff line change 1+ import pandas as pd
2+ from sqlalchemy .engine import Engine
3+
4+ def execute_query (engine : Engine , query : str , params : dict = None ) -> pd .DataFrame :
5+ """
6+ Executa uma consulta SQL e retorna os resultados como DataFrame.
7+ :param engine: Objeto de conexão do SQLAlchemy
8+ :param query: Consulta SQL
9+ :param params: Parâmetros opcionais para a consulta
10+ :return: DataFrame com os resultados
11+ """
12+ with engine .connect () as conn :
13+ return pd .read_sql_query (query , conn , params = params )
Original file line number Diff line number Diff line change 1+ import os
2+
3+ def load_query (query_name : str ) -> str :
4+ """
5+ Carrega o conteúdo de um arquivo SQL da pasta de queries.
6+ :param query_name: Nome do arquivo SQL sem a extensão
7+ :return: Conteúdo da query como string
8+ """
9+ base_dir = os .path .dirname (os .path .abspath (__file__ ))
10+ queries_dir = os .path .join (base_dir , '../queries' )
11+ query_path = os .path .join (queries_dir , f'{ query_name } .sql' )
12+
13+ if not os .path .exists (query_path ):
14+ raise FileNotFoundError (f'Query "{ query_name } .sql" não encontrada!' )
15+
16+ with open (query_path , 'r' , encoding = 'utf-8' ) as file :
17+ return file .read ()
You can’t perform that action at this time.
0 commit comments