@@ -36,26 +36,33 @@ class AssetLibrary:
3636
3737 def __init__ (self ):
3838 self .count = 0
39- self .__local_headers = {}
39+ self .__stack_instance = None
40+ self .__http_request = None
41+ self .__stack_headers = {}
4042 self .__query_params = {}
4143
44+ def instance (self , stack_instance ):
45+ self .__stack_instance = stack_instance
46+ self .__stack_headers .update (self .__stack_instance .headers )
47+ self .__http_request = self .__stack_instance .http_request
48+
4249 def set_header (self , key : str , value ):
4350 if key is not None and value is not None :
44- self .__local_headers [key ] = value
51+ self .__stack_headers [key ] = value
4552 return self
4653
4754 def headers (self , headers : dict ):
4855 if headers is not None and len (headers ) > 0 and isinstance (headers , dict ):
49- self .__local_headers = headers
50- if 'environment' in self .__local_headers :
51- env_value = self .__local_headers ['environment' ]
56+ self .__stack_headers = headers
57+ if 'environment' in self .__stack_headers :
58+ env_value = self .__stack_headers ['environment' ]
5259 self .__query_params ["environment" ] = env_value
5360 return self
5461
5562 def remove_header (self , key ):
5663 if key is not None :
57- if key in self .__local_headers :
58- self .__local_headers .pop (key )
64+ if key in self .__stack_headers :
65+ self .__stack_headers .pop (key )
5966 return self
6067
6168 def include_count (self ):
@@ -79,7 +86,7 @@ def get_count(self) -> int:
7986 # self.__post_params['desc'] = key
8087 # return self.__post_params
8188
82- def fetch_all (self ) -> tuple :
89+ def fetch_all (self ):
8390
8491 import requests
8592 from urllib import parse
@@ -90,11 +97,11 @@ def fetch_all(self) -> tuple:
9097
9198 error = None
9299 asset_url = Config ().endpoint ('assets' )
93- self .__local_headers .update (self .header_agents ())
100+ self .__stack_headers .update (self .header_agents ())
94101 payload = parse .urlencode (query = self .__query_params , encoding = 'UTF-8' )
95102
96103 try :
97- response : Response = requests .get (asset_url , params = payload , headers = self .__local_headers )
104+ response : Response = requests .get (asset_url , params = payload , headers = self .__stack_headers )
98105 list_asset : list [Asset ] = []
99106
100107 if response .ok :
@@ -105,16 +112,15 @@ def fetch_all(self) -> tuple:
105112 asset_instance = Asset ()
106113 asset_resp : Asset = asset_instance .configure (response = asset )
107114 list_asset .append (asset_resp )
108- else :
109115
116+ return list_asset
117+ else :
110118 error_dict = response .json ()
111- Error ().error (error_dict )
112-
113- return list_asset , error
119+ return Error ().error (error_dict )
114120
115121 except requests .exceptions .RequestException as e :
116122 raise ConnectionError (e .response )
117- pass
123+
118124
119125 @classmethod
120126 def header_agents (cls ) -> dict :
0 commit comments