o h@sddZddlmZddlmZddlmZddlmZGdddZGdd d eZ Gd d d Z d S) z&Statistic collection logic for Flake8.) annotations) Generator) NamedTuple) Violationc@s<eZdZdZdddZdddZdd d Z ddddZd S) Statisticsz5Manager of aggregated statistics for a run of Flake8.returnNonecCs i|_dS)z8Initialize the underlying dictionary for our statistics.N)_storeselfr G/var/www/html/ai/venv/lib/python3.10/site-packages/flake8/statistics.py__init__ s zStatistics.__init__ list[str]cCstdd|jDS)ziReturn all unique error codes stored. :returns: Sorted list of error codes. cSsh|]}|jqSr )code.0keyr r r sz)Statistics.error_codes..sortedr r r r r error_codesszStatistics.error_codeserrorrcCs6t|}||jvrt||j|<|j|dS)zAdd the fact that the error was seen in the file. :param error: The Violation instance containing the information about the violation. N)Key create_fromr Statistic increment)r rrr r r records  zStatistics.recordNprefixstrfilename str | None Generator[Statistic, None, None]c#s6tfdd|jD}|D]}|j|VqdS)aBGenerate statistics for the prefix and filename. If you have a :class:`Statistics` object that has recorded errors, you can generate the statistics for a prefix (e.g., ``E``, ``E1``, ``W50``, ``W503``) with the optional filter of a filename as well. .. code-block:: python >>> stats = Statistics() >>> stats.statistics_for('E12', filename='src/flake8/statistics.py') >>> stats.statistics_for('W') :param prefix: The error class or specific error code to find statistics for. :param filename: (Optional) The filename to further filter results by. :returns: Generator of instances of :class:`Statistic` c3s |] }|r|VqdSN)matchesrr rr r >s   z,Statistics.statistics_for..Nr)r rr matching_errors error_coder r%r statistics_for%szStatistics.statistics_forrr)rr)rrrrr#)rrr r!rr")__name__ __module__ __qualname____doc__rrrr)r r r r r s    rc@s:eZdZUdZded<ded<eddd Zdd dZdS)rzSimple key structure for the Statistics dictionary. To make things clearer, easier to read, and more understandable, we use a namedtuple here for all Keys in the underlying dictionary for the Statistics object. rr rrrrcCs||j|jdS)z6Create a Key from :class:`flake8.violation.Violation`.r rr/clsrr r r rPszKey.create_fromrr!boolcCs|j|o|dup|j|kS)aDetermine if this key matches some constraints. :param prefix: The error code prefix that this key's error code should start with. :param filename: The filename that we potentially want to match on. This can be None to only match on error prefix. :returns: True if the Key's code starts with the prefix and either filename is None, or the Key's filename matches the value passed in. N)r startswithr )r rr r r r r$Us z Key.matchesN)rrrr)rrr r!rr2)r+r,r-r.__annotations__ classmethodrr$r r r r rEs  rc@s2eZdZdZdd d ZedddZdddZdS)rzSimple wrapper around the logic of each statistic. Instead of maintaining a simple but potentially hard to reason about tuple, we create a class which has attributes and a couple convenience methods on it. r(rr messagecountintrrcCs||_||_||_||_dS)zInitialize our Statistic.Nr(r r6r7)r r(r r6r7r r r rns zStatistic.__init__rrcCs||j|j|jddS)z>Create a Statistic from a :class:`flake8.violation.Violation`.rr9)rr textr0r r r rws zStatistic.create_fromcCs|jd7_dS)zAIncrement the number of times we've seen this error in this file.N)r7r r r r rszStatistic.incrementN) r(rr rr6rr7r8rr)rrrrr*)r+r,r-r.rr5rrr r r r rfs    rN) r. __future__rtypingrrflake8.violationrrrrr r r r s    ;!