o DDiR@sXddlZejdkrddlmZmZnddlmZmZddlmZGdddeeZdS)N) )IterableSequence) CSTNodeT_coc@sTeZdZUdZeeed<deeddfddZde defdd Z de fd d Z dS) FlattenSentinela A :class:`FlattenSentinel` may be returned by a :meth:`CSTTransformer.on_leave` method when one wants to replace a node with multiple nodes. The replaced node must be contained in a `Sequence` attribute such as :attr:`~libcst.Module.body`. This is generally the case for :class:`~libcst.BaseStatement` and :class:`~libcst.BaseSmallStatement`. For example to insert a print before every return:: def leave_Return( self, original_node: cst.Return, updated_node: cst.Return ) -> Union[cst.Return, cst.RemovalSentinel, cst.FlattenSentinel[cst.BaseSmallStatement]]: log_stmt = cst.Expr(cst.parse_expression("print('returning')")) return cst.FlattenSentinel([log_stmt, updated_node]) Returning an empty :class:`FlattenSentinel` is equivalent to returning :attr:`cst.RemovalSentinel.REMOVE` and is subject to its requirements. nodesreturnNcCst||_dSN)tupler)selfrr Q/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/_flatten_sentinel.py__init__&szFlattenSentinel.__init__idxcCs |j|Sr )r)r rr r r __getitem__) zFlattenSentinel.__getitem__cCs t|jSr )lenr)r r r r__len__,rzFlattenSentinel.__len__) __name__ __module__ __qualname____doc__rr__annotations__rrintrrr r r rrs  r) sys version_infotypingrrcollections.abc libcst._typesrrr r r rs