Skip to content

Commit 5b7977a

Browse files
authored
Optimize retrieval of conn_id and dag_id fields in AirflowSecurityManagerV2 (#43420)
1 parent 132a755 commit 5b7977a

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

airflow/www/security_manager.py

+6-6
Original file line numberDiff line numberDiff line change
@@ -183,18 +183,18 @@ def _auth_manager_is_authorized_map(
183183
def get_connection_id(resource_pk):
184184
if not resource_pk:
185185
return None
186-
connection = session.scalar(select(Connection).where(Connection.id == resource_pk).limit(1))
187-
if not connection:
186+
conn_id = session.scalar(select(Connection.conn_id).where(Connection.id == resource_pk).limit(1))
187+
if not conn_id:
188188
raise AirflowException("Connection not found")
189-
return connection.conn_id
189+
return conn_id
190190

191191
def get_dag_id_from_dagrun_id(resource_pk):
192192
if not resource_pk:
193193
return None
194-
dagrun = session.scalar(select(DagRun).where(DagRun.id == resource_pk).limit(1))
195-
if not dagrun:
194+
dag_id = session.scalar(select(DagRun.dag_id).where(DagRun.id == resource_pk).limit(1))
195+
if not dag_id:
196196
raise AirflowException("DagRun not found")
197-
return dagrun.dag_id
197+
return dag_id
198198

199199
def get_dag_id_from_task_instance(resource_pk):
200200
if not resource_pk:

0 commit comments

Comments
 (0)