File: //usr/local/lib/python3.10/test/test_zoneinfo/data/__pycache__/update_test_data.cpython-310.pyc
o
�i^ � @ s� d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl Z ddl
Z
g d�Z e� e
�jZejdd�d dd ��Zd d d �Zd!dd�Zd"dd�Zd#dd�Zd$d%d d �Zed kr`e� dS dS )&a�
Script to automatically generate a JSON file containing time zone information.
This is done to allow "pinning" a small subset of the tzdata in the tests,
since we are testing properties of a file that may be subject to change. For
example, the behavior in the far future of any given zone is likely to change,
but "does this give the right answer for this file in 2040" is still an
important property to test.
This must be run from a computer with zoneinfo data installed.
� )� annotationsN) zAfrica/AbidjanzAfrica/CasablancazAmerica/Los_AngeleszAmerica/Santiagoz
Asia/TokyozAustralia/Sydneyz
Europe/Dublinz
Europe/Lisbonz
Europe/LondonzPacific/Kiritimati�UTC)�maxsize�return� pathlib.Pathc C s4 d} t tjtj�D ] }|| �� r| S q td��)zEGet the first zoneinfo directory on TZPATH containing the "UTC" zone.r zCannot find time zone data.)�map�pathlib�Path�zoneinfoZTZPATH�exists�OSError)�key�path� r �E/usr/local/lib/python3.10/test/test_zoneinfo/data/update_test_data.py�get_zoneinfo_path) s �r �typing.Dict[str, str]c C s� t � } | d }|�� std��t|d�� }t|�}W d � n1 s#w Y |�� �dd�\}}|dd� �� rHt|�dk sH|dd � � � sQt
d d
|� � ��d |iS ) Nz tzdata.ziz*Time zone data does not include tzdata.zi.�r� � r � � zJVersion string should be YYYYx, where YYYY is the year and x is a letter; zfound: �version) r r r �open�next�strip�rsplit�isdigit�len�isalpha�
ValueError)r Z tzdata_zi�fZ version_line�_r r r r �get_zoneinfo_metadata4 s&
�� ���r# r
�str�bytesc C s@ t � }t|| d�� }|�� W d � S 1 sw Y d S )N�rb)r r �read)r
r r! r r r � get_zoneinfoO s $�r( �data�typing.List[str]c C s0 t �| �}t�|�}|�d�}tj|dd�}|S )Nzutf-8�F )�width)�lzma�compress�base64Z b85encode�decode�textwrap�wrap)r) Zcompressed_zone�rawZ raw_data_strZdata_strr r r �encode_compressedV s
r4 �"typing.Dict[str, typing.List[str]]c C s dd� t D �} | S )Nc S s i | ] }|t t|���qS r )r4 r( )�.0r
r r r �
<dictcomp>a s z(load_compressed_keys.<locals>.<dictcomp>)�KEYS)�outputr r r �load_compressed_keys` s r: �zoneinfo_data.json�fname�Nonec C sx t jddd� tddd�}t� }t� }||d�}tt | d��}tj||fi |�� W d � d S 1 s5w Y d S )NT)�exist_ok�parents� )�indentZ sort_keys)�metadatar) �w)�
TEST_DATA_LOC�mkdir�dictr: r# r �json�dump)r< Z json_kwargsZcompressed_keysrB r9 r! r r r �update_test_dataf s ��"�rI �__main__)r r )r r )r
r$ r r% )r) r% r r* )r r5 )r; )r< r$ r r= )�__doc__Z
__future__r r/ � functoolsrG r- r r1 �typingr
r8 r �__file__�parentrD � lru_cacher r# r( r4 r: rI �__name__r r r r �<module> s,
�