rjU,UdZddlmZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZd ZdddddddZ eZded< iZded<ej*dZdZGdde ZGddZy)aFormat of the call count file used by ``test/aaa_profiling/``. ``test/profiles.txt`` stores the expected function call counts asserted by the profiling suite. Each line is keyed to a "platform key" that encodes the machine architecture, operating system, Python implementation and version, the database / DBAPI in use, and whether or not the C extensions were built: .. sourcecode:: text test.aaa_profiling.test_orm.MergeTest.test_merge_load x86_64_linux_cpython_3.14_sqlite_pysqlite_dbapiunicode_cextensions 1234 The key covers anything that moves the counts, which includes compiled code that isn't SQLAlchemy's own: a DBAPI that ships both a pure Python and a C accelerated build carries a ``pydbapi`` / ``cdbapi`` token naming which one was loaded, since the difference between them can be tens of thousands of calls. A call count is only asserted when the running environment matches a key that's present in the file; otherwise the test is skipped. Two different things read and write this file: :mod:`.testing.profiling`, which records counts as the suite runs, and ``tools/profiles.py``, which maintains the file from the outside. Both use this module so that the format is spelled out in exactly one place. .. note:: ``tools/profiles.py`` loads this module directly from the source tree, without importing the ``sqlalchemy`` package, so that the maintenance tooling and the noxfile that drives it work in an environment where SQLAlchemy itself isn't installed. Keep the imports here to the standard library only. ) annotationsN)Dict)KeysView)List) NamedTuple)Optional)Set)Tuple) TypeAlias)Uniona$# SQLAlchemy call count profiles. # This file is written out on a per-environment basis. # For each test in aaa_profiling, the corresponding function and # environment is located within this file. If it doesn't exist, # the test is skipped. # If a callcount does exist, it is compared to what we received. # assertions are raised if the counts do not match. # # To add a new callcount test, apply the function_call_count # decorator and re-run the tests using the --write-profiles # option - this file will be rewritten including the new count. # sqlite postgresqlmysqloraclemssql)r rrmariadbrrr TestKeyzDict[PlatformKey, List[int]] _EMPTY_COUNTSz ^\d+\.\d+t?$ dbapiunicodeceZdZUdZded< ded< ded< ded< ded< ded< d ed < d ed < dd ZeddZeddZ eddZ y) PlatformKeyz8The environment half of a line in ``test/profiles.txt``.strmachinesystemimplementationpythondialectdriverzTuple[str, ...] dbapi_flagsbool cextensionscdj|j|j|j|j|j |j g|jt|jS)N_) joinrrrrrrr_UNICODE_TOKEN cext_tokenselfs k/var/www/html/python-recuperauto-other/env/lib/python3.12/site-packages/sqlalchemy/testing/profiles_file.py__str__zPlatformKey.__str__slxx  ##   !!     c ,|jd}|ddvrtd||jdk(}|dtk7rtdtd||jt |D]\}}t j |sntd||d kst||d zkrtd ||dj|d |d z ||d z ||d z ||||d z||d zt||d zd |S)zParse a platform key string. Raises ``ValueError`` if the key isn't in the format written by :meth:`.PlatformKey.__str__`. r#)r! nocextensionsz"unrecognized cextensions token in r!z unrecognized z token in zno python version found in zcan't parse platform key N)rrrrrrrr!) split ValueErrorpopr% enumerate _PY_VERSIONmatchlenr$tuple)clskeytokensr!idxtokens r)parsezPlatformKey.parsesE3 ":= =A#IJ Jjjlm3 ": '}^, ))Q 7 > zz|ts304 -Hgv#.#4#4W#=L;A<<;L"M%3u:"MK"M EP %%h3LA7= X|34 > > >   #N!$99+QvhnTHBrdK > >sSD 7Ez%ProfilesFile.write..s&',E &  N)r^rPwriteHEADERsortedrQrr$)r(rdrgper_fnrjris r)rvzProfilesFile.writes$))S ! Y OOF #"4::. H- 2X =>$*6s$;L XX&06|0D&FOO(0,G     s BB??Cc6|jjS)z"All test keys present in the file.)rQkeysr's r) test_keyszProfilesFile.test_keysszz  r+c^|jj|tjS)z(The platform keys recorded for one test.)rQrBrr{r(rgs r) platforms_forzProfilesFile.platforms_fors"zz~~h 6;;==r+c`|jj|tj|S)zThe call counts recorded for a test on a platform, or ``None``. The list returned is the one stored here; mutating it in place updates what :meth:`.ProfilesFile.write` will emit. )rQrBrr(rgrjs r) counts_forzProfilesFile.counts_for"s$zz~~h 6::<HHr+cZ|jj|ij|gS)zLike :meth:`.ProfilesFile.counts_for`, adding an empty entry for the test and platform if there isn't one yet. )rQrcrs r)setdefault_countszProfilesFile.setdefault_counts.s(zz$$Xr2==lBOOr+c|jj|}|y|j|d|s|j|=yy)zmDrop one test / platform entry. The test itself is dropped once its last platform is gone. N)rQrBr4)r(rgrjrys r) remove_entryzProfilesFile.remove_entry8s@) >  <& 8$r+cLt|jj|iS)z=Drop every entry for one test, returning how many there were.)r8rQr4r~s r) remove_testzProfilesFile.remove_testHs4::>>(B/00r+cp|jjDchc] }|D]}| c}}Scc}}w)z/All platform keys present anywhere in the file.)rQvalues)r(ryrjs r) platform_keyszProfilesFile.platform_keysMsF **++-  &      s2cVtd|jjDS)zTotal number of recorded lines.c32K|]}t|ywrT)r8)rqrys r)rrz+ProfilesFile.entry_count..YsA63v;Ars)sumrQrr's r) entry_countzProfilesFile.entry_countVs"ATZZ->->-@AAAr+c<|jj||fSrT)rRrBrs r) lineno_forzProfilesFile.lineno_for[s||< 899r+N)rPzUnion[str, os.PathLike[str]])rENone)rEzKeysView[TestKey])rgrrEzKeysView[PlatformKey])rgrrjrrEzOptional[List[int]])rgrrjrrEz List[int])rgrrjrrEr)rgrrErb)rEzSet[PlatformKey])rErb)rgrrjrrEz Optional[int])rFrGrHrIrJrXrWrvr|rrrrrrrrrMr+r)rOrOs I 662 33D !>F ! > I I/: I  IPP/:P P%%/:% % 1  B ::/:: :r+rO)rI __future__rrUretypingrrrrrr r r r rwrArrrJrcompiler6r%rrOrMr+r)rs#J#      /1 +0bjj) aF*aFHR:R:r+