U hY @szddlZddlZddlmZddlmZddlmZdZ da edddZ e dd d Z e e d d d ZddddZdS)N)OpenerDirector)warn) __version__)returncCs&tdkr"tjtjtjatS)N)openerurllibrequest build_opener HTTPHandler HTTPSHandlerr r @/tmp/pip-unpacked-wheel-e8onvpoz/albumentations/check_version.py get_opener src Cstjtjtj}d}zV|j|dd>}|jtkrh|}| d}| |W5QRWSW5QRXWn4t k r}zt d|ddW5d}~XYnXdS)Nz)https://pypi.org/pypi/albumentations/json)timeoutzutf-8zError fetching version info  stacklevel)rr r r r openstatusSUCCESS_HTML_CODEreadinfoget_content_charsetdecode Exceptionr)rurlresponsedataencodinger r rfetch_version_infos &$r")rrcCsB|r>z t|}|diddWStjk r<YdSXdS)z,Parses the version from the given JSON data.rversionr)jsonloadsgetJSONDecodeError)rZ json_datar r r parse_version"s r(c Csxz:t}t|}|r8|tkr8td|dtdtddWn8tk rr}ztd|dtddW5d}~XYnXdS)Nz.A new version of Albumentations is available: z (you have z). Upgrade using: pip install -U albumentations. To disable automatic update checks, set the environment variable NO_ALBUMENTATIONS_UPDATE to 1.rrz8Failed to check for updates due to an unexpected error: za. To disable automatic update checks, set the environment variable NO_ALBUMENTATIONS_UPDATE to 1.)r"r(current_versionr UserWarningr)rZlatest_versionr!r r rcheck_for_updates/s   r+)r$urllib.requestrrwarningsrZalbumentations._versionrr)rrrstrr"r(r+r r r rs