33
44from .schema import load_schema
55from .treeutil import get_children , is_container
6- from .util import NotSet
76
87
98def _filter_tree (info , filters ):
@@ -139,7 +138,7 @@ def _get_schema_key(schema, key):
139138 return None
140139
141140
142- def create_tree (key , node , identifier = "root" , filters = None , refresh_extension_manager = NotSet , extension_manager = None ):
141+ def create_tree (key , node , identifier = "root" , filters = None , extension_manager = None ):
143142 """
144143 Create a `NodeSchemaInfo` tree which can be filtered from a base node.
145144
@@ -153,19 +152,13 @@ def create_tree(key, node, identifier="root", filters=None, refresh_extension_ma
153152 A list of functions that take a node and identifier and return True if the node should be included in the tree.
154153 preserve_list : bool
155154 If True, then lists are preserved. Otherwise, they are turned into dicts.
156- refresh_extension_manager : bool
157- DEPRECATED
158- If `True`, refresh the extension manager before looking up the
159- key. This is useful if you want to make sure that the schema
160- data for a given key is up to date.
161155 """
162156 filters = [] if filters is None else filters
163157
164158 schema_info = NodeSchemaInfo .from_root_node (
165159 key ,
166160 identifier ,
167161 node ,
168- refresh_extension_manager = refresh_extension_manager ,
169162 extension_manager = extension_manager ,
170163 )
171164
@@ -182,7 +175,6 @@ def collect_schema_info(
182175 identifier = "root" ,
183176 filters = None ,
184177 preserve_list = True ,
185- refresh_extension_manager = NotSet ,
186178 extension_manager = None ,
187179):
188180 """
@@ -201,19 +193,13 @@ def collect_schema_info(
201193 A list of functions that take a node and identifier and return True if the node should be included in the tree.
202194 preserve_list : bool
203195 If True, then lists are preserved. Otherwise, they are turned into dicts.
204- refresh_extension_manager : bool
205- DEPRECATED
206- If `True`, refresh the extension manager before looking up the
207- key. This is useful if you want to make sure that the schema
208- data for a given key is up to date.
209196 """
210197
211198 schema_info = create_tree (
212199 key ,
213200 node ,
214201 identifier = identifier ,
215202 filters = [] if filters is None else filters ,
216- refresh_extension_manager = refresh_extension_manager ,
217203 extension_manager = extension_manager ,
218204 )
219205
@@ -229,18 +215,10 @@ def collect_schema_info(
229215 return info
230216
231217
232- def _get_extension_manager (refresh_extension_manager ):
233- from ._asdf import AsdfFile , get_config
234- from .extension import ExtensionManager
235-
236- if refresh_extension_manager is NotSet :
237- refresh_extension_manager = False
218+ def _get_extension_manager ():
219+ from ._asdf import AsdfFile
238220
239221 af = AsdfFile ()
240- if refresh_extension_manager :
241- config = get_config ()
242- af ._extension_manager = ExtensionManager (config .extensions )
243-
244222 return af .extension_manager
245223
246224
@@ -364,15 +342,13 @@ def set_schema_from_node(self, node, extension_manager):
364342 self .schema = schema
365343
366344 @classmethod
367- def from_root_node (
368- cls , key , root_identifier , root_node , schema = None , refresh_extension_manager = NotSet , extension_manager = None
369- ):
345+ def from_root_node (cls , key , root_identifier , root_node , schema = None , extension_manager = None ):
370346 """
371347 Build a NodeSchemaInfo tree from the given ASDF root node.
372348 Intentionally processes the tree in breadth-first order so that recursively
373349 referenced nodes are displayed at their shallowest reference point.
374350 """
375- extension_manager = extension_manager or _get_extension_manager (refresh_extension_manager )
351+ extension_manager = extension_manager or _get_extension_manager ()
376352
377353 current_nodes = [(None , root_identifier , root_node )]
378354 seen = set ()
0 commit comments