import pandas as pd
import time
import influxdb_client
from influxdb_client.client.write_api import PointSettings

ONE_SECOND = pd.Timedelta('1s')
# Time for the first measures
t = pd.Timestamp.now(tz='UTC') - pd.Timedelta('2min').floor('s')

# Define a first client which receives measures for odd numbers
client_1 = influxdb_client.InfluxDBClient(**influx_configs)
w_api_1 = client_1.write_api()
q_api_1 = client_1.query_api()
# Write a point to the first client before the second client is created
point_1_1 = [
    {"measurement": "test",
     "tags": {"client_number": "1",
              "numbers": "odds"},
     "time": t.isoformat(),
     "fields": {"value": 1.}}
]
w_api_1.write(bucket='test_tags_bug', record=point_1_1)

# Define a second client which receives measures for even numbers
# This client uses default tags
client_2_tags = {"client_number": "2", "numbers": "even" }
client_2 = influxdb_client.InfluxDBClient(
    default_tags=client_2_tags, **influx_configs
)
# Define a write API for the second client with default arguments
w_api_2 = client_2.write_api()#point_settings=PointSettings()) #uncomment to see the difference
# Write a point to the second client
point_2_1 = [
    {"measurement": "test",
     "time": (t + ONE_SECOND).isoformat(),
     "fields": {"value": 2.}}
]
w_api_2.write(bucket='test_tags_bug', record=point_2_1)

# Define a third client which receives measures for odd numbers and a half
client_3 = influxdb_client.InfluxDBClient(**influx_configs)
w_api_3 = client_3.write_api()
w_api_3_2 = client_3.write_api(point_settings=PointSettings())
point_3_1 = [
    {"measurement": "test",
     "tags": {"client_number": "3", "numbers": "odds and a half"},
     "time":  (t + 2*ONE_SECOND).isoformat(),
     "fields": {"value": 1.5}}
]
w_api_3.write(bucket='test_tags_bug', record=point_3_1)

# Write a second point for each client, one minute after the first point
t_2 = (t + pd.Timedelta('1min'))
point_1_2 = [
    {"measurement": "test",
     "tags": {"client_number": "1", "numbers": "odds"},
     "time": t_2.isoformat(),
     "fields": {"value": 3.}}
]
point_2_2 = [
    {"measurement": "test",
     "time": (t_2 + ONE_SECOND).isoformat(),
     "fields": {"value": 4.}}
]
point_3_2 = [
    {"measurement": "test",
     "tags": {"client_number": "3", "numbers": "odds and a half"},
     "time": (t_2 + 2 * ONE_SECOND).isoformat(),
     "fields": {"value": 3.5}}
]
# client 1 and 2 uses the same write API as initially
w_api_1.write(bucket='test_tags_bug', record=point_1_2)
w_api_2.write(bucket='test_tags_bug', record=point_2_2)
# client 3 uses a different write API, for which a point settings was defined
# explicitly
w_api_3_2.write(bucket='test_tags_bug', record=point_3_2)
time.sleep(2)
q_api_1.query_data_frame('from(bucket: "test_tags_bug") |> range(start: -1h) |> filter(fn: (r) => r["_measurement"] == "test") |> pivot(rowKey:["_time"], columnKey: ["_field"], valueColumn: "_value")')