1414
1515import json
1616from typing import Dict , List , Optional , Tuple
17- from . base import BaseClient
17+
1818from . import util
19+ from .base import BaseClient , Response
1920
2021
2122class Record :
2223 def __init__ (
2324 self ,
24- id : int ,
25- owner : str ,
26- name : str ,
27- created_time : str ,
28- organization : str ,
29- client_ip : str ,
30- user : str ,
31- method : str ,
32- request_uri : str ,
33- action : str ,
34- language : str ,
35- object : str ,
36- response : str ,
37- provider : str ,
38- block : str ,
39- is_triggered : bool ,
25+ createdTime : Optional [ str ] = None ,
26+ organization : Optional [ str ] = None ,
27+ clientIp : Optional [ str ] = None ,
28+ user : Optional [ str ] = None ,
29+ method : Optional [ str ] = None ,
30+ requestUri : Optional [ str ] = None ,
31+ action : Optional [ str ] = None ,
32+ language : Optional [ str ] = None ,
33+ object : Optional [ str ] = None ,
34+ response : Optional [ str ] = None ,
35+ provider : Optional [ str ] = None ,
36+ block : Optional [ str ] = None ,
37+ isTriggered : Optional [ bool ] = None ,
38+ id : Optional [ int ] = None ,
39+ owner : Optional [ str ] = None ,
40+ name : Optional [ str ] = None ,
4041 ):
4142 self .id = id
4243 self .owner = owner
4344 self .name = name
44- self .created_time = created_time
45+ self .createdTime = createdTime
4546 self .organization = organization
46- self .client_ip = client_ip
47+ self .clientIp = clientIp
4748 self .user = user
4849 self .method = method
49- self .request_uri = request_uri
50+ self .requestUri = requestUri
5051 self .action = action
5152 self .language = language
5253 self .object = object
5354 self .response = response
5455 self .provider = provider
5556 self .block = block
56- self .is_triggered = is_triggered
57+ self .isTriggered = isTriggered
5758
5859 def to_dict (self ) -> Dict :
5960 return self .__dict__
@@ -64,30 +65,30 @@ def from_dict(cls, data: Dict) -> "Record":
6465
6566
6667class _RecordSDK :
67- def __init__ (self , base_client : BaseClient , organization_name : str ):
68- self .base_client = base_client
69- self .organization_name = organization_name
68+ def __init__ (self , base_client : BaseClient , organizationName : str ):
69+ self .baseClient = base_client
70+ self .organizationName = organizationName
7071
7172 def get_records (self ) -> List [Record ]:
72- query_map = {"owner" : self .organization_name }
73- url = util .get_url (self .base_client .endpoint , "get-records" , query_map )
74- bytes = self .base_client .do_get_bytes (url )
73+ query_map = {"owner" : self .organizationName }
74+ url = util .get_url (self .baseClient .endpoint , "get-records" , query_map )
75+ bytes = self .baseClient .do_get_bytes (url )
7576 return [Record .from_dict (record ) for record in json .loads (bytes )]
7677
7778 def get_record (self , name : str ) -> Record :
78- query_map = {"id" : f"{ self .organization_name } /{ name } " }
79- url = util .get_url (self .base_client .endpoint , "get-record" , query_map )
80- bytes = self .base_client .do_get_bytes (url )
79+ query_map = {"id" : f"{ self .organizationName } /{ name } " }
80+ url = util .get_url (self .baseClient .endpoint , "get-record" , query_map )
81+ bytes = self .baseClient .do_get_bytes (url )
8182 return Record .from_dict (json .loads (bytes ))
8283
8384 def get_pagination_records (
84- self , p : int , page_size : int , query_map : Dict [str , str ]
85+ self , p : int , pageSize : int , query_map : Dict [str , str ]
8586 ) -> Tuple [List [Record ], int ]:
86- query_map ["owner" ] = self .organization_name
87+ query_map ["owner" ] = self .organizationName
8788 query_map ["p" ] = str (p )
88- query_map ["page_size " ] = str (page_size )
89- url = util .get_url (self .base_client .endpoint , "get-records" , query_map )
90- response = self .base_client .do_get_response (url )
89+ query_map ["pageSize " ] = str (pageSize )
90+ url = util .get_url (self .baseClient .endpoint , "get-records" , query_map )
91+ response = self .baseClient .do_get_response (url )
9192 return [Record .from_dict (record ) for record in response .data ], response .data2
9293
9394 def update_record (self , record : Record ) -> bool :
@@ -96,9 +97,9 @@ def update_record(self, record: Record) -> bool:
9697
9798 def add_record (self , record : Record ) -> bool :
9899 if not record .owner :
99- record .owner = self .organization_name
100+ record .owner = self .organizationName
100101 if not record .organization :
101- record .organization = self .organization_name
102+ record .organization = self .organizationName
102103 _ , affected = self .modify_record ("add-record" , record , None )
103104 return affected
104105
@@ -108,12 +109,12 @@ def delete_record(self, record: Record) -> bool:
108109
109110 def modify_record (
110111 self , action : str , record : Record , columns : Optional [List [str ]]
111- ) -> Tuple [Dict , bool ]:
112+ ) -> Tuple [Response , bool ]:
112113 query_map = {"id" : f"{ record .owner } /{ record .name } " }
113114 if columns :
114115 query_map ["columns" ] = "," .join (columns )
115116 if not record .owner :
116117 record .owner = "admin"
117118 post_bytes = json .dumps (record .to_dict ()).encode ("utf-8" )
118- resp = self .base_client .do_post (action , query_map , post_bytes , False , False )
119- return resp , resp [ " data" ] == "Affected"
119+ resp = self .baseClient .do_post (action , query_map , post_bytes , False , False )
120+ return resp , resp . data == "Affected"
0 commit comments