Skip to content

Commit be710a9

Browse files
committed
fix code
1 parent 7ff35b6 commit be710a9

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

databunkerpro/api.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import json
44
import re
55
from datetime import datetime
6-
from typing import Any, Dict, List, Optional, TypedDict, Union
6+
from typing import Any, Dict, List, Optional, TypedDict, Union, cast
77

88
import requests
99

@@ -452,7 +452,7 @@ def create_user_x_token(
452452
"""Creates an access token for a user."""
453453
data = {"mode": mode, "identity": identity}
454454
if options:
455-
data.update(options)
455+
data.update(cast(Dict[str, Any], options))
456456
return self._make_request("XTokenCreateForUser", data, request_metadata)
457457

458458
def create_role_x_token(
@@ -464,7 +464,7 @@ def create_role_x_token(
464464
"""Creates an access token for a role."""
465465
data: Dict[str, Any] = {}
466466
if options:
467-
data.update(options)
467+
data.update(cast(Dict[str, Any], options))
468468
if isinstance(role_ref, int) or str(role_ref).isdigit():
469469
data["roleid"] = int(role_ref)
470470
else:
@@ -1095,7 +1095,7 @@ def create_token(
10951095
"""Create a token for sensitive data."""
10961096
data = {"tokentype": token_type, "record": record}
10971097
if options:
1098-
data.update(options)
1098+
data.update(cast(Dict[str, Any], options))
10991099
return self._make_request("TokenCreate", data, request_metadata)
11001100

11011101
def create_tokens_bulk(
@@ -1107,7 +1107,7 @@ def create_tokens_bulk(
11071107
"""Create multiple tokens in bulk."""
11081108
data = {"records": records}
11091109
if options:
1110-
data.update(options)
1110+
data.update(cast(Dict[str, Any], options))
11111111
return self._make_request("TokenCreateBulk", data, request_metadata)
11121112

11131113
def get_token(
@@ -1469,7 +1469,7 @@ def upsert_session(
14691469
"""Create or update a session (upsert operation)."""
14701470
data = {"sessionuuid": session_uuid, "sessiondata": session_data}
14711471
if options:
1472-
data.update(options)
1472+
data.update(cast(Dict[str, Any], options))
14731473
return self._make_request("SessionUpsert", data, request_metadata)
14741474

14751475
def delete_session(

0 commit comments

Comments
 (0)