🔐 Sid Gifari File Manager Pro
v8.0.5 | 2026-08-08 16:54:40 | PHP 8.2.32
📂
/ (Root)
/
opt
/
alt
/
python38
/
lib
/
python3.8
/
site-packages
/
influxdb_client
/
client
/
__pycache__
📍 /opt/alt/python38/lib/python3.8/site-packages/influxdb_client/client/__pycache__
🔄 Refresh
✏️
Editing: flux_csv_parser.cpython-38.pyc
Read Only
U v�i�= � @ s� d Z ddlZddlZddlZddlZddlmZ ddlm Z ddl m Z m Z m Z mZ ddlmZ ddlmZ dZd Zd ZeeegZG d d � d e�ZG d d� de�ZG dd� de�ZG dd� de�ZG dd� de �Z G dd� de �Z G dd� d�Z dS )z>Parsing response from InfluxDB to FluxStructures or DataFrame.� N)�Enum)�List)� FluxTable� FluxColumn� FluxRecord� TableList)�get_date_helper)�_UTF_8_encodingz#defaultz#groupz #datatypec @ s e Zd ZdZdd�dd�ZdS )�FluxQueryExceptionz The exception from InfluxDB.N��returnc C s || _ || _dS )zInitialize defaults.N)�message� reference)�selfr r � r �W/opt/alt/python38/lib/python3.8/site-packages/influxdb_client/client/flux_csv_parser.py�__init__ s zFluxQueryException.__init__)�__name__� __module__� __qualname__�__doc__r r r r r r s r c @ s e Zd ZdZdS )�FluxCsvParserExceptionz$The exception for not parsable data.N)r r r r r r r r r s r c @ s e Zd ZdZdZdZdZdS )�FluxSerializationModez'The type how we want to serialize data.� � � N)r r r r �tables�stream� dataFramer r r r r # s r c @ s e Zd ZdZdZdZdS )�FluxResponseMetadataModezIThe configuration for expected amount of metadata response from InfluxDB.r r N)r r r r �full� only_namesr r r r r + s r c @ s e Zd Zdd� ZdS )�_FluxCsvParserMetadatac C s( d| _ d| _d| _d | _g | _d| _d S )Nr ���F)� table_index�table_id�start_new_table�table�groups�parsing_state_error�r r r r r 4 s z _FluxCsvParserMetadata.__init__N)r r r r r r r r r" 3 s r" c @ s e Zd ZdZddejdfeee edd�dd�Z dd� Z d d � Z d d � Z d d �dd�Z dd �dd�Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zed"d#� �Zed$d%� �Zed&d'� �Zed(d)� �Zd*d+� Z e e d,�d-d.�Z e e d/�d0d1�Z!e"d �d2d3�Z#e d4�d5d6�Z$dS )7� FluxCsvParserzJParse to processing response from InfluxDB to FluxStructures or DataFrame.NF)�serialization_mode�data_frame_index�response_metadata_moder c C sp || _ t� | _|| _|| _|| _|| _g | _|dk r:|jnd| _ |dk rN|j nd| _ dt |�j krfdnd| _dS )z� Initialize defaults. :param response: HTTP response from a HTTP client. Acceptable types: `urllib3.response.HTTPResponse`, `aiohttp.client_reqrep.ClientResponse`. NZClientResponseTF)� _responser r �_serialization_mode�_response_metadata_mode�_use_extension_dtypes�_data_frame_index�_data_frame_valuesZ profilers� _profilersZprofiler_callback�_profiler_callback�typer Z _async_mode)r �responser, r- Z query_optionsr. Zuse_extension_dtypesr r r r @ s zFluxCsvParser.__init__c C s | j �� d S �N)r/ �closer* r r r �_closeU s zFluxCsvParser._closec C sV t | jd�r<| jjr<ddlm} t�|| jj�t ���| _ nt�t � | jt ��| _ | S )�Initialize CSV reader.�closedr )�StringIO) �hasattrr/ r= �ior> � csv_parser�reader�data�decoder �_reader�codecs� iterdecode)r r> r r r � __enter__X s zFluxCsvParser.__enter__c C s | � � dS )zClose HTTP response.N)r; �r �exc_type�exc_val�exc_tbr r r �__exit__b s zFluxCsvParser.__exit__r c � s" ddl m} |t| jj��| _| S )r< r )� AsyncReader)ZaiocsvrN � _StreamReaderToWithAsyncReadr/ �contentrE )r rN r r r � __aenter__f s zFluxCsvParser.__aenter__c � s | � |||� dS )zShutdown the client.N)rM rI r r r � __aexit__m s zFluxCsvParser.__aexit__c c s( | �}|� � D ] }|V qW 5 Q R X dS )zReturn Python generator.N)�_parse_flux_response)r �parser�valr r r � generatorq s zFluxCsvParser.generatorc C s | � � S )z Return Python async-generator.)�_parse_flux_response_asyncr* r r r �generator_asyncw s z FluxCsvParser.generator_asyncc c s^ t � }| jD ] }| �||�D ] }|V q q | jtjkt| d�@ rZ| �� }| �|j �sZ|V d S )N� _data_frame) r" rE �_parse_flux_response_rowr0 r r r? �_prepare_data_frame�_is_profiler_tabler' )r �metadata�csvrU �dfr r r rS { s z"FluxCsvParser._parse_flux_responsec C s� t � }z�zb| j2 z$3 d H W }| �||�D ] }|V q(q6 | jtjkt| d�@ rh| �� }| � |j �sh|V W nH t k r� } z*t |�j }d|ks�d|kr�|�d� � W 5 d }~X Y nX W 5 | �� X d S )NrY �CancelledError� TimeoutErrorzTStream cancelled during read. Recommended: Check Influxdb client `timeout` setting.)r" r; rE rZ r0 r r r? r[ r\ r' � BaseExceptionr7 r Zadd_note)r r] r^ rU r_ �e�e_typer r r rW � s z(FluxCsvParser._parse_flux_response_asyncc c s� t |�dk r�n�d|d kr2d|d kr2d|_�n�|jrR|d }|d }t||��|d }|tkrh|jrz| jtjkr�|js�| j t j kt | d�@ r�| � � }| �|j�s�|V d|_t� |_| �|j|j� |jd |_d|_n|jd kr�td ��t|k�r | �|j|� �n�t|k�r ||_�n�t|k�r:| �|j|� �n�|j�r�|jj�s�| jtjk�r�| �|jtt d d � |��� tt d d � |��|_| � |j|j� | � |j|� d |_| j t j k�r�ddl m } tt dd � |jj��}|j!g |d d�| _"n�t#|d �} |jdk�r| |_|j| k�r\|jj} t� |_|jj�$| � | �|j|j� |jd |_| |_| �%|jd |j|�} | �&| ��r�| �'| � nV| j t j(k�r�| j(|jd j)�*| � | j t j+k�r�| V | j t j k�r�| j,�*| j-� d S )Nr �errorr r Tr rY r# zAUnable to parse CSV response. FluxTable definition was not found.c S s dS )N�stringr ��columnr r r �<lambda>� � z8FluxCsvParser._parse_flux_response_row.<locals>.<lambda>c S s dS )N�falser rg r r r ri � rj F��pdc S s | j S r9 )�label)�itr r r ri � rj )rC �columns�index).�lenr) r � ANNOTATIONSr& r1 r r! r' r0 r r r? r[ r\ r � _insert_tabler$ r% r �ANNOTATION_DATATYPE�add_data_types�ANNOTATION_GROUPr( �ANNOTATION_DEFAULT�add_default_empty_valuesrp �list�map� add_groups�add_column_names_and_tags�extrasrm � DataFramerY �int�extend� parse_record�_is_profiler_record�_print_profiler_infor �records�appendr r4 �values) r r] r^ re Zreference_value�tokenr_ rm �labelsZ current_idZ flux_columns� flux_recordr r r rZ � s� �� � z&FluxCsvParser._parse_flux_response_rowc C sj ddl m} |�| j�}g | _| jr@| j�| j�| _|�| j�}|�| j�|j �|g�}| j rf|� � S |S )Nr rl ) r~ rm r r4 r3 rY Z set_index�concatZastypeZdtypesr2 Zconvert_dtypes)r rm Z_temp_dfr_ r r r r[ � s z!FluxCsvParser._prepare_data_framec C sP t |�}|jD ]<}|j}||jd }| �||�|j|< |j�|j| � q|S )zParse one record.r )r rp rn rq � _to_valuer� �rowr� )r r$ r' r^ �recordZ fluxColumnZ column_name�str_valr r r r� s zFluxCsvParser.parse_recordc C s� |dks|d kr^|j }|dks&|d krR| jtjkrN| jrJddlm} |jS d S d S | �||�S d|j krl|S d|j kr~d|kS d|j ks�d|j kr�t |�S d |j kr�t |�S d |j kr�t � |�S d |j ks�d |j kr�t� �|�S d |j kr�t |�S d S )N� r rl rf �boolean�trueZ unsignedLong�long�doubleZ base64BinaryzdateTime:RFC3339zdateTime:RFC3339Nano�duration)� default_valuer0 r r r2 r~ rm ZNAr� � data_typer� �float�base64� b64decoder Z parse_date)r r� rh r� rm r r r r� s0 zFluxCsvParser._to_valuec C s8 t dt|��D ]$}t|d || d�}| j�|� qdS )zAdd data types to columns.r )rq r� N)�rangerr r rp r� )r' � data_typesrq Z column_defr r r rv ; s z FluxCsvParser.add_data_typesc C s* d}| j D ]}|| dk|_|d7 }q dS )zAdd group keys to columns.r r� N)rp �group)r' r^ �irh r r r r| B s zFluxCsvParser.add_groupsc C s& d}| j D ]}|| |_|d7 }q dS )z Add default values to columns.r N)rp r� )r' �default_valuesr� rh r r r ry J s z&FluxCsvParser.add_default_empty_valuesc C sZ t |�t t|��kr4d|� d�}t�|t� t|� d}| jD ]}|| |_|d7 }q>dS )zAdd labels to columns.z6The response contains columns with duplicated names: 'z_'. You should use the 'record.row' to access your data instead of 'record.values' dictionary. r N)rr �set�warnings�warn� UserWarning�printrp rn )r' r^ r r� rh r r r r} R s z'FluxCsvParser.add_column_names_and_tagsc C s | j tjkr| j�||� d S r9 )r0 r r �insert)r r' r$ r r r rt a s zFluxCsvParser._insert_table)r� r c C s: | j s dS | j D ]$}d|jkr|d d| kr dS qdS )NFZ _measurement� profiler/T)r5 r� )r r� Zprofilerr r r r� e s z!FluxCsvParser._is_profiler_record)r' r c C s | j s dS ttdd� |j��S )NFc S s | j dko| jdkS )NZ _profiler�result)r� rn rg r r r ri t rj z2FluxCsvParser._is_profiler_table.<locals>.<lambda>)r5 �any�filterrp )r r' r r r r\ o s �z FluxCsvParser._is_profiler_tablec s( � j s � jS tt� fdd�� j��S dS )z Get the list of flux tables.c s � � | � S r9 )r\ )r' r* r r ri | rj z*FluxCsvParser.table_list.<locals>.<lambda>N)r5 r r r� r* r r* r � table_listw s zFluxCsvParser.table_list)r� c C s� |� � �d�r�| jr | �|� n�d|� � }tdt|�d � t|� tt|�d � |jD ]V}|| }t|t�r�t|�dkr�t|d�d|� �� q^|d k r^t|d�d|d��� q^d S ) Nr� z Profiler: � �=�2 z<20z: z: )�get_measurement� startswithr6 r� rr r� � isinstance�str)r r� �msg�namerU r r r r� ~ s z"FluxCsvParser._print_profiler_info)%r r r r r r r r r� r r; rH rM rQ rR rV rX rS rW rZ r[ r� r� � staticmethodrv r| ry r} rt r �boolr� r r\ r r� r� r r r r r+ = sH �� \ # r+ c @ s$ e Zd Zdd� Zeed�dd�ZdS )rO c C s || _ t�t�� | _d S r9 )r8 rF �getincrementaldecoderr �decoder)r r8 r r r r � s z%_StreamReaderToWithAsyncRead.__init__)�sizer c � s6 | j �|�I d H }|s&| jjddd�S | jj|dd�S )Nrj T)�finalF)r8 �readr� rD )r r� Z raw_bytesr r r r� � s z!_StreamReaderToWithAsyncRead.readN)r r r r r� r� r� r r r r rO � s rO ) r r� rF r^ rA r� �enumr �typingr Z!influxdb_client.client.flux_tabler r r r Z&influxdb_client.client.util.date_utilsr Zinfluxdb_client.restr rx rw ru rs � Exceptionr r r r �objectr" r+ rO r r r r �<module> s, T
💾 Save Changes
❌ Cancel