o @a'@sdZddlmZddlmZddlmZmZmZm Z ddl m Z ddl Z ddlZddlZedZGdd d eZGd d d e jZd d ZddZddZdS)z This module performs a few early syntax check on the input AST. It checks the conformance of the input code to Pythran specific constraints. )MODULES)Class)TupleListSetDict)isstrNZpythranc@seZdZdddZddZdS)PythranSyntaxErrorNcCs6t|||rt|dd|_|j|_|j|_dSdS)Nfilename) SyntaxError__init__getattrr lineno col_offsetoffset)selfmsgnoder0/usr/lib/python3/dist-packages/pythran/syntax.pyr s  zPythranSyntaxError.__init__cCs|jr<|jr<|jrrz---- z ---- ) r rropenrangereadlineformatrstripargs)rfiZextrarrrr__str__s(    zPythranSyntaxError.__str__N)__name__ __module__ __qualname__r r!rrrrr s  r c@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*S)+ SyntaxCheckerz Visit an AST and raise a PythranSyntaxError upon unsupported construct. Attributes ---------- attributes : {str} Possible attributes from Pythonic modules/submodules. cs0t_fddtD]}|qdS)z) Gather attributes from MODULES content. csJj||D]}t|tr|q t|tr"|jq dS)z< Recursively save Pythonic keywords as possible attributes. N) attributesupdatekeysvalues isinstancedictrZfields)module signaturesave_attributerrrr0As     z.SyntaxChecker.__init__..save_attributeN)setr'rr*)rr-rr/rr =s  zSyntaxChecker.__init__cCs^d}tjtjtjtjf}|jD]}t|tjrt|j rqt||r#qt ||| |dS)NzUTop level statements can only be assignments, strings,functions, comments, or imports) ast FunctionDefZImportZ ImportFromZAssignbodyr+ZExprrvaluer generic_visit)rrerrZ WhiteListnrrr visit_ModuleMs   zSyntaxChecker.visit_ModulecC td|)Nz!Interactive session not supportedr rrrrrvisit_InteractiveY zSyntaxChecker.visit_InteractivecCr:)Nz%Interactive expressions not supportedr;r<rrrvisit_Expression\r>zSyntaxChecker.visit_ExpressioncCr:)Nz/Suites are specific to Jython and not supportedr;r<rrr visit_Suite_szSyntaxChecker.visit_SuitecCstd)NzClasses not supportedr;)r_rrrvisit_ClassDefcszSyntaxChecker.visit_ClassDefcCs |||jrtd|jdS)Nz+Printing to a specific stream not supported)r6destr r<rrr visit_Printfs zSyntaxChecker.visit_PrintcCr:)NzWith statements not supportedr;r<rrr visit_Withlr>zSyntaxChecker.visit_WithcCr:)Nz&Call with star arguments not supportedr;r<rrr visit_StarredoszSyntaxChecker.visit_StarredcCs|jdur td|dS)NzCall with kwargs not supported)argr r<rrr visit_keywordss  zSyntaxChecker.visit_keywordcCs||dSr")r6r<rrr visit_CallwszSyntaxChecker.visit_CallcCst|jturt|dr|g}ng}tdg|Rtt}t|jtr6|j|jkr0|j ks8ntd|dSdS)NrzEllipsis are not supportedzlarge int not supported) r5Ellipsishasattrr npiinfointr+minmax)rrrrMrrrvisit_Constantzs    zSyntaxChecker.visit_ConstantcCsB|jrtd||jjrtd||jjrtd|||dS)Nzdecorators not supportedzVarargs not supportedzKeyword arguments not supported)decorator_listr rvarargkwargr6r<rrrvisit_FunctionDefs  zSyntaxChecker.visit_FunctionDefcCs|||jr td|dS)Nz'Cause in raise statements not supported)r6causer r<rrr visit_Raises zSyntaxChecker.visit_RaisecCs,|||j|jvrtd|j|dS)NzAttribute '{0}' unknown)r6attrr'r rr<rrrvisit_Attributes   zSyntaxChecker.visit_AttributecCr:)Nz@named expression are not supported yet, please open an issue :-)r;r<rrrvisit_NamedExprszSyntaxChecker.visit_NamedExprcCsH|jD]}t}|jdD]}||vrtd|j|||}q qdS)z- Check if imported module exists in MODULES. .Module '{0}' unknown.N)namesrnamesplitr r)rraliascurrent_modulepathrrr visit_Imports   zSyntaxChecker.visit_ImportcCs|jrtd||jstd||j}t}|dD]}||vr(td||||}q|jD]}|jdkr8q0|j|vrGtd|j||q0dS)a, Check validity of imported functions. Check: - no level specific value are provided. - a module is provided - module/submodule exists in MODULES - imported function exists in the given module/submodule zRelative import not supportedzimport from without moduler[r\*z*identifier '{0}' not found in module '{1}'N)levelr r-rr_rr]r^)rrr-rarbr`rrrvisit_ImportFroms4      zSyntaxChecker.visit_ImportFromcCr:)Nz#'exec' statements are not supportedr;r<rrr visit_Execr>zSyntaxChecker.visit_ExeccCr:)Nz%'global' statements are not supportedr;r<rrr visit_Globalr>zSyntaxChecker.visit_GlobalN)r#r$r%__doc__r r9r=r?r@rBrDrErFrHrIrQrUrWrYrZrcrfrgrhrrrrr&2s,     $ r&cCst|dS)z7Does nothing but raising PythranSyntaxError when neededN)r&Zvisit)rrrr check_syntaxsrjc sddlm}m}mddlm}m}m}|jD]@\}}||} |D]5} |fdd| D|} z ||| || Wq#|yXt d || d t t | jdd wqdS) zh Does nothing but raising PythranSyntaxError if specs are incompatible with the actual code r)unifyclonetr)Function TypeVariableInferenceErrorcsg|]}|qSrr).0prmrr szcheck_specs..z\Specification for `{}` does not match inferred type: expected `{}` got `Callable[[{}], ...]`z, N)Zpythran.types.togrkrlrmrnrorp functionsitemsr rjoinmapstrtypes) specsr{rkrlrnrorpfname signaturesZftyper.Zsig_typerrsr check_specss( rc Cs ddlm}dd|jD}|||}|jD]i\}}z||}Wnty1td|w||} |D]I} t |j j } t | | krMtd|t | | t |j j kr`td|t | D]\} } | | rt | ttttfrtd|| jjqdq8qd S) za Does nothing but raising PythranSyntaxError if specs references an undefined global r)ArgumentEffectscSs i|] }t|tjr|j|qSr)r+r2r3r^)rqrrrr s z!check_exports..z/Invalid spec: exporting undefined function `{}`z&Too many arguments when exporting `{}`z(Not enough arguments when exporting `{}`zvExporting function '{}' that modifies its {} argument. Beware that this argument won't be modified at Python call siteN)Z!pythran.analyses.argument_effectsrr4ZgatherrvrwKeyErrorr rlenrdefaults enumerater+rrrrloggerZwarning __class__r%)Zpmmodr|rZ mod_functionsZargument_effectsr}r~ZfnodeZaer.Z args_countrZtyrrr check_exportssL      r)riZpythran.tablesrZpythran.intrinsicrZpythran.typingrrrrZ pythran.utilsrZgastr2ZloggingZnumpyrLZ getLoggerrr r Z NodeVisitorr&rjrrrrrrs    *