Skip to content

Commit cc41ab4

Browse files
committed
mypy fixes
1 parent 6a0713c commit cc41ab4

File tree

5 files changed

+319
-211
lines changed

5 files changed

+319
-211
lines changed

packages/service-library/src/servicelib/rabbitmq/rpc_interfaces/webserver/v1/api_keys.py

Lines changed: 20 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
"""API Keys RPC API subclient."""
22

3+
from typing import cast
4+
35
from models_library.basic_types import IDStr
46
from models_library.products import ProductName
57
from models_library.rpc.webserver.auth.api_keys import ApiKeyCreate, ApiKeyGet
@@ -19,12 +21,15 @@ async def create_api_key(
1921
api_key: ApiKeyCreate,
2022
) -> ApiKeyGet:
2123
"""Create an API key."""
22-
return await self._request(
23-
"create_api_key",
24-
product_name=product_name,
25-
user_id=user_id,
26-
display_name=api_key.display_name,
27-
expiration=api_key.expiration,
24+
return cast(
25+
ApiKeyGet,
26+
await self._request(
27+
"create_api_key",
28+
product_name=product_name,
29+
user_id=user_id,
30+
display_name=api_key.display_name,
31+
expiration=api_key.expiration,
32+
),
2833
)
2934

3035
async def get_api_key(
@@ -35,11 +40,14 @@ async def get_api_key(
3540
api_key_id: IDStr,
3641
) -> ApiKeyGet:
3742
"""Get an API key by ID."""
38-
return await self._request(
39-
"get_api_key",
40-
product_name=product_name,
41-
user_id=user_id,
42-
api_key_id=api_key_id,
43+
return cast(
44+
ApiKeyGet,
45+
await self._request(
46+
"get_api_key",
47+
product_name=product_name,
48+
user_id=user_id,
49+
api_key_id=api_key_id,
50+
),
4351
)
4452

4553
async def delete_api_key_by_key(
@@ -50,7 +58,7 @@ async def delete_api_key_by_key(
5058
api_key: str,
5159
) -> None:
5260
"""Delete an API key by key value."""
53-
return await self._request(
61+
await self._request(
5462
"delete_api_key_by_key",
5563
product_name=product_name,
5664
user_id=user_id,

0 commit comments

Comments
 (0)