import requests
import json
import numpy as np
import random


def generatea_random_para():
    randint = int(np.random.normal(100, 10))
    para_1 = randint if randint < 100 else 100
    para_2 = random.randint(1, 100)
    para_3 = random.randint(1, 200)
    para_4 = random.randint(1, 300)
    para_5 = random.randint(1, 400)
    return f"DNS服务应答率{para_1}；icmp时延{para_2}；udp时延{para_3}；tcp时延{para_4}；pps为{para_5}"


def test_initial(init_data):
    url_initial = 'http://127.0.0.1:5000/initial'
    response_initial = requests.post(url_initial, json=init_data)
    print("Initial Response Body:", response_initial.json())


def test_adjust(adjust):
    url_adjustment = 'http://127.0.0.1:5000/adjustment'
    response_adjustment = requests.post(url_adjustment, json=adjust)
    print("Adjustment Response Body:", response_adjustment.json())


# 定义要发送的数据
init_data = {"clxz": "拒绝服务", "mbgz":"（1）ip地址 x.x.x.x；（2）未开启DNSSEC；（3）可以使用缓存投毒；（4）可使用DDoS；（5）不适用DNSSEC_DDoS；（6）适用放大攻击；（7）开启了DoH；（8）开启了DoT；（9）开启了IPv6功能；（10）-（18）不关键暂时不提供；", "ztgz":"DNS服务应答率100%；icmp时延20ms；udp时延5ms；tcp时延6ms；pps为200kpps", "script":"DoH_DDoS;doh_ddos_n-4"}
adjust_1 = {"ztgz": generatea_random_para()}
adjust_2 = {"ztgz": generatea_random_para()}
adjust_3 = {"ztgz": generatea_random_para()}
adjust_4 = {"ztgz": generatea_random_para()}
adjust_5 = {"ztgz": generatea_random_para()}
adjust_6 = {"ztgz": generatea_random_para()}



test_initial(init_data)
print(f"当前状态为{adjust_1}")
print(f"响应为{test_adjust(adjust_1)}")
print(f"当前状态为{adjust_2}")
print(f"响应为{test_adjust(adjust_2)}")
print(f"当前状态为{adjust_3}")
print(f"响应为{test_adjust(adjust_3)}")
print(f"当前状态为{adjust_4}")
print(f"响应为{test_adjust(adjust_4)}")
print(f"当前状态为{adjust_5}")
print(f"响应为{test_adjust(adjust_5)}")
print(f"当前状态为{adjust_6}")
print(f"响应为{test_adjust(adjust_6)}")