🔐 Sid Gifari File Manager Pro
v8.0.5 | 2026-08-08 08:01:55 | PHP 8.2.32
📂
/ (Root)
/
opt
/
alt
/
python38
/
lib
/
python3.8
/
site-packages
/
influxdb_client
/
client
/
write
/
__pycache__
📍 /opt/alt/python38/lib/python3.8/site-packages/influxdb_client/client/write/__pycache__
🔄 Refresh
✏️
Editing: dataframe_serializer.cpython-38.pyc
Read Only
U v�ii7 � @ sp d Z ddlZddlZddlZddlmZ ddlmZmZm Z m Z e� d�Z dd� Z G dd � d �Ze fd d �ZdS ) z� Functions for serialize Pandas DataFrame. Much of the code here is inspired by that in the aioinflux packet found here: https://github.com/gusutabopb/aioinflux � N)�WritePrecision)� _ESCAPE_KEY�_ESCAPE_STRING�_ESCAPE_MEASUREMENT�DEFAULT_WRITE_PRECISIONz1influxdb_client.client.write.dataframe_serializerc s, � fdd�t t� j��D �}t� jf|�� S )Nc s g | ]}� j d d �|f �qS �N)�iloc)�.0�k�� data_frame� �b/opt/alt/python38/lib/python3.8/site-packages/influxdb_client/client/write/dataframe_serializer.py� <listcomp> s z _itertuples.<locals>.<listcomp>)�range�len�columns�zip�index)r �colsr r r � _itertuples s r c @ s> e Zd ZdZedfedd�dd�Zd ed�dd�Zd d � ZdS ) �DataframeSerializerz'Serialize DataFrame into LineProtocols.N)� chunk_size�returnc # s� ddl m� m} t|� j�s.td�t|����|�d�}|dkrHtd��|�dd�}|�dd�} |j d d �}|dkrz|j n|| } t| � j �r�| � � } n� j | |d �} | r�t| � j�r�| �| �} n | j�| �} t| d �r�| jdkr�| �d �} |dkr�| |_ n| ||< |�d�} t| �pg �} g } g } g }g }|j�rh|j�� D ](\}}||jk�r>|||< | �|� �q>tt|j�� �dd� d�}|�� � � }d}|D �]�\}\}}t |�}dt | �� d�}| � |� t!�� |d }d|� d�}|| k�r8|j"| �rd|� d|� d|� d|� d � }nd |� d|� d �}| � |� �q�n |dk �rT||k�rT|}�q�t |�dk�rfd nd }t#|j|j$��s�t#|j|j%��s�t#|j|j&��r�t#|j|j$��r�d nd }|j"| �r�d!|� d"|� |� d#|� d$|� d%� } n|� |� d#|� d$|� �} n>|j"| �r&d&|� d'|� |� d(|� d)�} n|� |� d(|� d*�} |� |� |� | � �q�t |�� t'�} d �(| �} d �(|�}d+| } |t)j*k�r�d,| } n*|t)j+k�r�d-| } n|t)j,k�r�d.| } t-d/| � d0|� d0| � d1�| t!t.| � d2��} t/|j��� D ]&\} }!| | k�r�|�0| d i|j1�}�q� fd3d4�}"|| _2| | _3|| _4|j"|d d | _5|"| _6|dk �r�t7t8�9t |�t:|� ��| _;|| _<nd| _;dS )5aR Init serializer. :param data_frame: Pandas DataFrame to serialize :param point_settings: Default Tags :param precision: The precision for the unix timestamps within the body line-protocol. :param chunk_size: The size of chunk for serializing into chunks. :key data_frame_measurement_name: name of measurement for writing Pandas DataFrame :key data_frame_tag_columns: list of DataFrame columns which are tags, rest columns will be fields :key data_frame_timestamp_column: name of DataFrame column which contains a timestamp. The column can be defined as a :class:`~str` value formatted as `2018-10-26`, `2018-10-26 12:00`, `2018-10-26 12:00:00-05:00` or other formats and types supported by `pandas.to_datetime <https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.to_datetime.html#pandas.to_datetime>`_ - ``DataFrame`` :key data_frame_timestamp_timezone: name of the timezone which is used for timestamp column - ``DataFrame`` � )�pd�npz%Must be DataFrame, but type was: {0}.�data_frame_measurement_nameNz4"data_frame_measurement_name" is a Required ArgumentZdata_frame_timestamp_columnZ data_frame_timestamp_timezoneF)�deep)�unit�tzinfo�UTC�data_frame_tag_columnsc S s | d d S )N� r r )�colr r r �<lambda>� � z.DataframeSerializer.__init__.<locals>.<lambda>)�keyr z{keys[z]}r# zp[�]z${ '' if z == '' or pd.isna(z&) else f',z={str(z7).translate(_ESCAPE_STRING)}' }�,z).translate(_ESCAPE_KEY)}� �iz{"" if pd.isna(z ) else f"z={�}z"}z,{ '' if pd.isna(z%) else f'z="{str(z8).translate(_ESCAPE_STRING)}"' }z ).translate(_ESCAPE_STRING)}"z {p[%s].value}z{int(p[%s].value / 1e3)}z{int(p[%s].value / 1e6)}z{int(p[%s].value / 1e9)}z lambda p: f"""{measurement_name}� z""")�measurement_namer r �keysr c s t t� �fdd�|��S )Nc s �� � | � S r )Zisna��x)�pr r r r% � r& zDDataframeSerializer.__init__.<locals>._any_not_nan.<locals>.<lambda>)�any�map)r2 Zindexes�r )r2 r � _any_not_nan� s z2DataframeSerializer.__init__.<locals>._any_not_nan)=�extrasr r � isinstanceZ DataFrame� TypeError�format�type�get�copyr Z PeriodIndexZ to_timestamp� to_datetimeZ DatetimeIndexZ tz_localize�dt�hasattrr �setZ defaultTags�itemsr �add�sorted� enumerateZdtypesZisnullr3 �strr �append� translater r � issubclass�integerZfloatingZbool_r �joinr ZUSZMS�S�evalr �dict�replace�nanr �f� field_indexes�first_field_maybe_nullr6 �int�math�ceil�float�number_of_chunksr )#�selfr �point_settings� precisionr �kwargsr r Ztimestamp_columnZtimestamp_timezoneZdata_frame_timestampr" r/ �tags�fieldsrR r' �valuer Z null_columnsZtimestamp_indexr Z key_formatZ field_indexZ val_format� key_value�sep�suffixZ field_valuer. � timestamprQ r �vr6 r r5 r �__init__ s� 9 � ���� 0 $ ���� � z DataframeSerializer.__init__)� chunk_idxc s� |dkr� j }n2t�d|d � j� � j |� j |d � j � }� jrt� fdd�t� fdd�t|��D �}t|�S tt � j t|���S dS )z� Serialize chunk into LineProtocols. :param chunk_idx: The index of chunk to serialize. If `None` then serialize whole dataframe. NzSerialize chunk %s/%s ...r# c 3 s" | ]}t �d d� �|��V qdS )z ^(( |[^ ])* ),([a-zA-Z0-9])(.*)z\1\3\4N)�re�subrQ )r r2 �rY r r � <genexpr> s �z0DataframeSerializer.serialize.<locals>.<genexpr>c s � � | � j�S r )r6 rR r0 ri r r r% r& z/DataframeSerializer.serialize.<locals>.<lambda>) r �logger�debugrX r rS �filterr �listr4 rQ )rY rf �chunkZlpr ri r � serialize� s �z DataframeSerializer.serializec C s | j S )zy Return the number of chunks. :return: number of chunks or None if chunk_size is not specified. )rX ri r r r rX s z$DataframeSerializer.number_of_chunks)N) �__name__� __module__� __qualname__�__doc__r rT re rp rX r r r r r s � ^r c K s t | ||f|��� S )a� Serialize DataFrame into LineProtocols. :param data_frame: Pandas DataFrame to serialize :param point_settings: Default Tags :param precision: The precision for the unix timestamps within the body line-protocol. :key data_frame_measurement_name: name of measurement for writing Pandas DataFrame :key data_frame_tag_columns: list of DataFrame columns which are tags, rest columns will be fields :key data_frame_timestamp_column: name of DataFrame column which contains a timestamp. The column can be defined as a :class:`~str` value formatted as `2018-10-26`, `2018-10-26 12:00`, `2018-10-26 12:00:00-05:00` or other formats and types supported by `pandas.to_datetime <https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.to_datetime.html#pandas.to_datetime>`_ - ``DataFrame`` :key data_frame_timestamp_timezone: name of the timezone which is used for timestamp column - ``DataFrame`` )r rp )r rZ r[ r\ r r r � data_frame_to_list_of_points s ru )rt �loggingrU rg Zinfluxdb_clientr Z"influxdb_client.client.write.pointr r r r � getLoggerrk r r ru r r r r �<module> s
💾 Save Changes
❌ Cancel