@@ -52,29 +52,29 @@ def encrypt_data(self, fetch, data, role):
5252 if self .shared_key is None or type (self .shared_key ) != bytes :
5353 raise EvervaultError ("Retrieved key is invalid" )
5454
55- if type (data ) == bytes :
55+ if isinstance (data , bytes ) :
5656 return self .__encrypt_file (data , role )
57- elif type (data ) == bytearray :
57+ elif isinstance (data , bytearray ) :
5858 return self .__encrypt_file (bytes (data ), role )
59- elif type (data ) == dict or type (data ) == list or type (data ) == set :
59+ elif isinstance (data , dict ) or isinstance (data , list ) or isinstance (data , set ) :
6060 return self .__traverse_and_encrypt (data , role )
6161 elif self .__encryptable_data (data ):
6262 return self .__encrypt_string (data , role )
6363 else :
6464 raise EvervaultError (f"Cannot encrypt unsupported type { data } " )
6565
6666 def __traverse_and_encrypt (self , data , role ):
67- if type (data ) == list :
67+ if isinstance (data , list ) :
6868 encrypted_list = []
6969 for idx , item in enumerate (data ):
7070 if not self .__encryptable_data (item ):
7171 encrypted_list .insert (idx , self .__traverse_and_encrypt (item , role ))
7272 else :
7373 encrypted_list .insert (idx , self .__encrypt_string (item , role ))
7474 return encrypted_list
75- elif type (data ) == dict :
75+ elif isinstance (data , dict ) :
7676 return self .__encrypt_object (data , role )
77- elif type (data ) == set :
77+ elif isinstance (data , set ) :
7878 return self .__encrypt_set (data , role )
7979 elif self .__encryptable_data (data ):
8080 return self .__encrypt_string (data , role )
@@ -212,9 +212,9 @@ def __encrypt_file(self, data, role):
212212 )
213213
214214 def __coerce_type (self , data ):
215- if type (data ) == bool :
215+ if isinstance (data , bool ) :
216216 return "true" if data else "false"
217- elif type (data ) == int or type (data ) == float :
217+ elif isinstance (data , int ) or isinstance (data , float ) :
218218 return str (data )
219219 else :
220220 return data
@@ -254,10 +254,10 @@ def __base_64_remove_padding(self, data):
254254
255255 def __encryptable_data (self , data ):
256256 return data is not None and (
257- type (data ) == str
258- or type (data ) == bool
259- or type (data ) == int
260- or type (data ) == float
257+ isinstance (data , str )
258+ or isinstance (data , bool )
259+ or isinstance (data , int )
260+ or isinstance (data , float )
261261 )
262262
263263 def __fetch_cage_key (self , fetch ):
0 commit comments