yIf2[ UdZddlmZddlZddlmZddlmZddlm Z ddl Z ddl m Z ddl m Z dd lmZmZdd lmZe rdd lmZd d dddddddddd Zded<ejdZeddhZdZdZdZd Zegd!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjdkeeeRZddpZ ddtZ!ddvZ"ddzZ#dd|Z$Gd}d~e Z%ddZ&dS)zCheck for signs of poor design.) annotationsN) defaultdict)Iterator) TYPE_CHECKING)nodes) BaseChecker)is_enumonly_required_for_messages)MessageDefinitionTuple)PyLinter)zToo many ancestors (%s/%s)too-many-ancestorsznUsed when class has too many parent classes, try to reduce this to get a simpler (and so easier to use) class.)z$Too many instance attributes (%s/%s)too-many-instance-attributeszsUsed when class has too many instance attributes, try to reduce this to get a simpler (and so easier to use) class.)zToo few public methods (%s/%s)too-few-public-methodszLUsed when class has too few public methods, so be sure it's really worth it.)zToo many public methods (%s/%s)too-many-public-methodsznUsed when class has too many public methods, try to reduce this to get a simpler (and so easier to use) class.)z"Too many return statements (%s/%s)too-many-return-statementszWUsed when a function or method has too many return statement, making it hard to follow.)zToo many branches (%s/%s)too-many-brancheszOUsed when a function or method has too many branches, making it hard to follow.)zToo many arguments (%s/%s)too-many-argumentsz8Used when a function or method takes too many arguments.)z Too many local variables (%s/%s)too-many-localszNN$$ 8   44 >>     '   44   ?udiikk())K_*   i . . '!I)glG4E%FGG   i . . &>DD%D ) ) )  $ $%; < < *;..44 5bool_op nodes.BoolOpintcd}|D]4}t|tjr|t |z }/|dz }5|S)zCounts the number of boolean expressions in BoolOp `bool_op` (recursive). example: a and (b or c or (d and e)) ==> 5 boolean expressions r) get_childrenr4r5BoolOp_count_boolean_expressions)rE nb_bool_expr bool_exprs rBrLrLs_ L))++ i 0 0  6yAA ALL A LL rDnodes.ClassDefctd|D}|D]1}t|jr|jdkr|dz }2|S)Nc3NK|] }|jddV!dS_rINr: startswith.0methods rB z*_count_methods_in_class..s8XXFFK!! <<>>_ , ,   LLL KK      f..e.<< = = =  > > > > >rDset[nodes.ClassDef]c<tt||SN)r1rn)r&rbs rB _get_parentsrrs  77 8 88rDc,eZdZdZdZeZddddddfd d ddd dfd d ddddfddddddfddddddfddddddfddddddfddddddfd d!ddd"dfd#d$ddd%dfd&dddd'dfd(gd)d*d+dff ZdUfd0 ZdVd1Z dWd3Z e d4d5d6d7dXd:Z e d6d7dXd;Z e dd?d@dAdYdCZeZe dd?d@dYdDZeZdZdGZd[dIZd\dKZe dLd=d]dNZd]dOZd^dQZeZd_d`dTZxZS)aMisdesignCheckerzChecker of potential misdesigns. Checks for sign of poor/misdesign: * number of methods, attributes, local variables... * size, complexity of functions, methods designzmax-argsrGzz2Maximum number of arguments for function / method.)defaulttypemetavarhelpz max-localsz4Maximum number of locals for function / method body.z max-returnszz2Maximum number of parents for a class (see R0901).zignored-parentscsvz%zOList of qualified class names to ignore when counting class parents (see R0901)zmax-attributesz5Maximum number of attributes for a class (see R0902).zmin-public-methodsz9Minimum number of public methods for a class (see R0903).zmax-public-methodsz9Maximum number of public methods for a class (see R0904).z max-bool-exprzEMaximum number of boolean expressions in an if statement (see R0916).zexclude-too-few-public-methods regexp_csvz[,...]zfList of regular expressions of class ancestor names to ignore when counting public methods (see R0903)linterr r(NonecVt||||dSrq)superrZ)selfr __class__s rBrZzMisdesignChecker.__init__s3       @ rDc|jjg|_t t |_g|_|jjj |_ dS)zInitialize visit variables.N) rstatsreset_node_count_returnsrrG _branches_stmtsconfigexclude_too_few_public_methods_exclude_too_few_public_methods)rs rBopenzMisdesignChecker.opensO **,,, $S)) K  = ,,,rDamountcdt|jD]\}}|j|xx|z cc<dSrq) enumerater)rrirSs rB_inc_all_stmtszMisdesignChecker._inc_all_stmtssCdk** % %DAq KNNNf $NNNN % %rDr rrrr&rOc6t|t|jjj}t |}||jjjkr)|d|||jjjf| fd|j D}t ||jjj kr8|d|t ||jjj fdSdS)zFCheck size of inheritance hierarchy and number of instance attributes.r r&argscRg|]#\}}|du!|$S)r)r2)rWkvr2s rB z3MisdesignChecker.visit_classdef..s:   1a1Q499;;$;N;NA;N;N;NrDrN) rrSTDLIB_CLASSES_IGNORE_ANCESTORunionrrrblen max_parents add_messager2instance_attrsitemsmax_attributes)rr&rk nb_parentsfiltered_attrsr2s @rBvisit_classdefzMisdesignChecker.visit_classdefs4  * 0 01C1S T T  \\  *6 6 6   $ $+"4"@A    yy{{     /5577    ~  !3!B B B   ..))4;+=+LM       C BrDcHtd|D}||jjjkr)|d|||jjjf|jdkrA|jr:|D]%tfd|jDrdS&|jdkst|rdSt|}||jjj kr+|d|||jjj fdSdS)zCheck number of public methods.c3NK|] }|jddV!dSrRrTrVs rBrYz2MisdesignChecker.leave_classdef..sK  FK4J4J34O4O       rDrrclassc3fK|]+}|V,dSrq)matchr,)rWpatternr?s rBrYz2MisdesignChecker.leave_classdef..sMMM(.."2"233rDNr) r[r]rrmax_public_methodsrrxrr+anyrCramin_public_methods)rr& my_methodsr`r?s @rBleave_classdefzMisdesignChecker.leave_classdefs   NN,,       *= = =   ) $+"4"GH     9  D$H  NN,,  #'#GFF  9  #A$#G#G  F .d33 +> > >   (!4;#5#HI       ? >rDrrrrrzkeyword-arg-before-varargnodes.FunctionDefc|jd|jj|jjz|jjz}|jjj|}d}rtfd|D}t||z }||jjj kr6| d|t||jjj fnd}t|j |z }d|j vr|dz}||jjj kr)| d|||jjj f|jddS) zeCheck function name, docstring, arguments, redefinition, variable names, max locals. rNc3PK|] }|jdV!dS)rIN)rr:)rWargignored_argument_namess rBrYz5MisdesignChecker.visit_functiondef..sM'')?)E)Ech)O)O'''''''rDrrrSrIr)rappendr posonlyargs kwonlyargsrrrr[rmax_argsrr3 max_localsr)rr&rignored_args_numargnumlocnumrs @rBvisit_functiondefz"MisdesignChecker.visit_functiondefs Qy~ 55 8LL!%!3!J   % #&''''!%'''$$ YY!11F *333  (d))T[%7%@A! ! T[!!$44 $+   aKF DK&1 1 1   !dk0;<     1rDc|j}||jjjkr)|d|||jjjf|j|}||jjjkr)|d|||jjjf|j}||jjj kr+|d|||jjj fdSdS)zlMost of the work is done here on close: checks for max returns, branch, return in __init__. rrrrN) rrhrr max_returnsrr max_branchesrmax_statements)rr&returnsbranchesstmtss rBleave_functiondefz"MisdesignChecker.leave_functiondef9s-##%% T['3 3 3   ,t{1=>     >$' dk(5 5 5   # 2 ?@      !! 4;%4 4 4   %T[/>?       5 4rDrS nodes.ReturncB|jsdS|jdxxdz cc<dS)zCount number of returns.NrI)r)rrSs rB visit_returnzMisdesignChecker.visit_return]s5}  F bQrD nodes.NodeNGcB|jr|ddSdS)zXDefault visit method -> increments the statements counter if necessary. rIN) is_statementr)rr&s rB visit_defaultzMisdesignChecker.visit_defaultcs3   #    " " " " " # #rD nodes.Tryct|j}|jr|dz }|jr|dz }|||||dS Increments the branches counter.rIN)rhandlersorelse finalbody _inc_branchrrr&rs rB visit_tryzMisdesignChecker.visit_tryjsgt}%% ;  MH >  MH x((( H%%%%%rDrnodes.Ifc||d}|jrBt|jdks%t|jdtjs|dz }|||||dS)z?Increments the branches counter and checks boolean expressions.rIrN)_check_boolean_expressionsrrr4r5Ifrrrs rBvisit_ifzMisdesignChecker.visit_ifts ''--- ;    q 4;q>7:(N(N MH x((( H%%%%%rDc|j}t|tjsdSt |}||jjjkr+|d|||jjjfdSdS)zvGo through "if" node `node` and count its boolean expressions if the 'if' node test is a BoolOp node. Nrr) testr4r5rKrLrr max_bool_exprr)rr& conditionrMs rBrz+MisdesignChecker._check_boolean_expressionssI )W^44  F1)<< $+,: : :   ."DK$6$DE       ; :rD nodes.WhilecNd}|jr|dz }|||dSr)rrrs rB visit_whilezMisdesignChecker.visit_whiles6 ;  MH x(((((rDrI branchesnumcT|j|xx|z cc<dS)rN)rscope)rr&rs rBrzMisdesignChecker._inc_branchs. tzz||$$$ 3$$$$$rDrr r(r)r(r)rrGr(r)r&rOr(r)r&rr(r)rSrr(r)r&rr(r)r&rr(r)r&rr(r)r&rr(r)rI)r&rrrGr(r)__name__ __module__ __qualname____doc__r:r"msgsoptionsrZrrr rrrvisit_asyncfunctiondefrleave_asyncfunctiondefrrrrrr visit_forr __classcell__)rs@rBrtrts D D "L     "N     "     "N     "Q     "L     Bi     "     !"     !"     ")     -$5E    SsGj     %%%% & !    <  8:STT+++UT+Z $# ''''R/$ 6/ ####&&&&  >@STT & & &UT &    ))))I444444444rDrtrr rcJ|t|dSrq)register_checkerrt)rs rBregisterrs% ,V4455555rD)r&r'r(r))rErFr(rG)r&rOr(rG)r&rOrbrcr(rd)r&rOrbrcr(ror)'r __future__rre collectionsrcollections.abcrtypingrr5rpylint.checkersrpylint.checkers.utilsr r pylint.typingr pylint.lintr r"__annotations__compiler^ frozensetr<r>r-r.r/rrCrLrarnrrrtrrrDrBrs &%%"""""" ######$$$$$$ ''''''EEEEEEEE000000%$$$$$$           uAACCCCHbj+,, "K#9:: '%;"+OOO O  O  O  O O O "O O O !O "O O !O %!O" %#O$ )%O& )'O( *)O* $+O, $-O. $/O0 %1O2 &3O4 !5O6 %7O8 &9O: ;O< &=O> #?O@ *AOB 'COD $EOF %GOH &IOJ $KOL +MON &OOP QOR SOT UOV WOX YOZ [O\ ]O^ _O` aOb cOd eOf gOh iOj kOl mOn oOp qOr sOt uOv wOx yOz {O| }O~ O@ AOB COD !EOF GOH IOJ KOL MON OOP QOR %SOT UOV WOX YOZ [O\ $]OOQ"Q"h!!!!H    !>!>!>!>H9999 }4}4}4}4}4{}4}4}4@ 666666rD