U
vi4 @ sT d Z ddlZddlZddlmZ ddlmZmZmZ ddlm Z ddl
m Z ddl m
Z
ddlmZ ejdejd ZejZed
d d d
ddZed
dd d d
ddZedddZzddlZdZ W n e k
r dZ Y nX G dd de Z dd Z dd Z!ed d d Z"d*ed d d!Z#ed d"d#Z$ed d$d%Z%efd&d'Z&d(d) Z'dS )+z/Point data structure to represent LineProtocol. N)int)datetime timedeltatimezone)Decimal)Integral)get_date_helper)WritePrecision)tzz\,z\ z\nz\tz\r),
z\=)r =r r
r r z\"z\\)"\TFc @ s e Zd ZdZedd ZeefeedddZ dd Z
efd d
Z d d Z d
d Z
dddZedd Zedd Zdd Zdd ZdS )Pointz
Point defines the values that will be written to the database.
Ref: https://docs.influxdata.com/influxdb/latest/reference/key-concepts/data-elements/#point
c C s t | }|S )z3Create a new Point with specified measurement name.)r ) measurementp r S/opt/alt/python38/lib/python3.8/site-packages/influxdb_client/client/write/point.pyr 6 s zPoint.measurement)
dictionarywrite_precisionc
K sJ | dd}|dkr$| | dd }t|}| dd}|dk rd|D ] }|| krD||| | qDn*d| kr| d D ]\}}||| qx| dd}|dk r|D ] } | | kr|| | | qn"| d D ]\} }
|| |
q| d d
} | | kr|j| | |d | d i } d | kr.| d } ttd
d | |_|S )a,
Initialize point from 'dict' structure.
The expected dict structure is:
- measurement
- tags
- fields
- time
Example:
.. code-block:: python
# Use default dictionary structure
dict_structure = {
"measurement": "h2o_feet",
"tags": {"location": "coyote_creek"},
"fields": {"water_level": 1.0},
"time": 1
}
point = Point.from_dict(dict_structure, WritePrecision.NS)
Example:
.. code-block:: python
# Use custom dictionary structure
dictionary = {
"name": "sensor_pt859",
"location": "warehouse_125",
"version": "2021.06.05.5874",
"pressure": 125,
"temperature": 10,
"created": 1632208639,
}
point = Point.from_dict(dictionary,
write_precision=WritePrecision.S,
record_measurement_key="name",
record_time_key="created",
record_tag_keys=["location", "version"],
record_field_keys=["pressure", "temperature"])
Int Types:
The following example shows how to configure the types of integers fields.
It is useful when you want to serialize integers always as ``float`` to avoid ``field type conflict``
or use ``unsigned 64-bit integer`` as the type for serialization.
.. code-block:: python
# Use custom dictionary structure
dict_structure = {
"measurement": "h2o_feet",
"tags": {"location": "coyote_creek"},
"fields": {
"water_level": 1.0,
"some_counter": 108913123234
},
"time": 1
}
point = Point.from_dict(dict_structure, field_types={"some_counter": "uint"})
:param dictionary: dictionary for serialize into data Point
:param write_precision: sets the precision for the supplied time values
:key record_measurement_key: key of dictionary with specified measurement
:key record_measurement_name: static measurement name for data Point
:key record_time_key: key of dictionary with specified timestamp
:key record_tag_keys: list of dictionary keys to use as a tag
:key record_field_keys: list of dictionary keys to use as a field
:key field_types: optional dictionary to specify types of serialized fields. Currently, is supported customization for integer types.
Possible integers types:
- ``int`` - serialize integers as "**Signed 64-bit integers**" - ``9223372036854775807i`` (default behaviour)
- ``uint`` - serialize integers as "**Unsigned 64-bit integers**" - ``9223372036854775807u``
- ``float`` - serialize integers as "**IEEE-754 64-bit floating-point numbers**". Useful for unify number types in your pipeline to avoid field type conflict - ``9223372036854775807``
The ``field_types`` can be also specified as part of incoming dictionary. For more info see an example above.
:return: new data point
Zrecord_measurement_nameNZrecord_measurement_keyr record_tag_keystagsrecord_field_keysfieldsrecord_time_keytime)r field_typesc S s, | d | d dkrdn| d dkr&dndfS )Nr r iZuintu r )itemr r r z!Point.from_dict..) getr tagitemsfieldr dictmap _field_types)
r r kwargsZ measurement_pointr tag_key tag_valuer Z field_keyZ field_valuer r. r r r from_dict< s< M
zPoint.from_dictc C s( i | _ i | _|| _d| _t| _i | _dS )zInitialize defaults.N)_tags_fields_name_timeDEFAULT_WRITE_PRECISION_write_precisionr. )selfZmeasurement_namer r r __init__ s zPoint.__init__c C s || _ || _| S )a
Specify timestamp for DataPoint with declared precision.
If time doesn't have specified timezone we assume that timezone is UTC.
Examples::
Point.measurement("h2o").field("val", 1).time("2009-11-10T23:00:00.123456Z")
Point.measurement("h2o").field("val", 1).time(1257894000123456000)
Point.measurement("h2o").field("val", 1).time(datetime(2009, 11, 10, 23, 0, 0, 123456))
Point.measurement("h2o").field("val", 1).time(1257894000123456000, write_precision=WritePrecision.NS)
:param time: the timestamp for your data
:param write_precision: sets the precision for the supplied time values
:return: this point
)r9 r7 )r: r r r r r r s z
Point.timec C s || j |< | S )zAdd tag with key and value.)r4 )r: keyvaluer r r r) s
z Point.tagc C s || j |< | S )z Add field with key and value.)r5 )r: r+ r= r r r r+ s
z Point.fieldNc C s| t | jt}|dr.d| d}t|t t| j}t | j
| j }|sNdS t | j
|dkrb| jn|}| | | | S )z
Create LineProtocol.
:param precision: required precision of LineProtocol. If it's not set then use the precision from ``Point``.
#zThe measurement name 'z' start with '#'.
The output Line protocol will be interpret as a comment by InfluxDB. For more info see:
- https://docs.influxdata.com/influxdb/latest/reference/syntax/line-protocol/#comments
r$ N) _escape_keyr6 _ESCAPE_MEASUREMENT
startswithwarningswarn
SyntaxWarning _append_tagsr4 _append_fieldsr5 r. _append_timer7 r9 )r: precisionZ _measurementmessager4 r5 r7 r r r to_line_protocol s
zPoint.to_line_protocolc C s | j S )zGet precision.)r9 r: r r r r s zPoint.write_precisionc C s
|| _ dS )z-Set the string representation for all Points.N)Z_Point__str___rep)clsZ rep_functionr r r set_str_rep s zPoint.set_str_repc C s | S )z+Create string representation of this Point.)rJ rK r r r __str__ s z
Point.__str__c C sV t |tsdS | j|jkoT| j|jkoT| j|jkoT| j|jkoT| j|jkoT| j|jkS )z'Return true iff other is equal to self.F)
isinstancer r4 r5 r6 r7 r9 r. )r: otherr r r __eq__ s
z Point.__eq__)N)__name__
__module__ __qualname____doc__ staticmethodr r8 r, r r3 r; r r) r+ rJ propertyr classmethodrM rN rQ r r r r r / s
u
r c C st g }t | D ]F\}}|d kr"qt|}t|}|dkr|dkr|| d| q|r`dnd d| dS )Nr$ r r r )sortedr* r? _escape_tag_valueappendjoin)r _returnr1 r2 r) r= r r r rE s rE c C sR g }t | D ]2\}}|d kr$qt|tsBt|tsBt|drt|sNqt|}| drl|d d }|
t | d| qt|t st|drt|t
s||d}|
t | dt| | qt|t
r|
t | dt| qt|tr*|
t | dt| d qtd t| d
| d qd | S )
Nfloatz.0r r r" z="r zType: "z
" of field: "z" is not supported.r )rY r* rO r^ r _np_is_subtypemathisfinitestrendswithr[ r? r boolr( lower_escape_string
ValueErrortyper\ )r r r] r+ r= s_typer r r rF s(
"
" rF )returnc C s | d kr dS dt t| | S )Nr$ r )r _convert_timestamp)r r r r r rG 9 s rG c C s |d kr t }t| |S N) _ESCAPE_KEYrc translate)r) Z escape_listr r r r? ? s r? c C s t | }|dr|d7 }|S )Nr r )r? rd )r= retr r r rZ E s
rZ c C s t | tS rn )rc rp _ESCAPE_STRING)r= r r r rg L s rg c C s t }t| tr| S t| tr(|| } t| tst| tjS dS )NZdtypeFr^ r )
_HAS_NUMPYhasattrnpZ
issubdtypeZfloatinginteger)r= Znp_typer r r r` k s r` )N)(rU ra rB builtinsr r r r decimalr numbersr Z&influxdb_client.client.util.date_utilsr Z&influxdb_client.domain.write_precisionr
fromtimestamputcrs rt r8 rc maketransr@ ro rr numpyrz rx ModuleNotFoundErrorobjectr rE rF rG r? rZ rg rm r` r r r r sX
]