U
vi^ @ s d Z ddlmZ ddlZddlZddlZddlZddlmZmZ ddl m
Z
m Z m Z m
Z
mZmZ ddlmZ ddlmZmZmZmZmZmZmZmZmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z* ddl+m,Z,m-Z-m.Z. dd l/m0Z0m1Z1m2Z2 dd
l3m4Z4 dd l5m6Z6 dd l7m8Z8 dd
l9m:Z: ddl;mW n e?k
rb dZ>Y nX dddddddddg Z@G dd deAZBG d d d eAZCG d d d eAZDG d d! d!eAZEG d"d# d#eZFd$d% ZGd&d' ZHdS )(z+Commons function for Sync and Async client. )absolute_importN)datetime timedelta)List GeneratorAnyUnionIterableAsyncGenerator) HTTPResponse)
ConfigurationDialectQueryOptionStatementVariableAssignment
Identifier
ExpressionBooleanLiteralIntegerLiteral FloatLiteralDateTimeLiteralUnaryExpressionDurationLiteralDuration
StringLiteralArrayExpressionImportDeclarationMemberExpressionMemberAssignmentFile WriteService QueryService
DeleteServiceDeletePredicateRequest)FluxResponseMetadataMode
FluxCsvParserFluxSerializationMode)
FluxRecord TableList CSVIterator)get_date_helper)get_org_query_param)MissingPivotFunction)DataframeSerializer)_UTF_8_encodingTFz&influxdb_client.client.influxdb_clientz,influxdb_client.client.influxdb_client_asyncz influxdb_client.client.write_apiz&influxdb_client.client.write_api_asyncz"influxdb_client.client.write.retryz1influxdb_client.client.write.dataframe_serializerz2influxdb_client.client.util.multiprocessing_helperzinfluxdb_client.client.httpz!influxdb_client.client.exceptionsc @ sD e Zd Zd
eeeddddZededd d
Zedd d ZdS ) _BaseClientN' F)org default_tagshttp_client_loggerreturnc K s || _ || _|| _|| _t | _t| j ts4td| j drT| j d d | j_
n
| j | j_
|| j_ | dd| j_
| dd | j_| dd | j_| dd | j_| d d | j_| d
d | j_| d d | j_| d d | j_| d
| jj| j_|| j_t|| jjd< tD ]}
t|
| jj|
< q|| j_| dd | j_ | dd | j_ d | _ d | _ | jrd| _ d| j | _ | dd} | rd| _ dt !|" # | _ | jj r| jj rd | _ d | _ | dd| _$| dd | _%d S )Nz#"url" attribute is not str instance/
verify_sslT ssl_ca_cert cert_file
cert_key_filecert_key_password ssl_contextproxy
proxy_headersconnection_pool_maxsizer3 usernamepassword
AuthorizationzToken
auth_basicFzBasic retries profilers)&urltokenr1 r2 _Configurationconf
isinstancestr
ValueErrorendswithhost enable_gzipgetr7 r8 r9 r: r; r< r= r> r? timeoutlogging getLoggerloggers
LOGGERS_NAMESdebugr@ rA Zauth_header_nameZauth_header_valuebase64 b64encodeencodedecoderD rE ) selfrF rG rV rQ rO r1 r2 r3 kwargsZ
client_loggerrC r] M/opt/alt/python38/lib/python3.8/site-packages/influxdb_client/client/_base.py__init__/ sT
z_BaseClient.__init__
config.ini) config_filec s. t |dddz | W n@ t jk
rf t| }dd l}|| dW 5 Q R X Y nX td fdd }td fd d
}td fd d } |d
}
|d} d } |dr|d} d }
|dr|d}
d}|dr|d}d }|dr
|d}d }|dr |d}d }|dr6|d}d }|drL|d}d }|drb|d}d}|drx|d}d }| drr d }n dd dD }t
|}d }|drd d |d d D }d }|d r|d }| |
| f|t | |
||t
|||||t |t
|||d |S )!N config_nameZinflux2Fr Tkeyc s& rt | n
| }|dS )N")rK strip)rd valueconfigrb is_jsonr] r^
_config_valuet s z4_BaseClient._from_config_file.._config_valuec s r | kS | S N)
has_optionrc rh r] r^ _has_optionx s z2_BaseClient._from_config_file.._has_optionc s r | kS | S rl ) has_sectionrc )ri rj r] r^ _has_section{ s z3_BaseClient._from_config_file.._has_sectionrF rG rQ r1 r7 r8 r9 r: r; r? rC tagsc S s i | ]\}}|| d qS )re rf ).0kvr] r] r^
s z1_BaseClient._from_config_file..rE c S s g | ] }| qS r] rr rs xr] r] r^
s z1_BaseClient._from_config_file..,r= )rV rQ r1 r2 rO r7 r8 r9 r: r; r? rC rE r= ) configparser ConfigParserrP read ParsingErroropenjsonloadrK itemsdictsplit_to_int_to_bool)clsra rV rO r\ Z json_filer rk rn rp rF rG rQ r1 r7 r8 r9 r: r; r? rC r2 rq rE r= r] rh r^ _from_config_fileg s
z _BaseClient._from_config_filec K s& t dd}t dd}t dd}t dd}t d d
}t d d } t d d }
t d
d } t dd } t dd }
t dd}t dd }d }|d k rdd |dD }t }t j D ]&\}}|dr|||dd < q| ||f|t||||t || |
| | t|
t ||d
|S )NZINFLUXDB_V2_URLzhttp://localhost:8086ZINFLUXDB_V2_TOKENzmy-tokenZINFLUXDB_V2_TIMEOUTZ10000ZINFLUXDB_V2_ORGzmy-orgZINFLUXDB_V2_VERIFY_SSLTrueZINFLUXDB_V2_SSL_CA_CERTZINFLUXDB_V2_CERT_FILEZINFLUXDB_V2_CERT_KEY_FILEZ INFLUXDB_V2_CERT_KEY_PASSWORDZ#INFLUXDB_V2_CONNECTION_POOL_MAXSIZEZINFLUXDB_V2_AUTH_BASICFalseZINFLUXDB_V2_PROFILERSc S s g | ] }| qS r] rr rw r] r] r^ ry s z4_BaseClient._from_env_properties..rz ZINFLUXDB_V2_TAG_ )
rV rQ r1 r2 rO r7 r8 r9 r: r; r? rC rE )
osgetenvr r environr
startswithlowerr r )r rV rO r\ rF rG rQ r1 r7 r8 r9 r: r; r? rC profrE r2 rd rg r] r] r^ _from_env_properties s@
z _BaseClient._from_env_properties)Nr0 FNNN)r` NF)NF) __name__
__module__ __qualname__rK r r_ classmethodr r r] r] r] r^ r/ . s 8Sr/ c @ sl e Zd ZeddddddgddZd8d
d Zd ejfeed d
dZ d ejfeed ddZ
e e dddZ
d ejfeeeed f d ddZd ejfeed d ddZd ejdfeddd Zd ejdfedd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zed fe d+d,d-Z e e e d. d/d0d1Z e e!e"d f d2d3d4Z#e d9e e e$ d5d6d7Z%d S ):
_BaseQueryApiTrz #datatypegroupdefaultZRFC3339)header delimiterZcomment_prefix annotationsZdate_time_formatNc C s6 ddl m} |d kr| n|| _|| _t|j| _d S )Nr QueryOptions) influxdb_client.client.query_apir _query_options_influxdb_clientr!
api_clientZ
_query_api)r[ influxdb_client
query_optionsr r] r] r^ r_ s z_BaseQueryApi.__init__)response_metadata_moder4 c C s" | |||}t| | S )z
Parse HTTP response to TableList.
:param response: HTTP response from an HTTP client. Expected type: `urllib3.response.HTTPResponse`.
)_to_tables_parserlist generator
table_listr[ responser r _parserr] r] r^
_to_tables s z_BaseQueryApi._to_tablesc sZ | |||4 I dH 4}| 2 z 3 dH W }q 6 | W 5 Q I dH R S Q I dH R X dS )z
Parse HTTP response to TableList.
:param response: HTTP response from an HTTP client. Expected type: `aiohttp.client_reqrep.ClientResponse`.
N)r generator_asyncr )r[ r r r parser_r] r] r^ _to_tables_async s z _BaseQueryApi._to_tables_async)r r4 c C s t |S )zParse HTTP response to CSV.)r) )r[ r r] r] r^ _to_csv s z_BaseQueryApi._to_csvc C s | |||}| S )z
Parse HTTP response to FluxRecord stream.
:param response: HTTP response from an HTTP client. Expected type: `urllib3.response.HTTPResponse`.
) _to_flux_record_stream_parserr r r] r] r^ _to_flux_record_stream s z$_BaseQueryApi._to_flux_record_stream)r' Nc s | |||}| I dH S )z
Parse HTTP response to FluxRecord stream.
:param response: HTTP response from an HTTP client. Expected type: `aiohttp.client_reqrep.ClientResponse`.
N)r
__aenter__r r r] r] r^ _to_flux_record_stream_async s z*_BaseQueryApi._to_flux_record_stream_asyncF)r c C s | |||||}| S )z
Parse HTTP response to DataFrame stream.
:param response: HTTP response from an HTTP client. Expected type: `urllib3.response.HTTPResponse`.
) _to_data_frame_stream_parserr r[ data_frame_indexr r r use_extension_dtypesr r] r] r^ _to_data_frame_stream s z#_BaseQueryApi._to_data_frame_streamc s$ | |||||}| I dH S )z
Parse HTTP response to DataFrame stream.
:param response: HTTP response from an HTTP client. Expected type: `aiohttp.client_reqrep.ClientResponse`.
N)r r r r r] r] r^ _to_data_frame_stream_async% s z)_BaseQueryApi._to_data_frame_stream_asyncc C s t |tj||dS N)r serialization_moder r )r% r& Ztables)r[ r r r r] r] r^ r 1 s z _BaseQueryApi._to_tables_parserc C s t |tj||dS r )r% r& stream)r[ r r r r] r] r^ r 5 s z+_BaseQueryApi._to_flux_record_stream_parserc C s t |tj||||dS )N)r r r r r r )r% r& Z dataFrame)r[ r r r r r r] r] r^ r 9 s z*_BaseQueryApi._to_data_frame_stream_parserc C sZ ddl m} t|tr |}nt|}t|dkr>|jg ddS t|dkrR|d S |S dS )z.Parse stream of DataFrames into expected type. )pdr N)columnsindex )extrasr rJ r len DataFrame)r[ Z
_generatorr Z _dataFramesr] r] r^ _to_data_frames@ s
z _BaseQueryApi._to_data_framesc C s t || jdS )N)r1 client)r+ r )r[ r1 r] r] r^
_org_paramO s z_BaseQueryApi._org_paramc C s: | j r| j jr| j S | jjr6ddlm} || jjdS d S )Nr r )rE )r rE r r r )r[ r r] r] r^ _get_query_optionsR s
z _BaseQueryApi._get_query_options)paramsc K sn | }|d k r|jnd }t||t||d}|rTtd td td t| |ddrjt| |S )N)querydialectZexternz
===============zProfiler: queryz===============Zdataframe_queryF) r rE r r _build_flux_astprintrP r, Z
print_warning)r[ r r r r\ r rE qr] r] r^
_create_queryY s
z_BaseQueryApi._create_queryr )r r4 c
C sL g }| D ]:\}}t|}|d kr(q |tdtdtd|| q |S )Nr r r )r r _parm_to_extern_astappendr r r )r Z
statementsrd rg Z
expressionr] r] r^ _params_to_extern_asti s
z#_BaseQueryApi._params_to_extern_ast)r4 c C s< | d kr d S t | tr td| S t | tr8tdt| S t | trLtd| S t | trt
| } t | dd}| j d | d}t
d| d | d
S t | t rt| td d }|dk rtd
tdt| ddgddS tdt|ddgS nDt | trtdt| S t | tr4tdttdd | dS | S d S )Nr r r Z
nanosecondr i Z09dr z%Y-%m-%dT%H:%M:%S.Zr ) microsecondsr r us)Z magnitudeunit-)argumentoperatorr r c S s
t | S rl )r r )itr] r] r^ z3_BaseQueryApi._parm_to_extern_ast..elements)rJ boolr intr rK floatr r r* to_utcgetattr microsecondr strftimer r r r r r r r map)rg nanosecondsfractionZ _micro_deltar] r] r^ r w s:
z!_BaseQueryApi._parm_to_extern_ast)r rE c C s g }g }|d k rt |dkr|tdtddd g }|D ]}|td|d q e Zd Zdd Zeed fdd Zed fdd Z ZS ) rH c C s t | d| _d | _d | _d S )NF)r r_ rO r@ rA )r[ r] r] r^ r_ s
z_Configuration.__init__)r r c s@ t || | jr<|dkr,d|d< d|d< |dkr sP h
/ _9%