# 调用不同的脚本
import time
import apps.utiltcping as utping
from flask import Blueprint,request
from icmplib import ping
from dns import resolver
import dns.nameserver

bp=Blueprint("script",__name__,url_prefix="/script")

@bp.route("/test")
def hello():
    return "hello"

# icmp时延测试
@bp.route("/icmpdelay")
def icmpdelay():
    addr=request.args.get("ip")
    host=ping(addr,count=2,interval=0.1,timeout=2)
    return str(host.avg_rtt)


@bp.route("/tcpdelay")
def tcpdelay():
    addr=request.args.get("ip")
    port=request.args.get("port",default=53)
    tdelay=utping.cli(addr,port,2,2)
    return str(tdelay)

# dns时延测试函数，返回执行完成时的时间戳
@bp.route("/dnsdelay")
def dnsdelay():
    addr = request.args.get("ip")
    port = request.args.get("port",default=53)
    domain = request.args.get("domain", default="www.baidu.com")
    qtype=request.args.get("qtype",default="A")

    start_time=time.time()
    # 解析器配置
    myResolver = resolver.Resolver()
    myResolver.nameservers = [dns.nameserver.Do53Nameserver(addr, port=port)]

    # 开始解析
    ans=myResolver.resolve(domain,qtype,lifetime=4)
    stop_time=time.time()
    t=stop_time-start_time
    return  str(int(t*1000))