import urllib3
from influxdb import InfluxDBClient
from netmiko import Netmiko


urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)



def trunkStatus(node):
    global status,field
    command1 = net_connect.send_command("show sip-ua register status | i <trunk name>")
    try:
        a = command1.split()
        a = a[3]
        if a == "yes":
            status = 1
            field = node
            return status, field
        elif a == "no":
            status = 2
            field = node
            return status, field
        else:
            status =0
            field = node
            return status, field

    except:
        status = 0
        field = node
        return status,field

def writeDB(output1,output2):
    json = [
        {
            "measurement": "python",
            "tags":
              {
                  "host": output1
              },
              "fields": { output1 : output2 }
        }
    ]
    #print(json)
    client = InfluxDBClient(host='127.0.0.1', port=8086, timeout=5)
    client.switch_database('telegraf')
    client.write_points(json)

cisco1 = {
    "host": "<IOS_XE_IP>",
    "username": "admin",
    "password": "C1sc0",
    "device_type": "cisco_xr",
}

net_connect = Netmiko(**cisco1)

trunkStatus(f"trunk-status")
writeDB(field,status)

net_connect.disconnect()