tasks.py

from dns import tsigkeyring, update, query
from dns.tsig import HMAC_SHA256
from scrapli.driver.core import IOSXEDriver, NXOSDriver
import time
from config import *


def doDNSUpdate(data):    
    keyring = tsigkeyring.from_text({data.tsig_keyname: data.tsig_secret })
    IP = str(data.subst_IP) if data.ddos else str(data.prim_IP)

    upd = update.Update(ZONE, keyring=keyring, keyalgorithm=HMAC_SHA256)
    upd.replace(data.name, TTL, 'A', IP)
    response = query.tcp(upd, DNS_SERV, timeout=10)
    print("DNS update done!")
    return response


def waitTTLsecs(ttl):
    time.sleep(ttl)
    print("times's up")


def blackHole(ip, ddos=True):
    with NXOSDriver(**connection_options) as conn:     
        if ddos:
            cmd = f"ip route {ip} 255.255.255.255 Null0 tag 666"
        else:
            cmd = f"no ip route {ip} 255.255.255.255 Null0"

        res = conn.send_config(cmd)

        if not res.failed:            
            p = "RTHB went good" if ddos else "deativating RTHB went good"            
        else:
            p = "RTHB didn't make it" if ddos else "deactivating RTHB didn't make it"
        print(p)