Skip to content

Commit 43da538

Browse files
Altering the execute_query method based on Dave's feedback
1 parent ec3bdc7 commit 43da538

File tree

1 file changed

+9
-24
lines changed

1 file changed

+9
-24
lines changed

utils/oracle/oracle.py

Lines changed: 9 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -134,30 +134,15 @@ def execute_query(
134134
"""
135135
conn = self.connect_to_db()
136136
engine = create_engine("oracle+oracledb://", creator=lambda: conn)
137-
if parameters == None:
138-
try:
139-
logging.info("Attempting to execute query")
140-
df = pd.read_sql(query, engine)
141-
logging.info("Query execution successful!")
142-
except Exception as executionError:
143-
logging.error(
144-
f"Failed to execute query with execution error {executionError}"
145-
)
146-
finally:
147-
if conn is not None:
148-
self.disconnect_from_db(conn)
149-
else:
150-
try:
151-
logging.info("Attempting to execute query")
152-
df = pd.read_sql(query, engine, params = parameters)
153-
logging.info("Query execution successful!")
154-
except Exception as executionError:
155-
logging.error(
156-
f"Failed to execute query with execution error {executionError}"
157-
)
158-
finally:
159-
if conn is not None:
160-
self.disconnect_from_db(conn)
137+
try:
138+
df = pd.read_sql(query, engine) if parameters == None else pd.read_sql(query, engine, params = parameters)
139+
except Exception as executionError:
140+
logging.error(
141+
f"Failed to execute query with execution error {executionError}"
142+
)
143+
finally:
144+
if conn is not None:
145+
self.disconnect_from_db(conn)
161146
return df
162147

163148
def execute_stored_procedure(

0 commit comments

Comments
 (0)