Okta Network Zone - Auto Set your Home IP

okta_home_netzone.py

#!/usr/bin/env python

#
## Script to update an Okta Network Zome named "Home Allowed" with your home ISP's public IP address. 
## Ensure there is only 1 IP range in this Network Zone. 
##
## Created by: Adam Matthews
## Date: 11th Jan 2023
#

import requests
import json

policyName = "Home Allowed" # Change to match your policy name

orgURL = "https://<<yourorgtenant>>.oktapreview.com"
apiToken = "<<yourapitoken>>"

url = orgURL+"/api/v1/zones"

payload={}
headers = {
  'Authorization': 'SSWS '+apiToken,
  'Content-Type': 'application/json'
}

def get():
    endpoint = 'https://ipinfo.io/json'
    response = requests.get(endpoint, verify = True)

    if response.status_code != 200:
        return 'Status:', response.status_code, 'Problem with the request. Exiting.'
        exit()

    data = response.json()

    return data['ip']

#get my ip
my_ip = get()

response = requests.request("GET", url, headers=headers, data=payload)

jsonObject = json.loads(response.text)

ipRange = (my_ip+"-"+my_ip) # build single range for your hope IP

for key in jsonObject:
	if key['name'] == policyName:
		if key['gateways'][0]['value'] == ipRange :
			print("No Change")
		else: # IP has changes, we need to update Okta. 
			key['gateways'][0]['value'] = ipRange

			url = orgURL+"/api/v1/zones/"+key['id']
			payload = json.dumps(key)

			response = requests.request("PUT", url, headers=headers, data=payload)

			print(response.text)