123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475 |
- # Copyright (c) 2009-2014 LOGILAB S.A. (Paris, FRANCE) <contact@logilab.fr>
- # Copyright (c) 2013-2020 Claudiu Popa <pcmanticore@gmail.com>
- # Copyright (c) 2013-2014 Google, Inc.
- # Copyright (c) 2014 Alexander Presnyakov <flagist0@gmail.com>
- # Copyright (c) 2014 Eevee (Alex Munroe) <amunroe@yelp.com>
- # Copyright (c) 2015-2016 Ceridwen <ceridwenv@gmail.com>
- # Copyright (c) 2016-2017 Derek Gustafson <degustaf@gmail.com>
- # Copyright (c) 2016 Jared Garst <jgarst@users.noreply.github.com>
- # Copyright (c) 2017 Hugo <hugovk@users.noreply.github.com>
- # Copyright (c) 2017 Łukasz Rogalski <rogalski.91@gmail.com>
- # Copyright (c) 2017 rr- <rr-@sakuya.pl>
- # Copyright (c) 2018-2019 Ville Skyttä <ville.skytta@iki.fi>
- # Copyright (c) 2018 Tomas Gavenciak <gavento@ucw.cz>
- # Copyright (c) 2018 Serhiy Storchaka <storchaka@gmail.com>
- # Copyright (c) 2018 Nick Drozd <nicholasdrozd@gmail.com>
- # Copyright (c) 2018 Bryce Guinta <bryce.paul.guinta@gmail.com>
- # Copyright (c) 2019-2021 Ashley Whetter <ashley@awhetter.co.uk>
- # Copyright (c) 2019 Hugo van Kemenade <hugovk@users.noreply.github.com>
- # Copyright (c) 2019 Zbigniew Jędrzejewski-Szmek <zbyszek@in.waw.pl>
- # Copyright (c) 2021 Marc Mueller <30130371+cdce8p@users.noreply.github.com>
- # Copyright (c) 2021 Daniël van Noord <13665637+DanielNoord@users.noreply.github.com>
- # Copyright (c) 2021 Pierre Sassoulas <pierre.sassoulas@gmail.com>
- # Copyright (c) 2021 Federico Bond <federicobond@gmail.com>
- # Copyright (c) 2021 hippo91 <guillaume.peillex@gmail.com>
- # Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html
- # For details: https://github.com/PyCQA/astroid/blob/main/LICENSE
- """this module contains utilities for rebuilding an _ast tree in
- order to get a single Astroid representation
- """
- import sys
- from typing import (
- TYPE_CHECKING,
- Callable,
- Dict,
- Generator,
- List,
- Optional,
- Tuple,
- Type,
- TypeVar,
- Union,
- cast,
- overload,
- )
- from astroid import nodes
- from astroid._ast import ParserModule, get_parser_module, parse_function_type_comment
- from astroid.const import PY38, PY38_PLUS, Context
- from astroid.manager import AstroidManager
- from astroid.nodes import NodeNG
- if sys.version_info >= (3, 8):
- from typing import Final
- else:
- from typing_extensions import Final
- if TYPE_CHECKING:
- import ast
- REDIRECT: Final[Dict[str, str]] = {
- "arguments": "Arguments",
- "comprehension": "Comprehension",
- "ListCompFor": "Comprehension",
- "GenExprFor": "Comprehension",
- "excepthandler": "ExceptHandler",
- "keyword": "Keyword",
- "match_case": "MatchCase",
- }
- T_Doc = TypeVar(
- "T_Doc",
- "ast.Module",
- "ast.ClassDef",
- Union["ast.FunctionDef", "ast.AsyncFunctionDef"],
- )
- T_Function = TypeVar("T_Function", nodes.FunctionDef, nodes.AsyncFunctionDef)
- T_For = TypeVar("T_For", nodes.For, nodes.AsyncFor)
- T_With = TypeVar("T_With", nodes.With, nodes.AsyncWith)
- # noinspection PyMethodMayBeStatic
- class TreeRebuilder:
- """Rebuilds the _ast tree to become an Astroid tree"""
- def __init__(
- self, manager: AstroidManager, parser_module: Optional[ParserModule] = None
- ):
- self._manager = manager
- self._global_names: List[Dict[str, List[nodes.Global]]] = []
- self._import_from_nodes: List[nodes.ImportFrom] = []
- self._delayed_assattr: List[nodes.AssignAttr] = []
- self._visit_meths: Dict[
- Type["ast.AST"], Callable[["ast.AST", NodeNG], NodeNG]
- ] = {}
- if parser_module is None:
- self._parser_module = get_parser_module()
- else:
- self._parser_module = parser_module
- self._module = self._parser_module.module
- def _get_doc(self, node: T_Doc) -> Tuple[T_Doc, Optional[str]]:
- try:
- if node.body and isinstance(node.body[0], self._module.Expr):
- first_value = node.body[0].value
- if isinstance(first_value, self._module.Str) or (
- PY38_PLUS
- and isinstance(first_value, self._module.Constant)
- and isinstance(first_value.value, str)
- ):
- doc = first_value.value if PY38_PLUS else first_value.s
- node.body = node.body[1:]
- return node, doc
- except IndexError:
- pass # ast built from scratch
- return node, None
- def _get_context(
- self,
- node: Union[
- "ast.Attribute",
- "ast.List",
- "ast.Name",
- "ast.Subscript",
- "ast.Starred",
- "ast.Tuple",
- ],
- ) -> Context:
- return self._parser_module.context_classes.get(type(node.ctx), Context.Load)
- def visit_module(
- self, node: "ast.Module", modname: str, modpath: str, package: bool
- ) -> nodes.Module:
- """visit a Module node by returning a fresh instance of it
- Note: Method not called by 'visit'
- """
- node, doc = self._get_doc(node)
- newnode = nodes.Module(
- name=modname,
- doc=doc,
- file=modpath,
- path=[modpath],
- package=package,
- parent=None,
- )
- newnode.postinit([self.visit(child, newnode) for child in node.body])
- return newnode
- if sys.version_info >= (3, 10):
- @overload
- def visit(self, node: "ast.arg", parent: NodeNG) -> nodes.AssignName:
- ...
- @overload
- def visit(self, node: "ast.arguments", parent: NodeNG) -> nodes.Arguments:
- ...
- @overload
- def visit(self, node: "ast.Assert", parent: NodeNG) -> nodes.Assert:
- ...
- @overload
- def visit(
- self, node: "ast.AsyncFunctionDef", parent: NodeNG
- ) -> nodes.AsyncFunctionDef:
- ...
- @overload
- def visit(self, node: "ast.AsyncFor", parent: NodeNG) -> nodes.AsyncFor:
- ...
- @overload
- def visit(self, node: "ast.Await", parent: NodeNG) -> nodes.Await:
- ...
- @overload
- def visit(self, node: "ast.AsyncWith", parent: NodeNG) -> nodes.AsyncWith:
- ...
- @overload
- def visit(self, node: "ast.Assign", parent: NodeNG) -> nodes.Assign:
- ...
- @overload
- def visit(self, node: "ast.AnnAssign", parent: NodeNG) -> nodes.AnnAssign:
- ...
- @overload
- def visit(self, node: "ast.AugAssign", parent: NodeNG) -> nodes.AugAssign:
- ...
- @overload
- def visit(self, node: "ast.BinOp", parent: NodeNG) -> nodes.BinOp:
- ...
- @overload
- def visit(self, node: "ast.BoolOp", parent: NodeNG) -> nodes.BoolOp:
- ...
- @overload
- def visit(self, node: "ast.Break", parent: NodeNG) -> nodes.Break:
- ...
- @overload
- def visit(self, node: "ast.Call", parent: NodeNG) -> nodes.Call:
- ...
- @overload
- def visit(self, node: "ast.ClassDef", parent: NodeNG) -> nodes.ClassDef:
- ...
- @overload
- def visit(self, node: "ast.Continue", parent: NodeNG) -> nodes.Continue:
- ...
- @overload
- def visit(self, node: "ast.Compare", parent: NodeNG) -> nodes.Compare:
- ...
- @overload
- def visit(
- self, node: "ast.comprehension", parent: NodeNG
- ) -> nodes.Comprehension:
- ...
- @overload
- def visit(self, node: "ast.Delete", parent: NodeNG) -> nodes.Delete:
- ...
- @overload
- def visit(self, node: "ast.Dict", parent: NodeNG) -> nodes.Dict:
- ...
- @overload
- def visit(self, node: "ast.DictComp", parent: NodeNG) -> nodes.DictComp:
- ...
- @overload
- def visit(self, node: "ast.Expr", parent: NodeNG) -> nodes.Expr:
- ...
- # Not used in Python 3.8+
- @overload
- def visit(self, node: "ast.Ellipsis", parent: NodeNG) -> nodes.Const:
- ...
- @overload
- def visit(
- self, node: "ast.ExceptHandler", parent: NodeNG
- ) -> nodes.ExceptHandler:
- ...
- # Not used in Python 3.9+
- @overload
- def visit(self, node: "ast.ExtSlice", parent: nodes.Subscript) -> nodes.Tuple:
- ...
- @overload
- def visit(self, node: "ast.For", parent: NodeNG) -> nodes.For:
- ...
- @overload
- def visit(self, node: "ast.ImportFrom", parent: NodeNG) -> nodes.ImportFrom:
- ...
- @overload
- def visit(self, node: "ast.FunctionDef", parent: NodeNG) -> nodes.FunctionDef:
- ...
- @overload
- def visit(self, node: "ast.GeneratorExp", parent: NodeNG) -> nodes.GeneratorExp:
- ...
- @overload
- def visit(self, node: "ast.Attribute", parent: NodeNG) -> nodes.Attribute:
- ...
- @overload
- def visit(self, node: "ast.Global", parent: NodeNG) -> nodes.Global:
- ...
- @overload
- def visit(self, node: "ast.If", parent: NodeNG) -> nodes.If:
- ...
- @overload
- def visit(self, node: "ast.IfExp", parent: NodeNG) -> nodes.IfExp:
- ...
- @overload
- def visit(self, node: "ast.Import", parent: NodeNG) -> nodes.Import:
- ...
- @overload
- def visit(self, node: "ast.JoinedStr", parent: NodeNG) -> nodes.JoinedStr:
- ...
- @overload
- def visit(
- self, node: "ast.FormattedValue", parent: NodeNG
- ) -> nodes.FormattedValue:
- ...
- @overload
- def visit(self, node: "ast.NamedExpr", parent: NodeNG) -> nodes.NamedExpr:
- ...
- # Not used in Python 3.9+
- @overload
- def visit(self, node: "ast.Index", parent: nodes.Subscript) -> NodeNG:
- ...
- @overload
- def visit(self, node: "ast.keyword", parent: NodeNG) -> nodes.Keyword:
- ...
- @overload
- def visit(self, node: "ast.Lambda", parent: NodeNG) -> nodes.Lambda:
- ...
- @overload
- def visit(self, node: "ast.List", parent: NodeNG) -> nodes.List:
- ...
- @overload
- def visit(self, node: "ast.ListComp", parent: NodeNG) -> nodes.ListComp:
- ...
- @overload
- def visit(
- self, node: "ast.Name", parent: NodeNG
- ) -> Union[nodes.Name, nodes.Const, nodes.AssignName, nodes.DelName]:
- ...
- # Not used in Python 3.8+
- @overload
- def visit(self, node: "ast.NameConstant", parent: NodeNG) -> nodes.Const:
- ...
- @overload
- def visit(self, node: "ast.Nonlocal", parent: NodeNG) -> nodes.Nonlocal:
- ...
- # Not used in Python 3.8+
- @overload
- def visit(self, node: "ast.Str", parent: NodeNG) -> nodes.Const:
- ...
- # Not used in Python 3.8+
- @overload
- def visit(self, node: "ast.Bytes", parent: NodeNG) -> nodes.Const:
- ...
- # Not used in Python 3.8+
- @overload
- def visit(self, node: "ast.Num", parent: NodeNG) -> nodes.Const:
- ...
- @overload
- def visit(self, node: "ast.Constant", parent: NodeNG) -> nodes.Const:
- ...
- @overload
- def visit(self, node: "ast.Pass", parent: NodeNG) -> nodes.Pass:
- ...
- @overload
- def visit(self, node: "ast.Raise", parent: NodeNG) -> nodes.Raise:
- ...
- @overload
- def visit(self, node: "ast.Return", parent: NodeNG) -> nodes.Return:
- ...
- @overload
- def visit(self, node: "ast.Set", parent: NodeNG) -> nodes.Set:
- ...
- @overload
- def visit(self, node: "ast.SetComp", parent: NodeNG) -> nodes.SetComp:
- ...
- @overload
- def visit(self, node: "ast.Slice", parent: nodes.Subscript) -> nodes.Slice:
- ...
- @overload
- def visit(self, node: "ast.Subscript", parent: NodeNG) -> nodes.Subscript:
- ...
- @overload
- def visit(self, node: "ast.Starred", parent: NodeNG) -> nodes.Starred:
- ...
- @overload
- def visit(
- self, node: "ast.Try", parent: NodeNG
- ) -> Union[nodes.TryExcept, nodes.TryFinally]:
- ...
- @overload
- def visit(self, node: "ast.Tuple", parent: NodeNG) -> nodes.Tuple:
- ...
- @overload
- def visit(self, node: "ast.UnaryOp", parent: NodeNG) -> nodes.UnaryOp:
- ...
- @overload
- def visit(self, node: "ast.While", parent: NodeNG) -> nodes.While:
- ...
- @overload
- def visit(self, node: "ast.With", parent: NodeNG) -> nodes.With:
- ...
- @overload
- def visit(self, node: "ast.Yield", parent: NodeNG) -> nodes.Yield:
- ...
- @overload
- def visit(self, node: "ast.YieldFrom", parent: NodeNG) -> nodes.YieldFrom:
- ...
- @overload
- def visit(self, node: "ast.Match", parent: NodeNG) -> nodes.Match:
- ...
- @overload
- def visit(self, node: "ast.match_case", parent: NodeNG) -> nodes.MatchCase:
- ...
- @overload
- def visit(self, node: "ast.MatchValue", parent: NodeNG) -> nodes.MatchValue:
- ...
- @overload
- def visit(
- self, node: "ast.MatchSingleton", parent: NodeNG
- ) -> nodes.MatchSingleton:
- ...
- @overload
- def visit(
- self, node: "ast.MatchSequence", parent: NodeNG
- ) -> nodes.MatchSequence:
- ...
- @overload
- def visit(self, node: "ast.MatchMapping", parent: NodeNG) -> nodes.MatchMapping:
- ...
- @overload
- def visit(self, node: "ast.MatchClass", parent: NodeNG) -> nodes.MatchClass:
- ...
- @overload
- def visit(self, node: "ast.MatchStar", parent: NodeNG) -> nodes.MatchStar:
- ...
- @overload
- def visit(self, node: "ast.MatchAs", parent: NodeNG) -> nodes.MatchAs:
- ...
- @overload
- def visit(self, node: "ast.MatchOr", parent: NodeNG) -> nodes.MatchOr:
- ...
- @overload
- def visit(self, node: "ast.pattern", parent: NodeNG) -> nodes.Pattern:
- ...
- @overload
- def visit(self, node: "ast.AST", parent: NodeNG) -> NodeNG:
- ...
- @overload
- def visit(self, node: None, parent: NodeNG) -> None:
- ...
- def visit(self, node: Optional["ast.AST"], parent: NodeNG) -> Optional[NodeNG]:
- if node is None:
- return None
- cls = node.__class__
- if cls in self._visit_meths:
- visit_method = self._visit_meths[cls]
- else:
- cls_name = cls.__name__
- visit_name = "visit_" + REDIRECT.get(cls_name, cls_name).lower()
- visit_method = getattr(self, visit_name)
- self._visit_meths[cls] = visit_method
- return visit_method(node, parent)
- else:
- @overload
- def visit(self, node: "ast.arg", parent: NodeNG) -> nodes.AssignName:
- ...
- @overload
- def visit(self, node: "ast.arguments", parent: NodeNG) -> nodes.Arguments:
- ...
- @overload
- def visit(self, node: "ast.Assert", parent: NodeNG) -> nodes.Assert:
- ...
- @overload
- def visit(
- self, node: "ast.AsyncFunctionDef", parent: NodeNG
- ) -> nodes.AsyncFunctionDef:
- ...
- @overload
- def visit(self, node: "ast.AsyncFor", parent: NodeNG) -> nodes.AsyncFor:
- ...
- @overload
- def visit(self, node: "ast.Await", parent: NodeNG) -> nodes.Await:
- ...
- @overload
- def visit(self, node: "ast.AsyncWith", parent: NodeNG) -> nodes.AsyncWith:
- ...
- @overload
- def visit(self, node: "ast.Assign", parent: NodeNG) -> nodes.Assign:
- ...
- @overload
- def visit(self, node: "ast.AnnAssign", parent: NodeNG) -> nodes.AnnAssign:
- ...
- @overload
- def visit(self, node: "ast.AugAssign", parent: NodeNG) -> nodes.AugAssign:
- ...
- @overload
- def visit(self, node: "ast.BinOp", parent: NodeNG) -> nodes.BinOp:
- ...
- @overload
- def visit(self, node: "ast.BoolOp", parent: NodeNG) -> nodes.BoolOp:
- ...
- @overload
- def visit(self, node: "ast.Break", parent: NodeNG) -> nodes.Break:
- ...
- @overload
- def visit(self, node: "ast.Call", parent: NodeNG) -> nodes.Call:
- ...
- @overload
- def visit(self, node: "ast.ClassDef", parent: NodeNG) -> nodes.ClassDef:
- ...
- @overload
- def visit(self, node: "ast.Continue", parent: NodeNG) -> nodes.Continue:
- ...
- @overload
- def visit(self, node: "ast.Compare", parent: NodeNG) -> nodes.Compare:
- ...
- @overload
- def visit(
- self, node: "ast.comprehension", parent: NodeNG
- ) -> nodes.Comprehension:
- ...
- @overload
- def visit(self, node: "ast.Delete", parent: NodeNG) -> nodes.Delete:
- ...
- @overload
- def visit(self, node: "ast.Dict", parent: NodeNG) -> nodes.Dict:
- ...
- @overload
- def visit(self, node: "ast.DictComp", parent: NodeNG) -> nodes.DictComp:
- ...
- @overload
- def visit(self, node: "ast.Expr", parent: NodeNG) -> nodes.Expr:
- ...
- # Not used in Python 3.8+
- @overload
- def visit(self, node: "ast.Ellipsis", parent: NodeNG) -> nodes.Const:
- ...
- @overload
- def visit(
- self, node: "ast.ExceptHandler", parent: NodeNG
- ) -> nodes.ExceptHandler:
- ...
- # Not used in Python 3.9+
- @overload
- def visit(self, node: "ast.ExtSlice", parent: nodes.Subscript) -> nodes.Tuple:
- ...
- @overload
- def visit(self, node: "ast.For", parent: NodeNG) -> nodes.For:
- ...
- @overload
- def visit(self, node: "ast.ImportFrom", parent: NodeNG) -> nodes.ImportFrom:
- ...
- @overload
- def visit(self, node: "ast.FunctionDef", parent: NodeNG) -> nodes.FunctionDef:
- ...
- @overload
- def visit(self, node: "ast.GeneratorExp", parent: NodeNG) -> nodes.GeneratorExp:
- ...
- @overload
- def visit(self, node: "ast.Attribute", parent: NodeNG) -> nodes.Attribute:
- ...
- @overload
- def visit(self, node: "ast.Global", parent: NodeNG) -> nodes.Global:
- ...
- @overload
- def visit(self, node: "ast.If", parent: NodeNG) -> nodes.If:
- ...
- @overload
- def visit(self, node: "ast.IfExp", parent: NodeNG) -> nodes.IfExp:
- ...
- @overload
- def visit(self, node: "ast.Import", parent: NodeNG) -> nodes.Import:
- ...
- @overload
- def visit(self, node: "ast.JoinedStr", parent: NodeNG) -> nodes.JoinedStr:
- ...
- @overload
- def visit(
- self, node: "ast.FormattedValue", parent: NodeNG
- ) -> nodes.FormattedValue:
- ...
- @overload
- def visit(self, node: "ast.NamedExpr", parent: NodeNG) -> nodes.NamedExpr:
- ...
- # Not used in Python 3.9+
- @overload
- def visit(self, node: "ast.Index", parent: nodes.Subscript) -> NodeNG:
- ...
- @overload
- def visit(self, node: "ast.keyword", parent: NodeNG) -> nodes.Keyword:
- ...
- @overload
- def visit(self, node: "ast.Lambda", parent: NodeNG) -> nodes.Lambda:
- ...
- @overload
- def visit(self, node: "ast.List", parent: NodeNG) -> nodes.List:
- ...
- @overload
- def visit(self, node: "ast.ListComp", parent: NodeNG) -> nodes.ListComp:
- ...
- @overload
- def visit(
- self, node: "ast.Name", parent: NodeNG
- ) -> Union[nodes.Name, nodes.Const, nodes.AssignName, nodes.DelName]:
- ...
- # Not used in Python 3.8+
- @overload
- def visit(self, node: "ast.NameConstant", parent: NodeNG) -> nodes.Const:
- ...
- @overload
- def visit(self, node: "ast.Nonlocal", parent: NodeNG) -> nodes.Nonlocal:
- ...
- # Not used in Python 3.8+
- @overload
- def visit(self, node: "ast.Str", parent: NodeNG) -> nodes.Const:
- ...
- # Not used in Python 3.8+
- @overload
- def visit(self, node: "ast.Bytes", parent: NodeNG) -> nodes.Const:
- ...
- # Not used in Python 3.8+
- @overload
- def visit(self, node: "ast.Num", parent: NodeNG) -> nodes.Const:
- ...
- @overload
- def visit(self, node: "ast.Constant", parent: NodeNG) -> nodes.Const:
- ...
- @overload
- def visit(self, node: "ast.Pass", parent: NodeNG) -> nodes.Pass:
- ...
- @overload
- def visit(self, node: "ast.Raise", parent: NodeNG) -> nodes.Raise:
- ...
- @overload
- def visit(self, node: "ast.Return", parent: NodeNG) -> nodes.Return:
- ...
- @overload
- def visit(self, node: "ast.Set", parent: NodeNG) -> nodes.Set:
- ...
- @overload
- def visit(self, node: "ast.SetComp", parent: NodeNG) -> nodes.SetComp:
- ...
- @overload
- def visit(self, node: "ast.Slice", parent: nodes.Subscript) -> nodes.Slice:
- ...
- @overload
- def visit(self, node: "ast.Subscript", parent: NodeNG) -> nodes.Subscript:
- ...
- @overload
- def visit(self, node: "ast.Starred", parent: NodeNG) -> nodes.Starred:
- ...
- @overload
- def visit(
- self, node: "ast.Try", parent: NodeNG
- ) -> Union[nodes.TryExcept, nodes.TryFinally]:
- ...
- @overload
- def visit(self, node: "ast.Tuple", parent: NodeNG) -> nodes.Tuple:
- ...
- @overload
- def visit(self, node: "ast.UnaryOp", parent: NodeNG) -> nodes.UnaryOp:
- ...
- @overload
- def visit(self, node: "ast.While", parent: NodeNG) -> nodes.While:
- ...
- @overload
- def visit(self, node: "ast.With", parent: NodeNG) -> nodes.With:
- ...
- @overload
- def visit(self, node: "ast.Yield", parent: NodeNG) -> nodes.Yield:
- ...
- @overload
- def visit(self, node: "ast.YieldFrom", parent: NodeNG) -> nodes.YieldFrom:
- ...
- @overload
- def visit(self, node: "ast.AST", parent: NodeNG) -> NodeNG:
- ...
- @overload
- def visit(self, node: None, parent: NodeNG) -> None:
- ...
- def visit(self, node: Optional["ast.AST"], parent: NodeNG) -> Optional[NodeNG]:
- if node is None:
- return None
- cls = node.__class__
- if cls in self._visit_meths:
- visit_method = self._visit_meths[cls]
- else:
- cls_name = cls.__name__
- visit_name = "visit_" + REDIRECT.get(cls_name, cls_name).lower()
- visit_method = getattr(self, visit_name)
- self._visit_meths[cls] = visit_method
- return visit_method(node, parent)
- def _save_assignment(self, node: Union[nodes.AssignName, nodes.DelName]) -> None:
- """save assignment situation since node.parent is not available yet"""
- if self._global_names and node.name in self._global_names[-1]:
- node.root().set_local(node.name, node)
- else:
- assert node.parent
- node.parent.set_local(node.name, node)
- def visit_arg(self, node: "ast.arg", parent: NodeNG) -> nodes.AssignName:
- """visit an arg node by returning a fresh AssName instance"""
- return self.visit_assignname(node, parent, node.arg)
- def visit_arguments(self, node: "ast.arguments", parent: NodeNG) -> nodes.Arguments:
- """visit an Arguments node by returning a fresh instance of it"""
- vararg: Optional[str] = None
- kwarg: Optional[str] = None
- newnode = nodes.Arguments(
- node.vararg.arg if node.vararg else None,
- node.kwarg.arg if node.kwarg else None,
- parent,
- )
- args = [self.visit(child, newnode) for child in node.args]
- defaults = [self.visit(child, newnode) for child in node.defaults]
- varargannotation: Optional[NodeNG] = None
- kwargannotation: Optional[NodeNG] = None
- posonlyargs: List[nodes.AssignName] = []
- if node.vararg:
- vararg = node.vararg.arg
- varargannotation = self.visit(node.vararg.annotation, newnode)
- if node.kwarg:
- kwarg = node.kwarg.arg
- kwargannotation = self.visit(node.kwarg.annotation, newnode)
- if PY38:
- # In Python 3.8 'end_lineno' and 'end_col_offset'
- # for 'kwonlyargs' don't include the annotation.
- for arg in node.kwonlyargs:
- if arg.annotation is not None:
- arg.end_lineno = arg.annotation.end_lineno
- arg.end_col_offset = arg.annotation.end_col_offset
- kwonlyargs = [self.visit(child, newnode) for child in node.kwonlyargs]
- kw_defaults = [self.visit(child, newnode) for child in node.kw_defaults]
- annotations = [self.visit(arg.annotation, newnode) for arg in node.args]
- kwonlyargs_annotations = [
- self.visit(arg.annotation, newnode) for arg in node.kwonlyargs
- ]
- posonlyargs_annotations: List[Optional[NodeNG]] = []
- if PY38_PLUS:
- posonlyargs = [self.visit(child, newnode) for child in node.posonlyargs]
- posonlyargs_annotations = [
- self.visit(arg.annotation, newnode) for arg in node.posonlyargs
- ]
- type_comment_args = [
- self.check_type_comment(child, parent=newnode) for child in node.args
- ]
- type_comment_kwonlyargs = [
- self.check_type_comment(child, parent=newnode) for child in node.kwonlyargs
- ]
- type_comment_posonlyargs: List[Optional[NodeNG]] = []
- if PY38_PLUS:
- type_comment_posonlyargs = [
- self.check_type_comment(child, parent=newnode)
- for child in node.posonlyargs
- ]
- newnode.postinit(
- args=args,
- defaults=defaults,
- kwonlyargs=kwonlyargs,
- posonlyargs=posonlyargs,
- kw_defaults=kw_defaults,
- annotations=annotations,
- kwonlyargs_annotations=kwonlyargs_annotations,
- posonlyargs_annotations=posonlyargs_annotations,
- varargannotation=varargannotation,
- kwargannotation=kwargannotation,
- type_comment_args=type_comment_args,
- type_comment_kwonlyargs=type_comment_kwonlyargs,
- type_comment_posonlyargs=type_comment_posonlyargs,
- )
- # save argument names in locals:
- assert newnode.parent
- if vararg:
- newnode.parent.set_local(vararg, newnode)
- if kwarg:
- newnode.parent.set_local(kwarg, newnode)
- return newnode
- def visit_assert(self, node: "ast.Assert", parent: NodeNG) -> nodes.Assert:
- """visit a Assert node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.Assert(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Assert(node.lineno, node.col_offset, parent)
- msg: Optional[NodeNG] = None
- if node.msg:
- msg = self.visit(node.msg, newnode)
- newnode.postinit(self.visit(node.test, newnode), msg)
- return newnode
- def check_type_comment(
- self,
- node: Union[
- "ast.Assign",
- "ast.arg",
- "ast.For",
- "ast.AsyncFor",
- "ast.With",
- "ast.AsyncWith",
- ],
- parent: Union[
- nodes.Assign,
- nodes.Arguments,
- nodes.For,
- nodes.AsyncFor,
- nodes.With,
- nodes.AsyncWith,
- ],
- ) -> Optional[NodeNG]:
- type_comment = getattr(node, "type_comment", None) # Added in Python 3.8
- if not type_comment:
- return None
- try:
- type_comment_ast = self._parser_module.parse(type_comment)
- except SyntaxError:
- # Invalid type comment, just skip it.
- return None
- type_object = self.visit(type_comment_ast.body[0], parent=parent)
- if not isinstance(type_object, nodes.Expr):
- return None
- return type_object.value
- def check_function_type_comment(
- self, node: Union["ast.FunctionDef", "ast.AsyncFunctionDef"], parent: NodeNG
- ) -> Optional[Tuple[Optional[NodeNG], List[NodeNG]]]:
- type_comment = getattr(node, "type_comment", None) # Added in Python 3.8
- if not type_comment:
- return None
- try:
- type_comment_ast = parse_function_type_comment(type_comment)
- except SyntaxError:
- # Invalid type comment, just skip it.
- return None
- if not type_comment_ast:
- return None
- returns: Optional[NodeNG] = None
- argtypes: List[NodeNG] = [
- self.visit(elem, parent) for elem in (type_comment_ast.argtypes or [])
- ]
- if type_comment_ast.returns:
- returns = self.visit(type_comment_ast.returns, parent)
- return returns, argtypes
- def visit_asyncfunctiondef(
- self, node: "ast.AsyncFunctionDef", parent: NodeNG
- ) -> nodes.AsyncFunctionDef:
- return self._visit_functiondef(nodes.AsyncFunctionDef, node, parent)
- def visit_asyncfor(self, node: "ast.AsyncFor", parent: NodeNG) -> nodes.AsyncFor:
- return self._visit_for(nodes.AsyncFor, node, parent)
- def visit_await(self, node: "ast.Await", parent: NodeNG) -> nodes.Await:
- if sys.version_info >= (3, 8):
- newnode = nodes.Await(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Await(node.lineno, node.col_offset, parent)
- newnode.postinit(value=self.visit(node.value, newnode))
- return newnode
- def visit_asyncwith(self, node: "ast.AsyncWith", parent: NodeNG) -> nodes.AsyncWith:
- return self._visit_with(nodes.AsyncWith, node, parent)
- def visit_assign(self, node: "ast.Assign", parent: NodeNG) -> nodes.Assign:
- """visit a Assign node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.Assign(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Assign(node.lineno, node.col_offset, parent)
- type_annotation = self.check_type_comment(node, parent=newnode)
- newnode.postinit(
- targets=[self.visit(child, newnode) for child in node.targets],
- value=self.visit(node.value, newnode),
- type_annotation=type_annotation,
- )
- return newnode
- def visit_annassign(self, node: "ast.AnnAssign", parent: NodeNG) -> nodes.AnnAssign:
- """visit an AnnAssign node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.AnnAssign(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.AnnAssign(node.lineno, node.col_offset, parent)
- newnode.postinit(
- target=self.visit(node.target, newnode),
- annotation=self.visit(node.annotation, newnode),
- simple=node.simple,
- value=self.visit(node.value, newnode),
- )
- return newnode
- @overload
- def visit_assignname(
- self, node: "ast.AST", parent: NodeNG, node_name: str
- ) -> nodes.AssignName:
- ...
- @overload
- def visit_assignname(
- self, node: "ast.AST", parent: NodeNG, node_name: None
- ) -> None:
- ...
- def visit_assignname(
- self, node: "ast.AST", parent: NodeNG, node_name: Optional[str]
- ) -> Optional[nodes.AssignName]:
- """visit a node and return a AssignName node
- Note: Method not called by 'visit'
- """
- if node_name is None:
- return None
- if sys.version_info >= (3, 8):
- newnode = nodes.AssignName(
- name=node_name,
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.AssignName(
- node_name,
- node.lineno,
- node.col_offset,
- parent,
- )
- self._save_assignment(newnode)
- return newnode
- def visit_augassign(self, node: "ast.AugAssign", parent: NodeNG) -> nodes.AugAssign:
- """visit a AugAssign node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.AugAssign(
- op=self._parser_module.bin_op_classes[type(node.op)] + "=",
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.AugAssign(
- self._parser_module.bin_op_classes[type(node.op)] + "=",
- node.lineno,
- node.col_offset,
- parent,
- )
- newnode.postinit(
- self.visit(node.target, newnode), self.visit(node.value, newnode)
- )
- return newnode
- def visit_binop(self, node: "ast.BinOp", parent: NodeNG) -> nodes.BinOp:
- """visit a BinOp node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.BinOp(
- op=self._parser_module.bin_op_classes[type(node.op)],
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.BinOp(
- self._parser_module.bin_op_classes[type(node.op)],
- node.lineno,
- node.col_offset,
- parent,
- )
- newnode.postinit(
- self.visit(node.left, newnode), self.visit(node.right, newnode)
- )
- return newnode
- def visit_boolop(self, node: "ast.BoolOp", parent: NodeNG) -> nodes.BoolOp:
- """visit a BoolOp node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.BoolOp(
- op=self._parser_module.bool_op_classes[type(node.op)],
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.BoolOp(
- self._parser_module.bool_op_classes[type(node.op)],
- node.lineno,
- node.col_offset,
- parent,
- )
- newnode.postinit([self.visit(child, newnode) for child in node.values])
- return newnode
- def visit_break(self, node: "ast.Break", parent: NodeNG) -> nodes.Break:
- """visit a Break node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- return nodes.Break(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- return nodes.Break(node.lineno, node.col_offset, parent)
- def visit_call(self, node: "ast.Call", parent: NodeNG) -> nodes.Call:
- """visit a CallFunc node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.Call(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Call(node.lineno, node.col_offset, parent)
- newnode.postinit(
- func=self.visit(node.func, newnode),
- args=[self.visit(child, newnode) for child in node.args],
- keywords=[self.visit(child, newnode) for child in node.keywords],
- )
- return newnode
- def visit_classdef(
- self, node: "ast.ClassDef", parent: NodeNG, newstyle: bool = True
- ) -> nodes.ClassDef:
- """visit a ClassDef node to become astroid"""
- node, doc = self._get_doc(node)
- if sys.version_info >= (3, 8):
- newnode = nodes.ClassDef(
- name=node.name,
- doc=doc,
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.ClassDef(
- node.name, doc, node.lineno, node.col_offset, parent
- )
- metaclass = None
- for keyword in node.keywords:
- if keyword.arg == "metaclass":
- metaclass = self.visit(keyword, newnode).value
- break
- decorators = self.visit_decorators(node, newnode)
- newnode.postinit(
- [self.visit(child, newnode) for child in node.bases],
- [self.visit(child, newnode) for child in node.body],
- decorators,
- newstyle,
- metaclass,
- [
- self.visit(kwd, newnode)
- for kwd in node.keywords
- if kwd.arg != "metaclass"
- ],
- )
- return newnode
- def visit_continue(self, node: "ast.Continue", parent: NodeNG) -> nodes.Continue:
- """visit a Continue node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- return nodes.Continue(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- return nodes.Continue(node.lineno, node.col_offset, parent)
- def visit_compare(self, node: "ast.Compare", parent: NodeNG) -> nodes.Compare:
- """visit a Compare node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.Compare(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Compare(node.lineno, node.col_offset, parent)
- newnode.postinit(
- self.visit(node.left, newnode),
- [
- (
- self._parser_module.cmp_op_classes[op.__class__],
- self.visit(expr, newnode),
- )
- for (op, expr) in zip(node.ops, node.comparators)
- ],
- )
- return newnode
- def visit_comprehension(
- self, node: "ast.comprehension", parent: NodeNG
- ) -> nodes.Comprehension:
- """visit a Comprehension node by returning a fresh instance of it"""
- newnode = nodes.Comprehension(parent)
- newnode.postinit(
- self.visit(node.target, newnode),
- self.visit(node.iter, newnode),
- [self.visit(child, newnode) for child in node.ifs],
- bool(node.is_async),
- )
- return newnode
- def visit_decorators(
- self,
- node: Union["ast.ClassDef", "ast.FunctionDef", "ast.AsyncFunctionDef"],
- parent: NodeNG,
- ) -> Optional[nodes.Decorators]:
- """visit a Decorators node by returning a fresh instance of it
- Note: Method not called by 'visit'
- """
- if not node.decorator_list:
- return None
- # /!\ node is actually an _ast.FunctionDef node while
- # parent is an astroid.nodes.FunctionDef node
- if sys.version_info >= (3, 8):
- # Set the line number of the first decorator for Python 3.8+.
- lineno = node.decorator_list[0].lineno
- end_lineno = node.decorator_list[-1].end_lineno
- end_col_offset = node.decorator_list[-1].end_col_offset
- else:
- lineno = node.lineno
- end_lineno = None
- end_col_offset = None
- newnode = nodes.Decorators(
- lineno=lineno,
- col_offset=node.col_offset,
- end_lineno=end_lineno,
- end_col_offset=end_col_offset,
- parent=parent,
- )
- newnode.postinit([self.visit(child, newnode) for child in node.decorator_list])
- return newnode
- def visit_delete(self, node: "ast.Delete", parent: NodeNG) -> nodes.Delete:
- """visit a Delete node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.Delete(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Delete(node.lineno, node.col_offset, parent)
- newnode.postinit([self.visit(child, newnode) for child in node.targets])
- return newnode
- def _visit_dict_items(
- self, node: "ast.Dict", parent: NodeNG, newnode: nodes.Dict
- ) -> Generator[Tuple[NodeNG, NodeNG], None, None]:
- for key, value in zip(node.keys, node.values):
- rebuilt_key: NodeNG
- rebuilt_value = self.visit(value, newnode)
- if not key:
- # Extended unpacking
- if sys.version_info >= (3, 8):
- rebuilt_key = nodes.DictUnpack(
- lineno=rebuilt_value.lineno,
- col_offset=rebuilt_value.col_offset,
- end_lineno=rebuilt_value.end_lineno,
- end_col_offset=rebuilt_value.end_col_offset,
- parent=parent,
- )
- else:
- rebuilt_key = nodes.DictUnpack(
- rebuilt_value.lineno, rebuilt_value.col_offset, parent
- )
- else:
- rebuilt_key = self.visit(key, newnode)
- yield rebuilt_key, rebuilt_value
- def visit_dict(self, node: "ast.Dict", parent: NodeNG) -> nodes.Dict:
- """visit a Dict node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.Dict(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Dict(node.lineno, node.col_offset, parent)
- items = list(self._visit_dict_items(node, parent, newnode))
- newnode.postinit(items)
- return newnode
- def visit_dictcomp(self, node: "ast.DictComp", parent: NodeNG) -> nodes.DictComp:
- """visit a DictComp node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.DictComp(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.DictComp(node.lineno, node.col_offset, parent)
- newnode.postinit(
- self.visit(node.key, newnode),
- self.visit(node.value, newnode),
- [self.visit(child, newnode) for child in node.generators],
- )
- return newnode
- def visit_expr(self, node: "ast.Expr", parent: NodeNG) -> nodes.Expr:
- """visit a Expr node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.Expr(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Expr(node.lineno, node.col_offset, parent)
- newnode.postinit(self.visit(node.value, newnode))
- return newnode
- # Not used in Python 3.8+.
- def visit_ellipsis(self, node: "ast.Ellipsis", parent: NodeNG) -> nodes.Const:
- """visit an Ellipsis node by returning a fresh instance of Const"""
- return nodes.Const(
- value=Ellipsis,
- lineno=node.lineno,
- col_offset=node.col_offset,
- parent=parent,
- )
- def visit_excepthandler(
- self, node: "ast.ExceptHandler", parent: NodeNG
- ) -> nodes.ExceptHandler:
- """visit an ExceptHandler node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.ExceptHandler(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.ExceptHandler(node.lineno, node.col_offset, parent)
- newnode.postinit(
- self.visit(node.type, newnode),
- self.visit_assignname(node, newnode, node.name),
- [self.visit(child, newnode) for child in node.body],
- )
- return newnode
- # Not used in Python 3.9+.
- def visit_extslice(
- self, node: "ast.ExtSlice", parent: nodes.Subscript
- ) -> nodes.Tuple:
- """visit an ExtSlice node by returning a fresh instance of Tuple"""
- # ExtSlice doesn't have lineno or col_offset information
- newnode = nodes.Tuple(ctx=Context.Load, parent=parent)
- newnode.postinit([self.visit(dim, newnode) for dim in node.dims]) # type: ignore[attr-defined]
- return newnode
- @overload
- def _visit_for(
- self, cls: Type[nodes.For], node: "ast.For", parent: NodeNG
- ) -> nodes.For:
- ...
- @overload
- def _visit_for(
- self, cls: Type[nodes.AsyncFor], node: "ast.AsyncFor", parent: NodeNG
- ) -> nodes.AsyncFor:
- ...
- def _visit_for(
- self, cls: Type[T_For], node: Union["ast.For", "ast.AsyncFor"], parent: NodeNG
- ) -> T_For:
- """visit a For node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = cls(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = cls(node.lineno, node.col_offset, parent)
- type_annotation = self.check_type_comment(node, parent=newnode)
- newnode.postinit(
- target=self.visit(node.target, newnode),
- iter=self.visit(node.iter, newnode),
- body=[self.visit(child, newnode) for child in node.body],
- orelse=[self.visit(child, newnode) for child in node.orelse],
- type_annotation=type_annotation,
- )
- return newnode
- def visit_for(self, node: "ast.For", parent: NodeNG) -> nodes.For:
- return self._visit_for(nodes.For, node, parent)
- def visit_importfrom(
- self, node: "ast.ImportFrom", parent: NodeNG
- ) -> nodes.ImportFrom:
- """visit an ImportFrom node by returning a fresh instance of it"""
- names = [(alias.name, alias.asname) for alias in node.names]
- if sys.version_info >= (3, 8):
- newnode = nodes.ImportFrom(
- fromname=node.module or "",
- names=names,
- level=node.level or None,
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.ImportFrom(
- node.module or "",
- names,
- node.level or None,
- node.lineno,
- node.col_offset,
- parent,
- )
- # store From names to add them to locals after building
- self._import_from_nodes.append(newnode)
- return newnode
- @overload
- def _visit_functiondef(
- self, cls: Type[nodes.FunctionDef], node: "ast.FunctionDef", parent: NodeNG
- ) -> nodes.FunctionDef:
- ...
- @overload
- def _visit_functiondef(
- self,
- cls: Type[nodes.AsyncFunctionDef],
- node: "ast.AsyncFunctionDef",
- parent: NodeNG,
- ) -> nodes.AsyncFunctionDef:
- ...
- def _visit_functiondef(
- self,
- cls: Type[T_Function],
- node: Union["ast.FunctionDef", "ast.AsyncFunctionDef"],
- parent: NodeNG,
- ) -> T_Function:
- """visit an FunctionDef node to become astroid"""
- self._global_names.append({})
- node, doc = self._get_doc(node)
- lineno = node.lineno
- if PY38_PLUS and node.decorator_list:
- # Python 3.8 sets the line number of a decorated function
- # to be the actual line number of the function, but the
- # previous versions expected the decorator's line number instead.
- # We reset the function's line number to that of the
- # first decorator to maintain backward compatibility.
- # It's not ideal but this discrepancy was baked into
- # the framework for *years*.
- lineno = node.decorator_list[0].lineno
- if sys.version_info >= (3, 8):
- newnode = cls(
- name=node.name,
- doc=doc,
- lineno=lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = cls(node.name, doc, lineno, node.col_offset, parent)
- decorators = self.visit_decorators(node, newnode)
- returns: Optional[NodeNG]
- if node.returns:
- returns = self.visit(node.returns, newnode)
- else:
- returns = None
- type_comment_args = type_comment_returns = None
- type_comment_annotation = self.check_function_type_comment(node, newnode)
- if type_comment_annotation:
- type_comment_returns, type_comment_args = type_comment_annotation
- newnode.postinit(
- args=self.visit(node.args, newnode),
- body=[self.visit(child, newnode) for child in node.body],
- decorators=decorators,
- returns=returns,
- type_comment_returns=type_comment_returns,
- type_comment_args=type_comment_args,
- )
- self._global_names.pop()
- return newnode
- def visit_functiondef(
- self, node: "ast.FunctionDef", parent: NodeNG
- ) -> nodes.FunctionDef:
- return self._visit_functiondef(nodes.FunctionDef, node, parent)
- def visit_generatorexp(
- self, node: "ast.GeneratorExp", parent: NodeNG
- ) -> nodes.GeneratorExp:
- """visit a GeneratorExp node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.GeneratorExp(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.GeneratorExp(node.lineno, node.col_offset, parent)
- newnode.postinit(
- self.visit(node.elt, newnode),
- [self.visit(child, newnode) for child in node.generators],
- )
- return newnode
- def visit_attribute(
- self, node: "ast.Attribute", parent: NodeNG
- ) -> Union[nodes.Attribute, nodes.AssignAttr, nodes.DelAttr]:
- """visit an Attribute node by returning a fresh instance of it"""
- context = self._get_context(node)
- newnode: Union[nodes.Attribute, nodes.AssignAttr, nodes.DelAttr]
- if context == Context.Del:
- # FIXME : maybe we should reintroduce and visit_delattr ?
- # for instance, deactivating assign_ctx
- if sys.version_info >= (3, 8):
- newnode = nodes.DelAttr(
- attrname=node.attr,
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.DelAttr(node.attr, node.lineno, node.col_offset, parent)
- elif context == Context.Store:
- if sys.version_info >= (3, 8):
- newnode = nodes.AssignAttr(
- attrname=node.attr,
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.AssignAttr(
- node.attr, node.lineno, node.col_offset, parent
- )
- # Prohibit a local save if we are in an ExceptHandler.
- if not isinstance(parent, nodes.ExceptHandler):
- # mypy doesn't recognize that newnode has to be AssignAttr because it doesn't support ParamSpec
- # See https://github.com/python/mypy/issues/8645
- self._delayed_assattr.append(newnode) # type: ignore[arg-type]
- else:
- # pylint: disable-next=else-if-used
- # Preserve symmetry with other cases
- if sys.version_info >= (3, 8):
- newnode = nodes.Attribute(
- attrname=node.attr,
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Attribute(
- node.attr, node.lineno, node.col_offset, parent
- )
- newnode.postinit(self.visit(node.value, newnode))
- return newnode
- def visit_global(self, node: "ast.Global", parent: NodeNG) -> nodes.Global:
- """visit a Global node to become astroid"""
- if sys.version_info >= (3, 8):
- newnode = nodes.Global(
- names=node.names,
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Global(
- node.names,
- node.lineno,
- node.col_offset,
- parent,
- )
- if self._global_names: # global at the module level, no effect
- for name in node.names:
- self._global_names[-1].setdefault(name, []).append(newnode)
- return newnode
- def visit_if(self, node: "ast.If", parent: NodeNG) -> nodes.If:
- """visit an If node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.If(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.If(node.lineno, node.col_offset, parent)
- newnode.postinit(
- self.visit(node.test, newnode),
- [self.visit(child, newnode) for child in node.body],
- [self.visit(child, newnode) for child in node.orelse],
- )
- return newnode
- def visit_ifexp(self, node: "ast.IfExp", parent: NodeNG) -> nodes.IfExp:
- """visit a IfExp node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.IfExp(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.IfExp(node.lineno, node.col_offset, parent)
- newnode.postinit(
- self.visit(node.test, newnode),
- self.visit(node.body, newnode),
- self.visit(node.orelse, newnode),
- )
- return newnode
- def visit_import(self, node: "ast.Import", parent: NodeNG) -> nodes.Import:
- """visit a Import node by returning a fresh instance of it"""
- names = [(alias.name, alias.asname) for alias in node.names]
- if sys.version_info >= (3, 8):
- newnode = nodes.Import(
- names=names,
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Import(
- names,
- node.lineno,
- node.col_offset,
- parent,
- )
- # save import names in parent's locals:
- for (name, asname) in newnode.names:
- name = asname or name
- parent.set_local(name.split(".")[0], newnode)
- return newnode
- def visit_joinedstr(self, node: "ast.JoinedStr", parent: NodeNG) -> nodes.JoinedStr:
- if sys.version_info >= (3, 8):
- newnode = nodes.JoinedStr(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.JoinedStr(node.lineno, node.col_offset, parent)
- newnode.postinit([self.visit(child, newnode) for child in node.values])
- return newnode
- def visit_formattedvalue(
- self, node: "ast.FormattedValue", parent: NodeNG
- ) -> nodes.FormattedValue:
- if sys.version_info >= (3, 8):
- newnode = nodes.FormattedValue(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.FormattedValue(node.lineno, node.col_offset, parent)
- newnode.postinit(
- self.visit(node.value, newnode),
- node.conversion,
- self.visit(node.format_spec, newnode),
- )
- return newnode
- def visit_namedexpr(self, node: "ast.NamedExpr", parent: NodeNG) -> nodes.NamedExpr:
- if sys.version_info >= (3, 8):
- newnode = nodes.NamedExpr(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.NamedExpr(node.lineno, node.col_offset, parent)
- newnode.postinit(
- self.visit(node.target, newnode), self.visit(node.value, newnode)
- )
- return newnode
- # Not used in Python 3.9+.
- def visit_index(self, node: "ast.Index", parent: nodes.Subscript) -> NodeNG:
- """visit a Index node by returning a fresh instance of NodeNG"""
- return self.visit(node.value, parent) # type: ignore[attr-defined]
- def visit_keyword(self, node: "ast.keyword", parent: NodeNG) -> nodes.Keyword:
- """visit a Keyword node by returning a fresh instance of it"""
- if sys.version_info >= (3, 9):
- newnode = nodes.Keyword(
- arg=node.arg,
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Keyword(node.arg, parent=parent)
- newnode.postinit(self.visit(node.value, newnode))
- return newnode
- def visit_lambda(self, node: "ast.Lambda", parent: NodeNG) -> nodes.Lambda:
- """visit a Lambda node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.Lambda(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Lambda(node.lineno, node.col_offset, parent)
- newnode.postinit(self.visit(node.args, newnode), self.visit(node.body, newnode))
- return newnode
- def visit_list(self, node: "ast.List", parent: NodeNG) -> nodes.List:
- """visit a List node by returning a fresh instance of it"""
- context = self._get_context(node)
- if sys.version_info >= (3, 8):
- newnode = nodes.List(
- ctx=context,
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.List(
- ctx=context,
- lineno=node.lineno,
- col_offset=node.col_offset,
- parent=parent,
- )
- newnode.postinit([self.visit(child, newnode) for child in node.elts])
- return newnode
- def visit_listcomp(self, node: "ast.ListComp", parent: NodeNG) -> nodes.ListComp:
- """visit a ListComp node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.ListComp(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.ListComp(node.lineno, node.col_offset, parent)
- newnode.postinit(
- self.visit(node.elt, newnode),
- [self.visit(child, newnode) for child in node.generators],
- )
- return newnode
- def visit_name(
- self, node: "ast.Name", parent: NodeNG
- ) -> Union[nodes.Name, nodes.AssignName, nodes.DelName]:
- """visit a Name node by returning a fresh instance of it"""
- context = self._get_context(node)
- newnode: Union[nodes.Name, nodes.AssignName, nodes.DelName]
- if context == Context.Del:
- if sys.version_info >= (3, 8):
- newnode = nodes.DelName(
- name=node.id,
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.DelName(node.id, node.lineno, node.col_offset, parent)
- elif context == Context.Store:
- if sys.version_info >= (3, 8):
- newnode = nodes.AssignName(
- name=node.id,
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.AssignName(
- node.id, node.lineno, node.col_offset, parent
- )
- else:
- # pylint: disable-next=else-if-used
- # Preserve symmetry with other cases
- if sys.version_info >= (3, 8):
- newnode = nodes.Name(
- name=node.id,
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Name(node.id, node.lineno, node.col_offset, parent)
- # XXX REMOVE me :
- if context in (Context.Del, Context.Store): # 'Aug' ??
- newnode = cast(Union[nodes.AssignName, nodes.DelName], newnode)
- self._save_assignment(newnode)
- return newnode
- # Not used in Python 3.8+.
- def visit_nameconstant(
- self, node: "ast.NameConstant", parent: NodeNG
- ) -> nodes.Const:
- # For singleton values True / False / None
- return nodes.Const(
- node.value,
- node.lineno,
- node.col_offset,
- parent,
- )
- def visit_nonlocal(self, node: "ast.Nonlocal", parent: NodeNG) -> nodes.Nonlocal:
- """visit a Nonlocal node and return a new instance of it"""
- if sys.version_info >= (3, 8):
- return nodes.Nonlocal(
- names=node.names,
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- return nodes.Nonlocal(
- node.names,
- node.lineno,
- node.col_offset,
- parent,
- )
- def visit_constant(self, node: "ast.Constant", parent: NodeNG) -> nodes.Const:
- """visit a Constant node by returning a fresh instance of Const"""
- if sys.version_info >= (3, 8):
- return nodes.Const(
- value=node.value,
- kind=node.kind,
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- return nodes.Const(
- node.value,
- node.lineno,
- node.col_offset,
- parent,
- node.kind,
- )
- # Not used in Python 3.8+.
- def visit_str(
- self, node: Union["ast.Str", "ast.Bytes"], parent: NodeNG
- ) -> nodes.Const:
- """visit a String/Bytes node by returning a fresh instance of Const"""
- return nodes.Const(
- node.s,
- node.lineno,
- node.col_offset,
- parent,
- )
- # Not used in Python 3.8+
- visit_bytes = visit_str
- # Not used in Python 3.8+.
- def visit_num(self, node: "ast.Num", parent: NodeNG) -> nodes.Const:
- """visit a Num node by returning a fresh instance of Const"""
- return nodes.Const(
- node.n,
- node.lineno,
- node.col_offset,
- parent,
- )
- def visit_pass(self, node: "ast.Pass", parent: NodeNG) -> nodes.Pass:
- """visit a Pass node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- return nodes.Pass(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- return nodes.Pass(node.lineno, node.col_offset, parent)
- def visit_raise(self, node: "ast.Raise", parent: NodeNG) -> nodes.Raise:
- """visit a Raise node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.Raise(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Raise(node.lineno, node.col_offset, parent)
- # no traceback; anyway it is not used in Pylint
- newnode.postinit(
- exc=self.visit(node.exc, newnode),
- cause=self.visit(node.cause, newnode),
- )
- return newnode
- def visit_return(self, node: "ast.Return", parent: NodeNG) -> nodes.Return:
- """visit a Return node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.Return(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Return(node.lineno, node.col_offset, parent)
- if node.value is not None:
- newnode.postinit(self.visit(node.value, newnode))
- return newnode
- def visit_set(self, node: "ast.Set", parent: NodeNG) -> nodes.Set:
- """visit a Set node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.Set(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Set(node.lineno, node.col_offset, parent)
- newnode.postinit([self.visit(child, newnode) for child in node.elts])
- return newnode
- def visit_setcomp(self, node: "ast.SetComp", parent: NodeNG) -> nodes.SetComp:
- """visit a SetComp node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.SetComp(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.SetComp(node.lineno, node.col_offset, parent)
- newnode.postinit(
- self.visit(node.elt, newnode),
- [self.visit(child, newnode) for child in node.generators],
- )
- return newnode
- def visit_slice(self, node: "ast.Slice", parent: nodes.Subscript) -> nodes.Slice:
- """visit a Slice node by returning a fresh instance of it"""
- if sys.version_info >= (3, 9):
- newnode = nodes.Slice(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Slice(parent=parent)
- newnode.postinit(
- lower=self.visit(node.lower, newnode),
- upper=self.visit(node.upper, newnode),
- step=self.visit(node.step, newnode),
- )
- return newnode
- def visit_subscript(self, node: "ast.Subscript", parent: NodeNG) -> nodes.Subscript:
- """visit a Subscript node by returning a fresh instance of it"""
- context = self._get_context(node)
- if sys.version_info >= (3, 8):
- newnode = nodes.Subscript(
- ctx=context,
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Subscript(
- ctx=context,
- lineno=node.lineno,
- col_offset=node.col_offset,
- parent=parent,
- )
- newnode.postinit(
- self.visit(node.value, newnode), self.visit(node.slice, newnode)
- )
- return newnode
- def visit_starred(self, node: "ast.Starred", parent: NodeNG) -> nodes.Starred:
- """visit a Starred node and return a new instance of it"""
- context = self._get_context(node)
- if sys.version_info >= (3, 8):
- newnode = nodes.Starred(
- ctx=context,
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Starred(
- ctx=context,
- lineno=node.lineno,
- col_offset=node.col_offset,
- parent=parent,
- )
- newnode.postinit(self.visit(node.value, newnode))
- return newnode
- def visit_tryexcept(self, node: "ast.Try", parent: NodeNG) -> nodes.TryExcept:
- """visit a TryExcept node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.TryExcept(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.TryExcept(node.lineno, node.col_offset, parent)
- newnode.postinit(
- [self.visit(child, newnode) for child in node.body],
- [self.visit(child, newnode) for child in node.handlers],
- [self.visit(child, newnode) for child in node.orelse],
- )
- return newnode
- def visit_try(
- self, node: "ast.Try", parent: NodeNG
- ) -> Union[nodes.TryExcept, nodes.TryFinally, None]:
- # python 3.3 introduce a new Try node replacing
- # TryFinally/TryExcept nodes
- if node.finalbody:
- if sys.version_info >= (3, 8):
- newnode = nodes.TryFinally(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.TryFinally(node.lineno, node.col_offset, parent)
- body: Union[List[nodes.TryExcept], List[NodeNG]]
- if node.handlers:
- body = [self.visit_tryexcept(node, newnode)]
- else:
- body = [self.visit(child, newnode) for child in node.body]
- newnode.postinit(body, [self.visit(n, newnode) for n in node.finalbody])
- return newnode
- if node.handlers:
- return self.visit_tryexcept(node, parent)
- return None
- def visit_tryfinally(self, node: "ast.Try", parent: NodeNG) -> nodes.TryFinally:
- """visit a TryFinally node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.TryFinally(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.TryFinally(node.lineno, node.col_offset, parent)
- newnode.postinit(
- [self.visit(child, newnode) for child in node.body],
- [self.visit(n, newnode) for n in node.finalbody],
- )
- return newnode
- def visit_tuple(self, node: "ast.Tuple", parent: NodeNG) -> nodes.Tuple:
- """visit a Tuple node by returning a fresh instance of it"""
- context = self._get_context(node)
- if sys.version_info >= (3, 8):
- newnode = nodes.Tuple(
- ctx=context,
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Tuple(
- ctx=context,
- lineno=node.lineno,
- col_offset=node.col_offset,
- parent=parent,
- )
- newnode.postinit([self.visit(child, newnode) for child in node.elts])
- return newnode
- def visit_unaryop(self, node: "ast.UnaryOp", parent: NodeNG) -> nodes.UnaryOp:
- """visit a UnaryOp node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.UnaryOp(
- op=self._parser_module.unary_op_classes[node.op.__class__],
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.UnaryOp(
- self._parser_module.unary_op_classes[node.op.__class__],
- node.lineno,
- node.col_offset,
- parent,
- )
- newnode.postinit(self.visit(node.operand, newnode))
- return newnode
- def visit_while(self, node: "ast.While", parent: NodeNG) -> nodes.While:
- """visit a While node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.While(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.While(node.lineno, node.col_offset, parent)
- newnode.postinit(
- self.visit(node.test, newnode),
- [self.visit(child, newnode) for child in node.body],
- [self.visit(child, newnode) for child in node.orelse],
- )
- return newnode
- @overload
- def _visit_with(
- self, cls: Type[nodes.With], node: "ast.With", parent: NodeNG
- ) -> nodes.With:
- ...
- @overload
- def _visit_with(
- self, cls: Type[nodes.AsyncWith], node: "ast.AsyncWith", parent: NodeNG
- ) -> nodes.AsyncWith:
- ...
- def _visit_with(
- self,
- cls: Type[T_With],
- node: Union["ast.With", "ast.AsyncWith"],
- parent: NodeNG,
- ) -> T_With:
- if sys.version_info >= (3, 8):
- newnode = cls(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = cls(node.lineno, node.col_offset, parent)
- def visit_child(child: "ast.withitem") -> Tuple[NodeNG, Optional[NodeNG]]:
- expr = self.visit(child.context_expr, newnode)
- var = self.visit(child.optional_vars, newnode)
- return expr, var
- type_annotation = self.check_type_comment(node, parent=newnode)
- newnode.postinit(
- items=[visit_child(child) for child in node.items],
- body=[self.visit(child, newnode) for child in node.body],
- type_annotation=type_annotation,
- )
- return newnode
- def visit_with(self, node: "ast.With", parent: NodeNG) -> NodeNG:
- return self._visit_with(nodes.With, node, parent)
- def visit_yield(self, node: "ast.Yield", parent: NodeNG) -> NodeNG:
- """visit a Yield node by returning a fresh instance of it"""
- if sys.version_info >= (3, 8):
- newnode = nodes.Yield(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.Yield(node.lineno, node.col_offset, parent)
- if node.value is not None:
- newnode.postinit(self.visit(node.value, newnode))
- return newnode
- def visit_yieldfrom(self, node: "ast.YieldFrom", parent: NodeNG) -> NodeNG:
- if sys.version_info >= (3, 8):
- newnode = nodes.YieldFrom(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- else:
- newnode = nodes.YieldFrom(node.lineno, node.col_offset, parent)
- if node.value is not None:
- newnode.postinit(self.visit(node.value, newnode))
- return newnode
- if sys.version_info >= (3, 10):
- def visit_match(self, node: "ast.Match", parent: NodeNG) -> nodes.Match:
- newnode = nodes.Match(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- newnode.postinit(
- subject=self.visit(node.subject, newnode),
- cases=[self.visit(case, newnode) for case in node.cases],
- )
- return newnode
- def visit_matchcase(
- self, node: "ast.match_case", parent: NodeNG
- ) -> nodes.MatchCase:
- newnode = nodes.MatchCase(parent=parent)
- newnode.postinit(
- pattern=self.visit(node.pattern, newnode),
- guard=self.visit(node.guard, newnode),
- body=[self.visit(child, newnode) for child in node.body],
- )
- return newnode
- def visit_matchvalue(
- self, node: "ast.MatchValue", parent: NodeNG
- ) -> nodes.MatchValue:
- newnode = nodes.MatchValue(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- newnode.postinit(value=self.visit(node.value, newnode))
- return newnode
- def visit_matchsingleton(
- self, node: "ast.MatchSingleton", parent: NodeNG
- ) -> nodes.MatchSingleton:
- return nodes.MatchSingleton(
- value=node.value, # type: ignore[arg-type] # See https://github.com/python/mypy/pull/10389
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- def visit_matchsequence(
- self, node: "ast.MatchSequence", parent: NodeNG
- ) -> nodes.MatchSequence:
- newnode = nodes.MatchSequence(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- newnode.postinit(
- patterns=[self.visit(pattern, newnode) for pattern in node.patterns]
- )
- return newnode
- def visit_matchmapping(
- self, node: "ast.MatchMapping", parent: NodeNG
- ) -> nodes.MatchMapping:
- newnode = nodes.MatchMapping(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- # Add AssignName node for 'node.name'
- # https://bugs.python.org/issue43994
- newnode.postinit(
- keys=[self.visit(child, newnode) for child in node.keys],
- patterns=[self.visit(pattern, newnode) for pattern in node.patterns],
- rest=self.visit_assignname(node, newnode, node.rest),
- )
- return newnode
- def visit_matchclass(
- self, node: "ast.MatchClass", parent: NodeNG
- ) -> nodes.MatchClass:
- newnode = nodes.MatchClass(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- newnode.postinit(
- cls=self.visit(node.cls, newnode),
- patterns=[self.visit(pattern, newnode) for pattern in node.patterns],
- kwd_attrs=node.kwd_attrs,
- kwd_patterns=[
- self.visit(pattern, newnode) for pattern in node.kwd_patterns
- ],
- )
- return newnode
- def visit_matchstar(
- self, node: "ast.MatchStar", parent: NodeNG
- ) -> nodes.MatchStar:
- newnode = nodes.MatchStar(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- # Add AssignName node for 'node.name'
- # https://bugs.python.org/issue43994
- newnode.postinit(name=self.visit_assignname(node, newnode, node.name))
- return newnode
- def visit_matchas(self, node: "ast.MatchAs", parent: NodeNG) -> nodes.MatchAs:
- newnode = nodes.MatchAs(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- # Add AssignName node for 'node.name'
- # https://bugs.python.org/issue43994
- newnode.postinit(
- pattern=self.visit(node.pattern, newnode),
- name=self.visit_assignname(node, newnode, node.name),
- )
- return newnode
- def visit_matchor(self, node: "ast.MatchOr", parent: NodeNG) -> nodes.MatchOr:
- newnode = nodes.MatchOr(
- lineno=node.lineno,
- col_offset=node.col_offset,
- end_lineno=node.end_lineno,
- end_col_offset=node.end_col_offset,
- parent=parent,
- )
- newnode.postinit(
- patterns=[self.visit(pattern, newnode) for pattern in node.patterns]
- )
- return newnode
|