Skip to content

Commit ab6353a

Browse files
committed
fix code
1 parent df39eb9 commit ab6353a

1 file changed

Lines changed: 14 additions & 12 deletions

File tree

‎databunkerpro/api.py‎

Lines changed: 14 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -822,7 +822,7 @@ def get_group(
822822
if isinstance(group_id, int) or str(group_id).isdigit():
823823
data["groupid"] = group_id
824824
else:
825-
data["groupname"] = group_id
825+
data["groupname"] = str(group_id)
826826
return self._make_request("GroupGet", data, request_metadata)
827827

828828
def list_all_groups(
@@ -855,7 +855,7 @@ def update_group(
855855
if isinstance(group_id, int) or str(group_id).isdigit():
856856
data["groupid"] = group_id
857857
else:
858-
data["groupname"] = group_id
858+
data["groupname"] = str(group_id)
859859
return self._make_request("GroupUpdate", data, request_metadata)
860860

861861
def delete_group(
@@ -868,7 +868,7 @@ def delete_group(
868868
if isinstance(group_id, int) or str(group_id).isdigit():
869869
data["groupid"] = group_id
870870
else:
871-
data["groupname"] = group_id
871+
data["groupname"] = str(group_id)
872872
return self._make_request("GroupDelete", data, request_metadata)
873873

874874
def remove_user_from_group(
@@ -883,7 +883,7 @@ def remove_user_from_group(
883883
if isinstance(group_id, int) or str(group_id).isdigit():
884884
data["groupid"] = group_id
885885
else:
886-
data["groupname"] = group_id
886+
data["groupname"] = str(group_id)
887887
return self._make_request("GroupDeleteUser", data, request_metadata)
888888

889889
def add_user_to_group(
@@ -1049,7 +1049,7 @@ def update_role(
10491049
if isinstance(role_id, int) or str(role_id).isdigit():
10501050
data["roleid"] = role_id
10511051
else:
1052-
data["rolename"] = role_id
1052+
data["rolename"] = str(role_id)
10531053
return self._make_request("RoleUpdate", data, request_metadata)
10541054

10551055
def link_policy(
@@ -1063,11 +1063,11 @@ def link_policy(
10631063
if isinstance(role_id, int) or str(role_id).isdigit():
10641064
data["roleid"] = role_id
10651065
else:
1066-
data["rolename"] = role_id
1066+
data["rolename"] = str(role_id)
10671067
if isinstance(policy_id, int) or str(policy_id).isdigit():
10681068
data["policyid"] = policy_id
10691069
else:
1070-
data["policyname"] = policy_id
1070+
data["policyname"] = str(policy_id)
10711071
return self._make_request("RoleLinkPolicy", data, request_metadata)
10721072

10731073
# Policy Management
@@ -1095,7 +1095,7 @@ def update_policy(
10951095
if isinstance(policy_id, int) or str(policy_id).isdigit():
10961096
data["policyid"] = policy_id
10971097
else:
1098-
data["policyname"] = policy_id
1098+
data["policyname"] = str(policy_id)
10991099
return self._make_request("PolicyUpdate", data, request_metadata)
11001100

11011101
def get_policy(
@@ -1108,7 +1108,7 @@ def get_policy(
11081108
if isinstance(policy_id, int) or str(policy_id).isdigit():
11091109
data["policyid"] = policy_id
11101110
else:
1111-
data["policyname"] = policy_id
1111+
data["policyname"] = str(policy_id)
11121112
return self._make_request("PolicyGet", data, request_metadata)
11131113

11141114
def list_policies(
@@ -1218,11 +1218,11 @@ def bulk_delete_tokens(
12181218
# System Configuration
12191219
def get_ui_conf(self) -> Dict[str, Any]:
12201220
"""Get UI configuration."""
1221-
return self._make_request("TenantGetUIConf", None, None)
1221+
return self._make_request("TenantGetUIConf")
12221222

12231223
def get_tenant_conf(self) -> Dict[str, Any]:
12241224
"""Get tenant configuration."""
1225-
return self._make_request("TenantGetConf", None, None)
1225+
return self._make_request("TenantGetConf")
12261226

12271227
def get_user_html_report(
12281228
self,
@@ -1256,7 +1256,9 @@ def get_system_stats(
12561256
"""Gets system statistics."""
12571257
return self._make_request("SystemGetSystemStats", None, request_metadata)
12581258

1259-
def get_system_metrics(self) -> Dict[str, Any]:
1259+
def get_system_metrics(
1260+
self, request_metadata: Optional[Dict[str, Any]] = None
1261+
) -> Dict[str, Any]:
12601262
"""Get system metrics in Prometheus format."""
12611263
try:
12621264
response = requests.get(f"{self.base_url}/metrics")

0 commit comments

Comments
 (0)