from datetime import datetime, timedelta

from influxdb_client import InfluxDBClient, Point
from influxdb_client.client.write_api import SYNCHRONOUS

from dateutil import tz
from config import settings
import pandas as pd

client = InfluxDBClient(url="https://influxdb.devel.water.cn",
                        token="KwiGZ6Evee-A5ocaPqRyClNve0opdDUccukOWwr1jpxShvk6NlkzDC5D9jbSbF7LJwXA_EbKA==",
                        org="waterstrategy",
                        verify_ssl=False,
                        timeout=60_000)   # 不行时改值

write_api = client.write_api(write_options=SYNCHRONOUS)
query_api = client.query_api()

# time_index = pd.date_range(start=start, end=stop - timedelta(microseconds=1), freq='5min')

bucket_list = ["waterstrategy", "waterstrategy-1min", "waterstrategy-5min"]


def deleteData(bucket: str, start: datetime, stop: datetime, measurement: str):
    print(f"删除数据{bucket} measurement:{measurement} start:{start.isoformat()} stop:{stop.isoformat()}")
    ### 1先删除数据
    client.delete_api().delete(start=start, stop=stop,
                               predicate=f"_measurement=\"{measurement}\"",
                               bucket=bucket)

# 删除项
delete_lst = [
    # 'SCADA出厂水当日总量',
    # 'SCADA原水当日总量',
    # 'SCADA反冲洗当日水量',
    # 'SCADA回用水当日水量',
    # '出厂水流量总累计偏差',
    # '出厂水流量累计偏差',
    # '原水流量',
    # '反洗水量偏差',
    # '回用水量偏差',
    # '累计偏差',
    # '进厂水流量总累计偏差',
    # '进厂水流量累计偏差',
    # '预测原水恒流',
    # '预测原水流量累计',
    # '预测原水流量累计值偏差',
    # '预测水位',
]

for i in delete_lst:
    measurement = 'adv_model_rawwaterv3_forecast_' + i
    # measurement = 'adv_model_rawwaterv3_forecast_' + '预测原水恒流'
    start_time = datetime.fromisoformat('2023-04-05T00:00:00+08:00')
    end_time = datetime.fromisoformat('2023-04-05T23:59:59+08:00')
    print('删除时间查看： ', start_time, end_time, measurement)
    for bucket in bucket_list:
        deleteData(bucket=bucket, start=start_time, stop=end_time,
                   measurement=measurement)