16
16
# under the License.
17
17
from __future__ import annotations
18
18
19
- from collections .abc import Container
20
19
from functools import cache
21
20
from typing import TYPE_CHECKING , Annotated , Callable
22
21
@@ -111,13 +110,11 @@ def to_orm(self, select: Select) -> Select:
111
110
return select .where (DagModel .dag_id .in_ (self .value ))
112
111
113
112
114
- def permitted_dag_filter_factory (
115
- methods : Container [ResourceMethod ],
116
- ) -> Callable [[Request , BaseUser ], PermittedDagFilter ]:
113
+ def permitted_dag_filter_factory (method : ResourceMethod ) -> Callable [[Request , BaseUser ], PermittedDagFilter ]:
117
114
"""
118
115
Create a callable for Depends in FastAPI that returns a filter of the permitted dags for the user.
119
116
120
- :param methods : whether filter readable or writable.
117
+ :param method : whether filter readable or writable.
121
118
:return: The callable that can be used as Depends in FastAPI.
122
119
"""
123
120
@@ -126,14 +123,14 @@ def depends_permitted_dags_filter(
126
123
user : GetUserDep ,
127
124
) -> PermittedDagFilter :
128
125
auth_manager : BaseAuthManager = request .app .state .auth_manager
129
- permitted_dags : set [str ] = auth_manager .get_permitted_dag_ids (user = user , methods = methods )
126
+ permitted_dags : set [str ] = auth_manager .get_permitted_dag_ids (user = user , method = method )
130
127
return PermittedDagFilter (permitted_dags )
131
128
132
129
return depends_permitted_dags_filter
133
130
134
131
135
- EditableDagsFilterDep = Annotated [PermittedDagFilter , Depends (permitted_dag_filter_factory ([ "PUT" ] ))]
136
- ReadableDagsFilterDep = Annotated [PermittedDagFilter , Depends (permitted_dag_filter_factory ([ "GET" ] ))]
132
+ EditableDagsFilterDep = Annotated [PermittedDagFilter , Depends (permitted_dag_filter_factory ("PUT" ))]
133
+ ReadableDagsFilterDep = Annotated [PermittedDagFilter , Depends (permitted_dag_filter_factory ("GET" ))]
137
134
138
135
139
136
def requires_access_pool (method : ResourceMethod ) -> Callable [[Request , BaseUser ], None ]:
0 commit comments